From d3540d8cc5a26f786edc0a4c3e856d5d08ebbc0e Mon Sep 17 00:00:00 2001 From: "DESKTOP-RTLN3BA\\$punk" Date: Sat, 10 May 2025 20:19:08 -0700 Subject: [PATCH] chore: more context to reranker --- .../app/agents/researcher/sub_section_writer/nodes.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py b/surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py index d52a425..765b619 100644 --- a/surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py +++ b/surfsense_backend/app/agents/researcher/sub_section_writer/nodes.py @@ -39,7 +39,9 @@ async def rerank_documents(state: State, config: RunnableConfig) -> Dict[str, An try: # Use the sub-section questions for reranking context # rerank_query = "\n".join(sub_section_questions) - rerank_query = configuration.user_query + # rerank_query = configuration.user_query + + rerank_query = configuration.user_query + "\n" + "\n".join(sub_section_questions) # Convert documents to format expected by reranker if needed reranker_input_docs = [