haive.core.engine.embedding.providers.CohereEmbeddingConfig

Cohere embedding configuration.

Classes

CohereEmbeddingConfig

Configuration for Cohere embeddings.

Module Contents

class haive.core.engine.embedding.providers.CohereEmbeddingConfig.CohereEmbeddingConfig

Bases: haive.core.engine.embedding.base.BaseEmbeddingConfig

Configuration for Cohere embeddings.

This configuration provides access to Cohere’s embedding models including embed-english-v3.0, embed-multilingual-v3.0, and other specialized models.

Examples

Basic usage:

config = CohereEmbeddingConfig(
    name="cohere_embeddings",
    model="embed-english-v3.0",
    api_key="your-api-key"
)

embeddings = config.instantiate()

With custom input type:

config = CohereEmbeddingConfig(
    name="cohere_embeddings",
    model="embed-english-v3.0",
    input_type="search_document",
    api_key="your-api-key"
)

Using environment variables:

# Set COHERE_API_KEY environment variable
config = CohereEmbeddingConfig(
    name="cohere_embeddings",
    model="embed-english-v3.0"
)

embedding_type

Always EmbeddingType.COHERE

model

Cohere model name (e.g., “embed-english-v3.0”)

api_key

Cohere API key (auto-resolved from COHERE_API_KEY env var)

input_type

Input type for embeddings (search_document, search_query, etc.)

truncate

How to handle inputs longer than max length

get_default_model()

Get the default model for Cohere embeddings.

Return type:

str

get_input_types()

Get list of supported input types.

Return type:

list[str]

get_model_info()

Get information about the configured model.

Return type:

dict

get_supported_models()

Get list of supported Cohere embedding models.

Return type:

list[str]

get_truncate_options()

Get list of supported truncate options.

Return type:

list[str]

instantiate()

Create a Cohere embeddings instance.

Returns:

CohereEmbeddings instance configured with the provided parameters

Raises:

• ImportError – If langchain-cohere is not installed

• ValueError – If configuration is invalid

Return type:

Any

classmethod validate_input_type(v)

Validate input type.

Return type:

Any

classmethod validate_model(v)

Validate the Cohere model name.

Return type:

Any

classmethod validate_truncate(v)

Validate truncate parameter.

Return type:

Any