harivansh-afk/sandbox-agent
1
0
Fork 0
mirror of https://github.com/harivansh-afk/sandbox-agent.git synced 2026-04-21 02:04:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: delay idle event until turn finishes (#132)

Browse source
This commit is contained in:
NathanFlurry 2026-02-07 09:09:49 +00:00
parent 54d537fb23
commit 479c8468e8
No known key found for this signature in database
GPG key ID: 6A5F43A4F3241BCA
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
server/packages/sandbox-agent/src/opencode_compat.rs
View file

@ -240,6 +240,8 @@ struct OpenCodeSessionRuntime {
tool_name_by_call: HashMap<String, String>, tool_name_by_call: HashMap<String, String>,
/// Tool arguments by call_id, persisted from ToolCall for use in ToolResult events /// Tool arguments by call_id, persisted from ToolCall for use in ToolResult events
tool_args_by_call: HashMap<String, String>, tool_args_by_call: HashMap<String, String>,
/// Tool calls that have been requested but not yet resolved.
open_tool_calls: HashSet<String>,
} }
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
@ -1822,6 +1824,13 @@ async fn apply_universal_event(state: Arc<OpenCodeAppState>, event: UniversalEve
{ {
if let Some(ContentPart::Status { label, .. }) = item.content.first() { if let Some(ContentPart::Status { label, .. }) = item.content.first() {
if label == "turn.completed" || label == "session.idle" { if label == "turn.completed" || label == "session.idle" {
let runtime = state
.opencode
.update_runtime(&event.session_id, |_| {})
.await;
if !runtime.open_tool_calls.is_empty() {
return;
}
let session_id = event.session_id.clone(); let session_id = event.session_id.clone();
state.opencode.emit_event(json!({ state.opencode.emit_event(json!({
"type": "session.status", "type": "session.status",
@ -2234,6 +2243,7 @@ async fn apply_item_event(
runtime runtime
.tool_args_by_call .tool_args_by_call
.insert(call_id.clone(), arguments.clone()); .insert(call_id.clone(), arguments.clone());
runtime.open_tool_calls.insert(call_id.clone());
}) })
.await; .await;
} }
@ -2291,6 +2301,7 @@ async fn apply_item_event(
runtime runtime
.tool_message_by_call .tool_message_by_call
.insert(call_id.clone(), message_id.clone()); .insert(call_id.clone(), message_id.clone());
runtime.open_tool_calls.remove(call_id);
}) })
.await; .await;
} }
@ -2556,6 +2567,14 @@ async fn apply_tool_item_event(
.tool_args_by_call .tool_args_by_call
.insert(call_id.clone(), args.clone()); .insert(call_id.clone(), args.clone());
} }
if item.kind == ItemKind::ToolCall {
runtime.open_tool_calls.insert(call_id.clone());
}
if item.kind == ItemKind::ToolResult
&& event.event_type == UniversalEventType::ItemCompleted
{
runtime.open_tool_calls.remove(&call_id);
}
}) })
.await; .await;
} }