graphragzen.text_embedding.embedding_models.BaseEmbedder

class graphragzen.text_embedding.embedding_models.BaseEmbedder[source]

Attributes

vector_size

Methods

embed(text[, task, show_progress_bar])

Text embed strings

abstract embed(text, task='embed_document', show_progress_bar=False)[source]

Text embed strings

Parameters:
  • text (Union[str, List[str]]) – String(s) to embed

  • task (str, optional) – Some embedding models create different vectors for different tasks.

  • show_progress_bar (bool, optional) – If True shows a progress bar. Defaults to False.

Returns:

np.array

Return type:

ndarray

vector_size: float = 0