query • AI Agent SDK
query: Query knowledge and get a structured answer with citations.
query
Method
</div>
Query knowledge and get a structured answer with citations.
This is the recommended method for getting answers with source citations. Returns a structured result with answer, citations, context used, and metadata.
Signature
def query(question: str) -> 'RAGResult'Parameters
ParamField
strThe question to answer **kwargs: Additional arguments (top_k, rerank, etc.)
Returns
ResponseField
'RAGResult'RAGResult with answer, citations, context_used, and metadata
Exceptions
ValueError
If no knowledge is configured
ImportError
If RAG module is not available
Usage
agent = Agent(knowledge=["doc.pdf"], retrieval_config={"citations": True})
result = agent.query("What is the main finding?")
print(result.answer)
for citation in result.citations:
print(f" [{citation.id}] {citation.source}")Uses
ValueErrorImportErrorkwargs.setdefaultquery
Used By
Agent.queryAgent.rag_queryChromaMemoryAdapter.search_long_termMemory.search_short_termMemory.search_long_term
Source
View on GitHubpraisonaiagents/agent/agent.py at line 6859