haive.core.engine.embedding.providers

Embedding provider configurations.

This module contains all the embedding provider configurations for the Haive framework. Each provider is implemented as a separate class that extends BaseEmbeddingConfig.

Available Providers:

  • OpenAI: OpenAI embedding models (text-embedding-3-large, etc.)

  • Azure OpenAI: Azure-hosted OpenAI embedding models

  • HuggingFace: HuggingFace Hub and local transformer models

  • Cohere: Cohere embedding models (embed-english-v3.0, etc.)

  • Google Vertex AI: Google Cloud Vertex AI embedding models

  • Ollama: Locally hosted Ollama embedding models

  • Fake: Fake embeddings for testing

Examples

Basic usage:

from haive.core.engine.embedding.providers import OpenAIEmbeddingConfig

config = OpenAIEmbeddingConfig(
    name="my_embeddings",
    model="text-embedding-3-large",
    api_key="sk-..."
)

embeddings = config.instantiate()

Discovering providers:

from haive.core.engine.embedding.base import BaseEmbeddingConfig

# List all registered providers
providers = BaseEmbeddingConfig.list_registered_types()
print(f"Available providers: {list(providers.keys())}")

# Get a specific provider
provider_class = BaseEmbeddingConfig.get_config_class("OpenAI")

Submodules