apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-21 18:55:16 +02:00
Code Issues Projects Releases Packages Wiki Activity

refactor: enhance report generation tool with improved source strategies and revision handling

Browse source
This commit is contained in:
Anish Sarkar 2026-02-19 15:33:32 +05:30
parent f3ec48fb00
commit 69dfe943b7
4 changed files with 843 additions and 172 deletions
Download patch file Download diff file Expand all files Collapse all files

73
surfsense_backend/app/tasks/chat/stream_new_chat.py
View file

@ -387,26 +387,19 @@ async def _stream_agent_events(
if isinstance(tool_input, dict)
else "Report"
)
report_style = (
tool_input.get("report_style", "detailed")
if isinstance(tool_input, dict)
else "detailed"
is_revision = bool(
isinstance(tool_input, dict)
and tool_input.get("parent_report_id")
)
content_len = len(
tool_input.get("source_content", "")
if isinstance(tool_input, dict)
else ""
)
last_active_step_title = "Generating report"
step_title = "Revising report" if is_revision else "Generating report"
last_active_step_title = step_title
last_active_step_items = [
f"Topic: {report_topic}",
f"Style: {report_style}",
f"Source content: {content_len:,} characters",
"Generating report with LLM...",
"Analyzing source content...",
]
yield streaming_service.format_thinking_step(
step_id=tool_step_id,
title="Generating report",
title=step_title,
status="in_progress",
items=last_active_step_items,
)
@ -594,12 +587,18 @@ async def _stream_agent_events(
if isinstance(tool_output, dict)
else 0
)
is_revision = (
tool_output.get("is_revision", False)
if isinstance(tool_output, dict)
else False
)
step_title = "Revising report" if is_revision else "Generating report"
if report_status == "ready":
completed_items = [
f"Title: {report_title}",
f"Words: {word_count:,}",
"Report generated successfully",
f"Topic: {report_title}",
f"{word_count:,} words",
"Report ready",
]
elif report_status == "failed":
error_msg = (
@ -608,7 +607,7 @@ async def _stream_agent_events(
else "Unknown error"
)
completed_items = [
f"Title: {report_title}",
f"Topic: {report_title}",
f"Error: {error_msg[:50]}",
]
else:
@ -616,7 +615,7 @@ async def _stream_agent_events(
yield streaming_service.format_thinking_step(
step_id=original_step_id,
title="Generating report",
title=step_title,
status="completed",
items=completed_items,
)
@ -817,6 +816,42 @@ async def _stream_agent_events(
f"Tool {tool_name} completed", "success"
)
elif event_type == "on_custom_event" and event.get("name") == "report_progress":
# Live progress updates from inside the generate_report tool
data = event.get("data", {})
message = data.get("message", "")
if message and last_active_step_id:
phase = data.get("phase", "")
# Always keep the "Topic: ..." line
topic_items = [
item for item in last_active_step_items
if item.startswith("Topic:")
]
if phase in ("revising_section", "adding_section"):
# During section-level ops: keep plan summary + show current op
plan_items = [
item for item in last_active_step_items
if item.startswith("Topic:") or item.startswith("Modifying ")
or item.startswith("Adding ") or item.startswith("Removing ")
]
# Only keep plan_items that don't end with "..." (not progress lines)
plan_items = [
item for item in plan_items
if not item.endswith("...")
]
last_active_step_items = [*plan_items, message]
else:
# Phase transitions: replace everything after topic
last_active_step_items = [*topic_items, message]
yield streaming_service.format_thinking_step(
step_id=last_active_step_id,
title=last_active_step_title,
status="in_progress",
items=last_active_step_items,
)
elif event_type in ("on_chain_end", "on_agent_end"):
if current_text_id is not None:
yield streaming_service.format_text_end(current_text_id)