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 {
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 !pending.arguments.trim().is_empty() {
let temp_tool_call = ToolCall {
id: pending.id.clone().unwrap_or_default(),
function: FunctionCall {
name: name.clone(),
arguments: pending.arguments.clone(),
},
call_type: pending
.call_type
.clone()
.unwrap_or_default(),
code_interpreter: None,
retrieval: None,
};
let temp_tool_call = ToolCall {
id: pending.id.clone().unwrap_or_default(),
function: FunctionCall {
name: name.clone(),
arguments: pending.arguments.clone(),
},
call_type: pending.call_type.clone().unwrap_or_default(),
code_interpreter: None,
retrieval: None,
};
let _ = tx
.send(Ok(Message::assistant(
None,
Some(vec![temp_tool_call]),
Some(MessageProgress::InProgress),
)))
.await;
}
let _ = tx
.send(Ok(Message::assistant(
None,
Some(vec![temp_tool_call]),
Some(MessageProgress::InProgress),
)))
.await;
}
}
}