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

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

@ -424,19 +424,15 @@ 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: pending.call_type.clone().unwrap_or_default(),
.call_type
.clone()
.unwrap_or_default(),
code_interpreter: None, code_interpreter: None,
retrieval: None, retrieval: None,
}; };
@ -453,7 +449,6 @@ impl Agent {
} }
} }
} }
}
Err(e) => { Err(e) => {
let _ = tx.send(Err(anyhow::Error::from(e))).await; let _ = tx.send(Err(anyhow::Error::from(e))).await;
return Ok(()); return Ok(());