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

refactor: coderabbit

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-04-20 00:19:21 -07:00
parent 7be68ebf41
commit 796e326cc8
2 changed files with 15 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -12,7 +12,6 @@ from .sub_section_writer.graph import graph as sub_section_writer_graph
from app.utils.connector_service import ConnectorService from app.utils.connector_service import ConnectorService
from app.utils.reranker_service import RerankerService from app.utils.reranker_service import RerankerService
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
import copy
class Section(BaseModel): class Section(BaseModel):
"""A section in the answer outline.""" """A section in the answer outline."""
@ -133,7 +132,6 @@ async def fetch_relevant_documents(
""" """
# Initialize services # Initialize services
connector_service = ConnectorService(db_session) connector_service = ConnectorService(db_session)
reranker_service = RerankerService.get_reranker_instance(app_config)
all_raw_documents = [] # Store all raw documents before reranking all_raw_documents = [] # Store all raw documents before reranking
@ -289,13 +287,20 @@ async def process_sections(state: State, config: RunnableConfig) -> Dict[str, An
relevant_documents = [] relevant_documents = []
async with session_maker() as db_session: async with session_maker() as db_session:
relevant_documents = await fetch_relevant_documents( try:
research_questions=all_questions, relevant_documents = await fetch_relevant_documents(
user_id=configuration.user_id, research_questions=all_questions,
search_space_id=configuration.search_space_id, user_id=configuration.user_id,
db_session=db_session, search_space_id=configuration.search_space_id,
connectors_to_search=configuration.connectors_to_search db_session=db_session,
) connectors_to_search=configuration.connectors_to_search
)
except Exception as e:
print(f"Error fetching relevant documents: {str(e)}")
# Log the error and continue with an empty list of documents
# This allows the process to continue, but the report might lack information
relevant_documents = []
# Consider adding more robust error handling or reporting if needed
print(f"Fetched {len(relevant_documents)} relevant documents for all sections") print(f"Fetched {len(relevant_documents)} relevant documents for all sections")

2
surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py
View file

@ -1,7 +1,7 @@
from .configuration import Configuration from .configuration import Configuration
from langchain_core.runnables import RunnableConfig from langchain_core.runnables import RunnableConfig
from .state import State from .state import State
from typing import Any, Dict, List from typing import Any, Dict
from app.config import config as app_config from app.config import config as app_config
from .prompts import get_citation_system_prompt from .prompts import get_citation_system_prompt
from langchain_core.messages import HumanMessage, SystemMessage from langchain_core.messages import HumanMessage, SystemMessage