gemini_client

module wittgenstein_embeddings.providers.gemini_client

Classes

• GeminiEmbedder — Gemini embeddings via google-genai, using its native query/document task types — RETRIEVAL_QUERY / RETRIEVAL_DOCUMENT — instead of a text prefix convention.

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder

class GeminiEmbedder(api_key: str, model: str = _DEFAULT_MODEL, output_dimensionality: int | None = None)

Bases : EmbeddingClient

Gemini embeddings via google-genai, using its native query/document task types — RETRIEVAL_QUERY / RETRIEVAL_DOCUMENT — instead of a text prefix convention.

Methods

• query

• passage

• query_batch

• passage_batch

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder.query

method GeminiEmbedder.query(text: str) → list[float]

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder.passage

method GeminiEmbedder.passage(text: str) → list[float]

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder.query_batch

method GeminiEmbedder.query_batch(texts: list[str]) → list[list[float]]

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder.passage_batch

method GeminiEmbedder.passage_batch(texts: list[str]) → list[list[float]]

wittgenstein_embeddings.providers.gemini_client.GeminiEmbedder.dimension

property GeminiEmbedder.dimension: int