vibe.embedding_providers.berget

Berget AI embedding provider.

Uses Berget's OpenAI-compatible API for embeddings via intfloat/multilingual-e5-large.

BergetAPIError

Error from Berget API with clean message.

BergetEmbeddingProvider

Embedding provider using Berget AI API.

Uses intfloat/multilingual-e5-large-instruct (1024 dimensions) by default.

Configuration

api_key: Berget API key (or BERGET_API_KEY env var) base_url: API base URL (default: https://api.berget.ai/v1) model: Model to use (default: intfloat/multilingual-e5-large-instruct) timeout: Request timeout in seconds (default: 60) max_retries: Max retries on transient errors (default: 6)

embed

embed(text: str) -> list[float]

Generate embedding for a single text.

embed_batch

embed_batch(texts: list[str]) -> list[list[float]]

Generate embeddings for multiple texts.

Includes retry logic with exponential backoff for transient errors.

close

close() -> None

Close the HTTP client.