vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-11 04:50:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Document thinkingStepId on tool-call parts and first-key metadata merge.

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

13
surfsense_backend/app/tasks/chat/content_builder.py
View file

@ -52,6 +52,12 @@ _MEANINGFUL_PART_TYPES: frozenset[str] = frozenset({"text", "reasoning", "tool-c
def _merge_tool_part_metadata(part: dict[str, Any], metadata: dict[str, Any] | None) -> None:
"""Shallow-merge ``metadata`` into ``part["metadata"]``; first key wins.
Used for tool-call linkage (``spanId``, ``thinkingStepId``, …): a later
event must not overwrite an existing key so chunk order vs ``on_tool_start``
stays stable.
"""
if not metadata:
return
md = part.setdefault("metadata", {})
@ -70,6 +76,7 @@ class AssistantContentBuilder:
| { type: "reasoning"; text: string }
| { type: "tool-call"; toolCallId: str; toolName: str;
args: dict; result?: any; argsText?: str; langchainToolCallId?: str;
metadata?: { spanId?: str; thinkingStepId?: str; ... };
state?: "aborted" }
| { type: "data-thinking-steps"; data: { steps: ThinkingStepData[] } }
| { type: "data-step-separator"; data: { stepIndex: int } }
@ -189,7 +196,11 @@ class AssistantContentBuilder:
*,
metadata: dict[str, Any] | None = None,
) -> None:
"""Register a tool-call card. Args are filled in by later events."""
"""Register a tool-call card. Args are filled in by later events.
Optional ``metadata`` (``spanId``, ``thinkingStepId``, …) is stored on the
part; duplicate ``tool-input-start`` calls merge with first-key-wins.
"""
if not ui_id:
return
# Skip duplicate registration: the stream may emit