From 32092c0b65914a046a0f6a02e76bfa6a561af71b Mon Sep 17 00:00:00 2001 From: CREDO23 Date: Fri, 8 May 2026 23:17:05 +0200 Subject: [PATCH] Pass thinkingStepId through tool-input start and available metadata. --- .../app/tasks/chat/streaming/handlers/tool_start.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/surfsense_backend/app/tasks/chat/streaming/handlers/tool_start.py b/surfsense_backend/app/tasks/chat/streaming/handlers/tool_start.py index 3b3537567..e0cac307c 100644 --- a/surfsense_backend/app/tasks/chat/streaming/handlers/tool_start.py +++ b/surfsense_backend/app/tasks/chat/streaming/handlers/tool_start.py @@ -99,20 +99,21 @@ def iter_tool_start_frames( ) span_md = state.span_metadata_if_active() + tool_md = state.tool_activity_metadata(thinking_step_id=tool_step_id) if matched_meta is None: yield streaming_service.format_tool_input_start( tool_call_id, tool_name, langchain_tool_call_id=langchain_tool_call_id, - metadata=span_md, + metadata=tool_md, ) if content_builder is not None: content_builder.on_tool_input_start( tool_call_id, tool_name, langchain_tool_call_id, - metadata=span_md, + metadata=tool_md, ) thinking = resolve_tool_start_thinking(tool_name, tool_input) @@ -148,7 +149,7 @@ def iter_tool_start_frames( tool_name, _safe_input, langchain_tool_call_id=langchain_tool_call_id, - metadata=span_md, + metadata=tool_md, ) if content_builder is not None: content_builder.on_tool_input_available( @@ -156,5 +157,5 @@ def iter_tool_start_frames( tool_name, _safe_input, langchain_tool_call_id, - metadata=span_md, + metadata=tool_md, )