apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-26 01:06:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: enhance report generation with length constraints and user-friendly language guidelines

Browse source
This commit is contained in:
Anish Sarkar 2026-02-19 23:33:08 +05:30
parent e059ee4193
commit e568ded743
2 changed files with 24 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

19
surfsense_backend/app/agents/new_chat/tools/report.py
View file

@ -78,6 +78,8 @@ _REPORT_PROMPT = """You are an expert report writer. Generate a comprehensive Ma
---
{length_instruction}
Write a well-structured Markdown report with a # title, executive summary, organized sections, and conclusion. Cite facts from the source content. Be thorough and professional.
{formatting_rules}
@ -102,6 +104,8 @@ _REVISION_PROMPT = """You are an expert report editor. Apply ONLY the requested
---
{length_instruction}
Preserve all structure and content not affected by the modification.
{formatting_rules}
@ -696,8 +700,7 @@ def create_generate_report_tool(
search_queries: When source_strategy is "kb_search" or "auto",
provide 1-5 targeted search queries for the knowledge base.
These should be specific, not just the topic repeated.
report_style: "detailed", "executive_summary", "deep_research",
or "brief".
report_style: "detailed", "deep_research", or "brief".
user_instructions: Optional focus or modification instructions.
When revising (parent_report_id set), describe WHAT TO CHANGE.
parent_report_id: ID of a previous report to revise (creates new
@ -897,6 +900,16 @@ def create_generate_report_tool(
capped_source = effective_source[:100000] # Cap source content
# Length constraint — only when user explicitly asks for brevity
length_instruction = ""
if report_style == "brief":
length_instruction = (
"**LENGTH CONSTRAINT (MANDATORY):** The user wants a SHORT report. "
"Keep it concise — aim for ~500 words (~1 page) unless a different "
"length is specified in the Additional Instructions above. "
"Prioritize brevity over thoroughness. Do NOT write a long report."
)
# ── Phase 2: LLM GENERATION (no DB connection held) ──────────
report_content: str | None = None
@ -944,6 +957,7 @@ def create_generate_report_tool(
or "Improve and refine the report.",
source_content=capped_source,
previous_report_content=parent_report_content,
length_instruction=length_instruction,
formatting_rules=_FORMATTING_RULES,
)
response = await llm.ainvoke([HumanMessage(content=prompt)])
@ -966,6 +980,7 @@ def create_generate_report_tool(
user_instructions_section=user_instructions_section,
previous_version_section="",
source_content=capped_source,
length_instruction=length_instruction,
formatting_rules=_FORMATTING_RULES,
)
response = await llm.ainvoke([HumanMessage(content=prompt)])