Struct EmbeddingsClient 

pub struct EmbeddingsClient { /* private fields */ }

Client that wraps an embeddings service and keeps track of the model name.

Implementations

impl EmbeddingsClient

pub fn new(model: impl Into<String>) -> Self

Create a client that uses the default OpenAI configuration (OPENAI_API_KEY).

pub fn with_api_key( model: impl Into<String>, api_key: impl Into<String>, ) -> Self

Create a client that uses the provided API key.

pub fn with_config(model: impl Into<String>, config: OpenAIConfig) -> Self

Create a client that uses the provided OpenAI configuration.

pub fn with_service( model: impl Into<String>, service: Arc<dyn EmbeddingsService>, ) -> Self

Create a client around a custom embeddings service implementation.

pub async fn embed_batch<T>(&self, inputs: &[T]) -> Result<Vec<Vec<f32>>>

where T: AsRef<str>,

Ask the underlying service for embeddings for a batch of inputs.

pub async fn embed(&self, input: &str) -> Result<Vec<f32>>

Ask the underlying service for an embedding for a single input.

Trait Implementations

impl Clone for EmbeddingsClient

fn clone(&self) -> EmbeddingsClient

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.