agents.rag.agentic.query_rewriter

Query Rewriting Agent for Agentic RAG.

This agent rewrites queries to improve retrieval using existing models from common.

Functions

create_query_rewriter_agent([name, temperature])	Create a query rewriter agent using direct SimpleAgent instantiation.
rewrite_query(agent, query[, context])	Rewrite a query to improve retrieval.

Module Contents

agents.rag.agentic.query_rewriter.create_query_rewriter_agent(name='query_rewriter', temperature=0.7, **kwargs)

Create a query rewriter agent using direct SimpleAgent instantiation.

Parameters:

• name (str) – Agent name (default: “query_rewriter”)

• temperature (float) – LLM temperature (default: 0.7 for creativity)

• **kwargs – Additional configuration options

Returns:

SimpleAgent configured for query refinement

Return type:

haive.agents.simple.SimpleAgent

Example

# Create query rewriter agent
rewriter = create_query_rewriter_agent(
name="query_rewriter",
temperature=0.7
)

# Rewrite a query
result = await rewriter.arun({
"query": "quantum computing basics"
})

# Access results
print(f"Original: {result.original_query}")
print(f"Best rewrite: {result.best_refined_query}")
for suggestion in result.refinement_suggestions:
print(f"- {suggestion.refined_query} ({suggestion.improvement_type})")

async agents.rag.agentic.query_rewriter.rewrite_query(agent, query, context='')

Rewrite a query to improve retrieval.

Parameters:

• agent (haive.agents.simple.SimpleAgent) – The query rewriter agent

• query (str) – The original query to rewrite

• context (str) – Optional context to help with rewriting

Returns:

QueryRefinementResponse with refinement suggestions

Return type:

haive.agents.rag.common.query_refinement.models.QueryRefinementResponse