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

refactor(chat): stream agent events via stream_output and remove parity v2 flag

Browse source
This commit is contained in:
CREDO23 2026-05-07 19:40:10 +02:00
parent 7e07092f67
commit 78f4747382
17 changed files with 76 additions and 1676 deletions
Download patch file Download diff file Expand all files Collapse all files

3
surfsense_backend/app/tasks/chat/streaming/relay/state.py
View file

@ -22,7 +22,6 @@ class AgentEventRelayState:
active_tool_depth: int = 0
called_update_memory: bool = False
current_reasoning_id: str | None = None
parity_v2: bool = False
pending_tool_call_chunks: list[dict[str, Any]] = field(default_factory=list)
lc_tool_call_id_by_run: dict[str, str] = field(default_factory=dict)
file_path_by_run: dict[str, str] = field(default_factory=dict)
@ -39,7 +38,6 @@ class AgentEventRelayState:
initial_step_id: str | None = None,
initial_step_title: str = "",
initial_step_items: list[str] | None = None,
parity_v2: bool,
) -> AgentEventRelayState:
counter = 1 if initial_step_id else 0
return cls(
@ -47,7 +45,6 @@ class AgentEventRelayState:
last_active_step_id=initial_step_id,
last_active_step_title=initial_step_title,
last_active_step_items=list(initial_step_items or []),
parity_v2=parity_v2,
)
def next_thinking_step_id(self, step_prefix: str) -> str: