apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-12 17:22:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Pass thinkingStepId through tool-input start and available metadata.

Browse source
This commit is contained in:
CREDO23 2026-05-08 23:17:05 +02:00
parent a309e830d3
commit 32092c0b65
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
surfsense_backend/app/tasks/chat/streaming/handlers/tool_start.py
View file

@ -99,20 +99,21 @@ def iter_tool_start_frames(
) )
span_md = state.span_metadata_if_active() span_md = state.span_metadata_if_active()
tool_md = state.tool_activity_metadata(thinking_step_id=tool_step_id)
if matched_meta is None: if matched_meta is None:
yield streaming_service.format_tool_input_start( yield streaming_service.format_tool_input_start(
tool_call_id, tool_call_id,
tool_name, tool_name,
langchain_tool_call_id=langchain_tool_call_id, langchain_tool_call_id=langchain_tool_call_id,
metadata=span_md, metadata=tool_md,
) )
if content_builder is not None: if content_builder is not None:
content_builder.on_tool_input_start( content_builder.on_tool_input_start(
tool_call_id, tool_call_id,
tool_name, tool_name,
langchain_tool_call_id, langchain_tool_call_id,
metadata=span_md, metadata=tool_md,
) )
thinking = resolve_tool_start_thinking(tool_name, tool_input) thinking = resolve_tool_start_thinking(tool_name, tool_input)
@ -148,7 +149,7 @@ def iter_tool_start_frames(
tool_name, tool_name,
_safe_input, _safe_input,
langchain_tool_call_id=langchain_tool_call_id, langchain_tool_call_id=langchain_tool_call_id,
metadata=span_md, metadata=tool_md,
) )
if content_builder is not None: if content_builder is not None:
content_builder.on_tool_input_available( content_builder.on_tool_input_available(
@ -156,5 +157,5 @@ def iter_tool_start_frames(
tool_name, tool_name,
_safe_input, _safe_input,
langchain_tool_call_id, langchain_tool_call_id,
metadata=span_md, metadata=tool_md,
) )