CalvinBackup/claw-code-parity
1
0
Fork 0
mirror of https://github.com/ultraworkers/claw-code-parity.git synced 2026-04-28 15:56:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): split openai compat stream state into nested flags

Browse Source 
Resolves clippy::struct_excessive_bools in openai_compat.rs.

Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent)

Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
YeonGyu-Kim
2026-04-02 14:57:07 +09:00
parent 464a870180
commit 23ba21bdd5
1 changed files with 25 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rust/crates/api/src/providers/openai_compat.rs
+25 -17
View File
@@ -299,23 +299,31 @@ impl OpenAiSseParser {
#[derive(Debug)]
struct StreamState {
model: String,
message_started: bool,
text_started: bool,
text_finished: bool,
finished: bool,
message: MessageState,
text: TextState,
stop_reason: Option<String>,
usage: Option<Usage>,
tool_calls: BTreeMap<u32, ToolCallState>,
}
#[derive(Debug, Default)]
struct MessageState {
started: bool,
finished: bool,
}
#[derive(Debug, Default)]
struct TextState {
started: bool,
finished: bool,
}
impl StreamState {
fn new(model: String) -> Self {
Self {
model,
message_started: false,
text_started: false,
text_finished: false,
finished: false,
message: MessageState::default(),
text: TextState::default(),
stop_reason: None,
usage: None,
tool_calls: BTreeMap::new(),
@@ -324,8 +332,8 @@ impl StreamState {
fn ingest_chunk(&mut self, chunk: ChatCompletionChunk) -> Result<Vec<StreamEvent>, ApiError> {
let mut events = Vec::new();
if !self.message_started {
self.message_started = true;
if !self.message.started {
self.message.started = true;
events.push(StreamEvent::MessageStart(MessageStartEvent {
message: MessageResponse {
id: chunk.id.clone(),
@@ -357,8 +365,8 @@ impl StreamState {
for choice in chunk.choices {
if let Some(content) = choice.delta.content.filter(|value| !value.is_empty()) {
if !self.text_started {
self.text_started = true;
if !self.text.started {
self.text.started = true;
events.push(StreamEvent::ContentBlockStart(ContentBlockStartEvent {
index: 0,
content_block: OutputContentBlock::Text {
@@ -414,14 +422,14 @@ impl StreamState {
}
fn finish(&mut self) -> Result<Vec<StreamEvent>, ApiError> {
if self.finished {
if self.message.finished {
return Ok(Vec::new());
}
self.finished = true;
self.message.finished = true;
let mut events = Vec::new();
if self.text_started && !self.text_finished {
self.text_finished = true;
if self.text.started && !self.text.finished {
self.text.finished = true;
events.push(StreamEvent::ContentBlockStop(ContentBlockStopEvent {
index: 0,
}));
@@ -445,7 +453,7 @@ impl StreamState {
}
}
if self.message_started {
if self.message.started {
events.push(StreamEvent::MessageDelta(MessageDeltaEvent {
delta: MessageDelta {
stop_reason: Some(