Struct ExtractorBuilder 

pub struct ExtractorBuilder<M, T>
where
    M: CompletionModel,
    T: JsonSchema + for<'a> Deserialize<'a> + Serialize + WasmCompatSend + WasmCompatSync + 'static,
{ /* private fields */ }

Builder for the Extractor

Implementations

impl<M, T> ExtractorBuilder<M, T>

where M: CompletionModel, T: JsonSchema + for<'a> Deserialize<'a> + Serialize + WasmCompatSend + WasmCompatSync + 'static,

pub fn new(model: M) -> ExtractorBuilder<M, T>

pub fn preamble(self, preamble: &str) -> ExtractorBuilder<M, T>

Add additional preamble to the extractor

pub fn context(self, doc: &str) -> ExtractorBuilder<M, T>

Add a context document to the extractor

pub fn additional_params(self, params: Value) -> ExtractorBuilder<M, T>

pub fn max_tokens(self, max_tokens: u64) -> ExtractorBuilder<M, T>

Set the maximum number of tokens for the completion

pub fn retries(self, retries: u64) -> ExtractorBuilder<M, T>

Set the maximum number of retries for the extractor.

pub fn tool_choice(self, choice: ToolChoice) -> ExtractorBuilder<M, T>

Set the tool_choice option for the inner Agent.

pub fn build(self) -> Extractor<M, T>

Build the Extractor

pub fn dynamic_context( self, sample: usize, dynamic_context: impl VectorStoreIndexDyn + Send + Sync + 'static, ) -> ExtractorBuilder<M, T>

Add dynamic context (RAG) to the extractor.

On each prompt, sample documents will be retrieved from the index based on the RAG text and inserted in the request.