Interface EmbeddingProviderTrait

core/src/embeddings.rs:170–200

/// Trait for embedding providers
#[async_trait]
pub trait EmbeddingProviderTrait: Send + Sync {
    /// Generate embeddings for the given request
    async fn generate_embeddings(
        &self,
        request: EmbeddingRequest,
    ) -> GraphBitResult<EmbeddingResponse>;

    /// Get the provider name
    fn provider_name(&self) -> &str;

    /// Get the model name
    fn model_name(&self) -> &str;

    /// Get embedding dimensions for this model
    async fn get_embedding_dimensions(&self) -> GraphBitResult<usize>;

    /// Check if the provider supports batch processing
    fn supports_batch(&self) -> bool {
        true
    }

    /// Get maximum batch size supported by the provider
    fn max_batch_size(&self) -> usize {
        100
    }

    /// Validate the configuration
    fn validate_config(&self) -> GraphBitResult<()> {
        Ok(())
    }
}

/// `OpenAI` embedding provider
#[derive(Debug, Clone)]