hoops_ai.ml.embeddings.TextEmbeddingsModel

class hoops_ai.ml.embeddings.TextEmbeddingsModel

Bases: ABC

INTERFACE Text embeddings for semantic search and similarity matching.

abstract embed_text(text)

Compute an embedding for a single text string.

Parameters:

text (str)

Return type:

Embedding

abstract embed_text_batch(texts)

Compute embeddings for multiple text strings.

Parameters:

texts (Sequence[str])

Return type:

EmbeddingBatch

abstract property embedding_dim: int

Return the dimensionality of embeddings produced by this model.

Returns:

Embedding dimension (e.g., 384 for all-MiniLM-L6-v2, 1536 for text-embedding-3-small)

Return type:

int

abstract property model_id: str

all-MiniLM-L6-v2’ or ‘openai:text-embedding-3-small’

Type:

Example

Type:

‘hf

abstract property model_name: str