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

refactor: optimize report generation tool to use short-lived database sessions for improved performance and connection management

Browse source
This commit is contained in:
Anish Sarkar 2026-02-14 18:48:36 +05:30
parent e1124d170d
commit 3ec30d94ce
2 changed files with 93 additions and 76 deletions
Download patch file Download diff file Expand all files Collapse all files

5
surfsense_backend/app/agents/new_chat/tools/registry.py
View file

@ -119,16 +119,15 @@ BUILTIN_TOOLS: list[ToolDefinition] = [
),
requires=["search_space_id", "db_session", "thread_id"],
),
# Report generation tool (inline, no Celery)
# Report generation tool (inline, short-lived sessions for DB ops)
ToolDefinition(
name="generate_report",
description="Generate a structured Markdown report from provided content",
factory=lambda deps: create_generate_report_tool(
search_space_id=deps["search_space_id"],
db_session=deps["db_session"],
thread_id=deps["thread_id"],
),
requires=["search_space_id", "db_session", "thread_id"],
requires=["search_space_id", "thread_id"],
),
# Link preview tool - fetches Open Graph metadata for URLs
ToolDefinition(