danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Search tool in chat agent needs no search context

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-01-12 15:13:43 +02:00
parent 5adab04e50
commit 2da2c6d13a
No known key found for this signature in database
1 changed files with 1 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
haiku_rag_slim/haiku/rag/agents/chat/agent.py
View file

@ -9,7 +9,6 @@ from haiku.rag.agents.chat.state import (
CitationInfo, CitationInfo,
QAResponse, QAResponse,
build_document_filter, build_document_filter,
format_conversation_context,
rank_qa_history_by_similarity, rank_qa_history_by_similarity,
) )
from haiku.rag.agents.research.dependencies import ResearchContext from haiku.rag.agents.research.dependencies import ResearchContext
@ -46,17 +45,12 @@ def create_chat_agent(config: AppConfig) -> Agent[ChatDeps, str]:
query: The search query (what to search for) query: The search query (what to search for)
document_name: Optional document name/title to search within (e.g., "tbmed593", "army manual") document_name: Optional document name/title to search within (e.g., "tbmed593", "army manual")
""" """
# Build context from conversation history
context = None
if ctx.deps.session_state and ctx.deps.session_state.qa_history:
context = format_conversation_context(ctx.deps.session_state.qa_history)
# Build filter from document_name # Build filter from document_name
doc_filter = build_document_filter(document_name) if document_name else None doc_filter = build_document_filter(document_name) if document_name else None
# Use search agent for query expansion and deduplication # Use search agent for query expansion and deduplication
search_agent = SearchAgent(ctx.deps.client, ctx.deps.config) search_agent = SearchAgent(ctx.deps.client, ctx.deps.config)
results = await search_agent.search(query, context=context, filter=doc_filter) results = await search_agent.search(query, filter=doc_filter)
# Store for potential citation resolution # Store for potential citation resolution
ctx.deps.search_results = results ctx.deps.search_results = results