diff --git a/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/report.py b/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/report.py index f12ca8a90..97f012414 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/report.py +++ b/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/report.py @@ -16,7 +16,7 @@ from app.agents.shared.receipt import make_receipt from app.agents.shared.receipt_command import with_receipt from app.db import Report, shielded_async_session from app.services.connector_service import ConnectorService -from app.services.llm_service import get_document_summary_llm +from app.services.llm_service import get_agent_llm logger = logging.getLogger(__name__) @@ -767,7 +767,7 @@ def create_generate_report_tool( "creating standalone report" ) - llm = await get_document_summary_llm(read_session, search_space_id) + llm = await get_agent_llm(read_session, search_space_id) # read_session closed — connection returned to pool if not llm: diff --git a/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/resume.py b/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/resume.py index ad16b7ba7..86332ccbe 100644 --- a/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/resume.py +++ b/surfsense_backend/app/agents/multi_agent_chat/subagents/builtins/deliverables/tools/resume.py @@ -17,7 +17,7 @@ from langgraph.types import Command from app.agents.shared.receipt import make_receipt from app.agents.shared.receipt_command import with_receipt from app.db import Report, shielded_async_session -from app.services.llm_service import get_document_summary_llm +from app.services.llm_service import get_agent_llm logger = logging.getLogger(__name__) @@ -578,7 +578,7 @@ def create_generate_resume_tool( f"(group {report_group_id})" ) - llm = await get_document_summary_llm(read_session, search_space_id) + llm = await get_agent_llm(read_session, search_space_id) if not llm: error_msg = ( diff --git a/surfsense_backend/app/agents/new_chat/tools/report.py b/surfsense_backend/app/agents/new_chat/tools/report.py index 6bc1b7d57..e8629aef3 100644 --- a/surfsense_backend/app/agents/new_chat/tools/report.py +++ b/surfsense_backend/app/agents/new_chat/tools/report.py @@ -35,7 +35,7 @@ from langchain_core.tools import tool from app.db import Report, shielded_async_session from app.services.connector_service import ConnectorService -from app.services.llm_service import get_document_summary_llm +from app.services.llm_service import get_agent_llm logger = logging.getLogger(__name__) @@ -768,7 +768,7 @@ def create_generate_report_tool( "creating standalone report" ) - llm = await get_document_summary_llm(read_session, search_space_id) + llm = await get_agent_llm(read_session, search_space_id) # read_session closed — connection returned to pool if not llm: diff --git a/surfsense_backend/app/agents/new_chat/tools/resume.py b/surfsense_backend/app/agents/new_chat/tools/resume.py index 4abe48ba6..17849bce7 100644 --- a/surfsense_backend/app/agents/new_chat/tools/resume.py +++ b/surfsense_backend/app/agents/new_chat/tools/resume.py @@ -26,7 +26,7 @@ from langchain_core.messages import HumanMessage from langchain_core.tools import tool from app.db import Report, shielded_async_session -from app.services.llm_service import get_document_summary_llm +from app.services.llm_service import get_agent_llm logger = logging.getLogger(__name__) @@ -547,7 +547,7 @@ def create_generate_resume_tool( f"(group {report_group_id})" ) - llm = await get_document_summary_llm(read_session, search_space_id) + llm = await get_agent_llm(read_session, search_space_id) if not llm: error_msg = (