Query Rewriter Agent • AI Agent SDK

QueryRewriterAgent: Agent for rewriting queries to improve retrieval quality in RAG applications.

QueryRewriterAgent

Defined in the Query Rewriter Agent module.

Agent for rewriting queries to improve retrieval quality in RAG applications.

This agent transforms user queries using various strategies to bridge the gap between how users ask questions and how information is stored in knowledge bases.

Attributes: name: Name of the agent model: LLM model to use for rewriting (alias for llm=) max_queries: Maximum number of queries to generate for multi-query strategy abbreviations: Dictionary of common abbreviations to expand

Constructor

Methods

Usage

agent = QueryRewriterAgent(model="gpt-4o-mini")
    result = agent.rewrite("ML best practices", strategy=RewriteStrategy.MULTI_QUERY)
    for query in result.rewritten_queries:
        print(query)

Source

Related Documentation