Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

ok streaming it back

This commit is contained in:
dal 2025-02-10 12:29:13 -07:00
parent 8c8372b50e
commit 118ef9c691
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 18 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
api/src/utils/agent/agent.rs
View File

@ -424,31 +424,26 @@ impl Agent {
for tool_call in tool_calls { for tool_call in tool_calls {
pending.update_from_delta(tool_call); pending.update_from_delta(tool_call);
// Only send intermediate updates if we have a function name // Send an update if we have a name, regardless of arguments
if let Some(name) = &pending.function_name { if let Some(name) = &pending.function_name {
if !pending.arguments.trim().is_empty() { let temp_tool_call = ToolCall {
let temp_tool_call = ToolCall { id: pending.id.clone().unwrap_or_default(),
id: pending.id.clone().unwrap_or_default(), function: FunctionCall {
function: FunctionCall { name: name.clone(),
name: name.clone(), arguments: pending.arguments.clone(),
arguments: pending.arguments.clone(), },
}, call_type: pending.call_type.clone().unwrap_or_default(),
call_type: pending code_interpreter: None,
.call_type retrieval: None,
.clone() };
.unwrap_or_default(),
code_interpreter: None,
retrieval: None,
};
let _ = tx let _ = tx
.send(Ok(Message::assistant( .send(Ok(Message::assistant(
None, None,
Some(vec![temp_tool_call]), Some(vec![temp_tool_call]),
Some(MessageProgress::InProgress), Some(MessageProgress::InProgress),
))) )))
.await; .await;
}
} }
} }
} }