feat: add metric for per-turn tool count and add tmp_mem flag (#13456)

2026-07-01 00:31:56 +08:00 · 2026-03-04 11:25:58 +00:00
parent bda3c49dc4
commit e07eaff0d3
3 changed files with 31 additions and 5 deletions
@@ -118,6 +118,14 @@ impl ToolRegistry {
        let mcp_server_ref = mcp_server.as_deref();
        let mcp_server_origin_ref = mcp_server_origin.as_deref();

+        {
+            let mut active = invocation.session.active_turn.lock().await;
+            if let Some(active_turn) = active.as_mut() {
+                let mut turn_state = active_turn.turn_state.lock().await;
+                turn_state.tool_calls = turn_state.tool_calls.saturating_add(1);
+            }
+        }
+
        let handler = match self.handler(tool_name.as_ref()) {
            Some(handler) => handler,
            None => {