apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-08 20:25:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: capture agent text from on_chat_model_end for non-streaming LLM calls

Browse source
This commit is contained in:
CREDO23 2026-04-07 18:06:40 +02:00
parent e8f82a0590
commit 5f354ef6ac
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
surfsense_backend/app/agents/autocomplete/autocomplete_agent.py
View file

@ -275,7 +275,7 @@ async def create_autocomplete_agent(
# ---------------------------------------------------------------------------
# JSON suggestion parsing (robust fallback)
# JSON suggestion parsing (with fallback)
# ---------------------------------------------------------------------------
@ -383,7 +383,6 @@ async def stream_autocomplete_agent(
input_data, config=config, version="v2"
):
event_type = event.get("event", "")
if event_type == "on_chat_model_stream":
if active_tool_depth > 0:
continue
@ -395,6 +394,17 @@ async def stream_autocomplete_agent(
if content and isinstance(content, str):
text_buffer.append(content)
elif event_type == "on_chat_model_end":
if active_tool_depth > 0:
continue
if "surfsense:internal" in event.get("tags", []):
continue
output = event.get("data", {}).get("output")
if output and hasattr(output, "content"):
content = output.content
if content and isinstance(content, str) and not text_buffer:
text_buffer.append(content)
elif event_type == "on_tool_start":
active_tool_depth += 1
tool_name = event.get("name", "unknown_tool")