Struct OpenAIEmbedding 

pub struct OpenAIEmbedding { /* private fields */ }

OpenAI embedding provider

Implementations

impl OpenAIEmbedding

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

Create a new OpenAI embedding provider

pub fn from_env() -> Result<Self>

Create from environment variable

pub fn with_model(self, model: impl Into<String>, dimension: usize) -> Self

Use a specific model

pub fn large() -> Result<Self>

Use text-embedding-3-large model

pub fn ada() -> Result<Self>

Use text-embedding-ada-002 model

Trait Implementations

impl EmbeddingProvider for OpenAIEmbedding

fn embed<'life0, 'life1, 'async_trait>( &'life0 self, text: &'life1 str, ) -> Pin<Box<dyn Future<Output = Result<Vec<f32>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Generate embedding for a single text

fn embed_batch<'life0, 'life1, 'life2, 'async_trait>( &'life0 self, texts: &'life1 [&'life2 str], ) -> Pin<Box<dyn Future<Output = Result<Vec<Vec<f32>>>> + Send + 'async_trait>>

where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait, 'life2: 'async_trait,

Generate embeddings for multiple texts

fn dimension(&self) -> usize

Get the embedding dimension