Rewrite Result • Rust AI Agent SDK

RewriteResult: Result of query rewriting

RewriteResult

Defined in the Specialized Agents module.

Rust AI Agent SDK

Result of query rewriting

Fields

NameTypeDescription
original_queryStringOriginal query
rewritten_queriesVec<String>Rewritten queries
strategy_usedRewriteStrategyStrategy used
hypothetical_documentOption<String>Hypothetical document (for HyDE strategy)
step_back_questionOption<String>Step-back question (for step-back strategy)
sub_queriesOption<Vec<String>>Sub-queries (for sub-queries strategy)
metadataHashMap<StringAdditional metadata
serde_json:Value>Additional metadata

Methods

new

rust
fn new(
        original: impl Into<String>,
        rewritten: Vec<String>,
        strategy: RewriteStrategy,
    ) -> Self

Create a new rewrite result

Parameters:

NameType
originalimpl Into&lt;String&gt;
rewrittenVec&lt;String&gt;
strategyRewriteStrategy

with_hypothetical_document

rust
fn with_hypothetical_document(mut self, doc: impl Into<String>) -> Self

Set hypothetical document

Parameters:

NameType
docimpl Into&lt;String&gt;

with_step_back_question

rust
fn with_step_back_question(mut self, question: impl Into<String>) -> Self

Set step-back question

Parameters:

NameType
questionimpl Into&lt;String&gt;

with_sub_queries

rust
fn with_sub_queries(mut self, queries: Vec<String>) -> Self

Set sub-queries

Parameters:

NameType
queriesVec&lt;String&gt;

with_metadata

rust
fn with_metadata(mut self, key: impl Into<String>, value: impl Into<serde_json::Value>) -> Self

Add metadata

Parameters:

NameType
keyimpl Into&lt;String&gt;
valueimpl Into&lt;serde_json::Value&gt;

primary_query

rust
fn primary_query(&self) -> Option<&str>

Get the primary rewritten query

all_queries

rust
fn all_queries(&self) -> Vec<&str>

Get all queries for retrieval

Source

View on GitHub

praisonai/src/specialized_agents.rs at line 457