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

Merge pull request #151 from MODSetter/dev

Browse source 
fix: q/a agent
This commit is contained in:
Rohan Verma 2025-06-06 00:03:47 -07:00 • committed by GitHub
commit ecf0bccbbf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1174,7 +1174,7 @@ async def handle_qna_workflow(state: State, config: RunnableConfig, writer: Stre
await connector_service.initialize_counter()
# Use the reformulated query as a single research question
research_questions = [reformulated_query]
research_questions = [reformulated_query, user_query]
relevant_documents = await fetch_relevant_documents(
research_questions=research_questions,
@ -1210,7 +1210,8 @@ async def handle_qna_workflow(state: State, config: RunnableConfig, writer: Stre
# Prepare configuration for the QNA agent
qna_config = {
"configurable": {
"user_query": reformulated_query, # Use the reformulated query
"user_query": user_query, # Use the reformulated query
"reformulated_query": reformulated_query,
"relevant_documents": all_documents, # Use combined documents
"user_id": configuration.user_id,
"search_space_id": configuration.search_space_id

1
surfsense_backend/app/agents/researcher/qna_agent/configuration.py
View file

@ -14,6 +14,7 @@ class Configuration:
# Configuration parameters for the Q&A agent
user_query: str # The user's question to answer
reformulated_query: str # The reformulated query
relevant_documents: List[Any] # Documents provided directly to the agent for answering
user_id: str # User identifier
search_space_id: int # Search space identifier

3
surfsense_backend/app/agents/researcher/qna_agent/nodes.py
View file

@ -26,6 +26,7 @@ async def rerank_documents(state: State, config: RunnableConfig) -> Dict[str, An
configuration = Configuration.from_runnable_config(config)
documents = configuration.relevant_documents
user_query = configuration.user_query
reformulated_query = configuration.reformulated_query
# If no documents were provided, return empty list
if not documents or len(documents) == 0:
@ -57,7 +58,7 @@ async def rerank_documents(state: State, config: RunnableConfig) -> Dict[str, An
]
# Rerank documents using the user's query
reranked_docs = reranker_service.rerank_documents(user_query, reranker_input_docs)
reranked_docs = reranker_service.rerank_documents(user_query + "\n" + reformulated_query, reranker_input_docs)
# Sort by score in descending order
reranked_docs.sort(key=lambda x: x.get("score", 0), reverse=True)