vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-01 18:19:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Updated section streaming for report sections.

Browse source
This commit is contained in:
Utkarsh-Patel-13 2025-07-19 21:08:19 -07:00
parent 92781e726c
commit ef761f1a43
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
surfsense_backend/app/agents/researcher/nodes.py
View file

@ -1065,6 +1065,13 @@ async def process_sections(state: State, config: RunnableConfig, writer: StreamW
# Skip adding the section header since the content already contains the title
final_report.append(content)
final_report.append("\n")
# Stream each section with its title
writer(
{
"yield_value": state.streaming_service.format_text_chunk(f"# {section.section_title}\n\n{content}")
}
)
# Join all sections with newlines
@ -1219,13 +1226,6 @@ async def process_section_with_documents(
complete_answer.extend(content_lines)
complete_answer.append("") # Empty line after content
# Update answer in UI in real-time
state.streaming_service.only_update_answer(complete_answer)
writer(
{
"yield_value": state.streaming_service._format_annotations()
}
)
# Set default if no content was received
if not complete_content: