Hare/yoi
1
1
Fork 0
Code Actions Packages Releases Activity

chore: format Worker command telemetry

Browse Source
This commit is contained in:
Hare
2026-09-29 15:19:01 +09:00
parent 60137e0aad
commit 8c78e55b28
1 changed files with 19 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/worker/src/in_flight.rs
+19 -11
View File
@@ -284,11 +284,11 @@ impl InFlightInner {
CommandEvent::Started { CommandEvent::Started {
tool_call_id: None, .. tool_call_id: None, ..
} => false, } => false,
CommandEvent::Output { command_id, .. } CommandEvent::Output { command_id, .. } | CommandEvent::Terminal { command_id, .. } => {
| CommandEvent::Terminal { command_id, .. } => self self.commands
.commands .iter()
.iter() .any(|command| command.command_id == *command_id)
.any(|command| command.command_id == *command_id), }
} }
} }
@@ -433,9 +433,7 @@ impl InFlightInner {
} }
fn clear(&mut self) -> bool { fn clear(&mut self) -> bool {
if self.blocks.is_empty() if self.blocks.is_empty() && self.known_tool_call_ids.is_empty() && self.commands.is_empty()
&& self.known_tool_call_ids.is_empty()
&& self.commands.is_empty()
{ {
false false
} else { } else {
@@ -764,7 +762,10 @@ mod tests {
let mut rx = working_event_tx.subscribe(); let mut rx = working_event_tx.subscribe();
let in_flight = InFlightEvents::new(working_event_tx); let in_flight = InFlightEvents::new(working_event_tx);
in_flight.tool_call_start("tool-1".into(), "Bash".into()); in_flight.tool_call_start("tool-1".into(), "Bash".into());
assert!(matches!(rx.try_recv().unwrap(), Event::ToolCallStart { .. })); assert!(matches!(
rx.try_recv().unwrap(),
Event::ToolCallStart { .. }
));
in_flight.publish_command_event(CommandEvent::Started { in_flight.publish_command_event(CommandEvent::Started {
command_id: "command-1".into(), command_id: "command-1".into(),
tool_call_id: Some("tool-1".into()), tool_call_id: Some("tool-1".into()),
@@ -817,7 +818,10 @@ mod tests {
let mut rx = working_event_tx.subscribe(); let mut rx = working_event_tx.subscribe();
let in_flight = InFlightEvents::new(working_event_tx); let in_flight = InFlightEvents::new(working_event_tx);
in_flight.tool_call_start("own-tool".into(), "Bash".into()); in_flight.tool_call_start("own-tool".into(), "Bash".into());
assert!(matches!(rx.try_recv().unwrap(), Event::ToolCallStart { .. })); assert!(matches!(
rx.try_recv().unwrap(),
Event::ToolCallStart { .. }
));
in_flight.publish_command_event(CommandEvent::Started { in_flight.publish_command_event(CommandEvent::Started {
command_id: "foreign-command".into(), command_id: "foreign-command".into(),
@@ -842,7 +846,11 @@ mod tests {
}); });
assert!(rx.try_recv().is_err()); assert!(rx.try_recv().is_err());
assert!(snapshot_from_guard(&in_flight.snapshot_guard()).commands.is_empty()); assert!(
snapshot_from_guard(&in_flight.snapshot_guard())
.commands
.is_empty()
);
in_flight.replace_command_snapshot(vec![ in_flight.replace_command_snapshot(vec![
command_snapshot("own-command", Some("own-tool")), command_snapshot("own-command", Some("own-tool")),