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

Set search filters from ducment filter in example

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-12-11 14:38:23 +02:00
parent 5f4f5f70f0
commit 39cbec1b88
No known key found for this signature in database
2 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/ag-ui-research/backend/agent.py
View file

@ -33,6 +33,7 @@ class AgentDeps:
client: HaikuRAG client: HaikuRAG
agui_emitter: "AGUIEmitter[ResearchState, ResearchReport] | None" = None agui_emitter: "AGUIEmitter[ResearchState, ResearchReport] | None" = None
search_filter: str | None = None
model = get_model(Config.research.model, Config) model = get_model(Config.research.model, Config)
@ -74,6 +75,7 @@ async def run_research(ctx: RunContext[AgentDeps], question: str) -> str:
graph = build_research_graph(Config) graph = build_research_graph(Config)
context = ResearchContext(original_question=question) context = ResearchContext(original_question=question)
state = ResearchState.from_config(context=context, config=Config) state = ResearchState.from_config(context=context, config=Config)
state.search_filter = ctx.deps.search_filter
graph_deps = ResearchDeps( graph_deps = ResearchDeps(
client=ctx.deps.client, client=ctx.deps.client,

13
examples/ag-ui-research/backend/main.py
View file

@ -92,8 +92,19 @@ async def stream_research_agent(request: Request) -> StreamingResponse:
effective_db_path = Path(effective_db_path) effective_db_path = Path(effective_db_path)
client = get_client(effective_db_path) client = get_client(effective_db_path)
# Build search filter from document IDs if provided
document_ids = input_data.state.get("documentFilter")
search_filter = None
if document_ids:
ids_str = ", ".join(f"'{id}'" for id in document_ids)
search_filter = f"id IN ({ids_str})"
# Create agent dependencies with shared emitter # Create agent dependencies with shared emitter
agent_deps = AgentDeps(client=client, agui_emitter=emitter) agent_deps = AgentDeps(
client=client,
agui_emitter=emitter,
search_filter=search_filter,
)
# Start run with empty initial state # Start run with empty initial state
emitter.start_run( emitter.start_run(