apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-10 08:12:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix: Fix rerank_documents node in sub_section_writer & qna_agent

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-07-03 14:08:32 -07:00
parent d005f810f1
commit f99878c07c
6 changed files with 2708 additions and 1923 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2,7 +2,7 @@ from .configuration import Configuration
from langchain_core.runnables import RunnableConfig
from .state import State
from typing import Any, Dict
from app.config import config as app_config
from app.utils.reranker_service import RerankerService
from .prompts import get_citation_system_prompt, get_no_documents_system_prompt
from langchain_core.messages import HumanMessage, SystemMessage
from .configuration import SubSectionType
@ -35,7 +35,7 @@ async def rerank_documents(state: State, config: RunnableConfig) -> Dict[str, An
}
# Get reranker service from app config
reranker_service = getattr(app_config, "reranker_service", None)
reranker_service = RerankerService.get_reranker_instance()
# Use documents as is if no reranker service is available
reranked_docs = documents