mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-06 20:15:17 +02:00
feat(backend): Use agent LLM for deliverable generation
This commit is contained in:
parent
f3866b9e7e
commit
b5a9bcf87b
4 changed files with 8 additions and 8 deletions
|
|
@ -16,7 +16,7 @@ from app.agents.shared.receipt import make_receipt
|
||||||
from app.agents.shared.receipt_command import with_receipt
|
from app.agents.shared.receipt_command import with_receipt
|
||||||
from app.db import Report, shielded_async_session
|
from app.db import Report, shielded_async_session
|
||||||
from app.services.connector_service import ConnectorService
|
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__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
@ -767,7 +767,7 @@ def create_generate_report_tool(
|
||||||
"creating standalone report"
|
"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
|
# read_session closed — connection returned to pool
|
||||||
|
|
||||||
if not llm:
|
if not llm:
|
||||||
|
|
|
||||||
|
|
@ -17,7 +17,7 @@ from langgraph.types import Command
|
||||||
from app.agents.shared.receipt import make_receipt
|
from app.agents.shared.receipt import make_receipt
|
||||||
from app.agents.shared.receipt_command import with_receipt
|
from app.agents.shared.receipt_command import with_receipt
|
||||||
from app.db import Report, shielded_async_session
|
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__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
@ -578,7 +578,7 @@ def create_generate_resume_tool(
|
||||||
f"(group {report_group_id})"
|
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:
|
if not llm:
|
||||||
error_msg = (
|
error_msg = (
|
||||||
|
|
|
||||||
|
|
@ -35,7 +35,7 @@ from langchain_core.tools import tool
|
||||||
|
|
||||||
from app.db import Report, shielded_async_session
|
from app.db import Report, shielded_async_session
|
||||||
from app.services.connector_service import ConnectorService
|
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__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
@ -768,7 +768,7 @@ def create_generate_report_tool(
|
||||||
"creating standalone report"
|
"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
|
# read_session closed — connection returned to pool
|
||||||
|
|
||||||
if not llm:
|
if not llm:
|
||||||
|
|
|
||||||
|
|
@ -26,7 +26,7 @@ from langchain_core.messages import HumanMessage
|
||||||
from langchain_core.tools import tool
|
from langchain_core.tools import tool
|
||||||
|
|
||||||
from app.db import Report, shielded_async_session
|
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__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
@ -547,7 +547,7 @@ def create_generate_resume_tool(
|
||||||
f"(group {report_group_id})"
|
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:
|
if not llm:
|
||||||
error_msg = (
|
error_msg = (
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue