highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-22 22:44:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Adding research topics for better search context (#4448)

Browse Source 
* research topics addition

* allow for question to overwrite research area
This commit is contained in:
joachim-danswer 2025-04-04 09:53:39 -07:00 committed by GitHub
parent 9dd56a5c80
commit 68f9f157a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
backend/onyx/agents/agent_search/dc_search_analysis/nodes/a2_research_object_source.py
View File

@ -61,7 +61,11 @@ def research_object_source(
raise ValueError("Agent 2 task not found")
agent_2_time_cutoff = extract_section(
agent_2_instructions, "Time Cutoff:", "Output Objective:"
agent_2_instructions, "Time Cutoff:", "Research Topics:"
)
agent_2_research_topics = extract_section(
agent_2_instructions, "Research Topics:", "Output Objective"
)
agent_2_output_objective = extract_section(
@ -99,8 +103,15 @@ def research_object_source(
document_sources = [document_source] if document_source else None
if len(question.strip()) > 0:
research_area = f"{question} for {object}"
elif agent_2_research_topics and len(agent_2_research_topics.strip()) > 0:
research_area = f"{agent_2_research_topics} for {object}"
else:
research_area = object
retrieved_docs = research(
question=object,
question=research_area,
search_tool=search_tool,
document_sources=document_sources,
time_cutoff=agent_2_source_start_time,