Bases: ABC
BaseEmbeddingModel Abstract
Source code in llmfy/llmfy_core/embeddings/base_embedding_model.py
| class BaseEmbeddingModel(ABC):
"""BaseEmbeddingModel Abstract"""
def __init__(self):
"""Model provider."""
self.provider: ServiceProvider
self.model: str
@abstractmethod
def encode(
self,
text: str,
) -> List[float]:
pass
@abstractmethod
def encode_batch(
self,
texts: List[str] | str,
batch_size: int = 10,
max_retries: int = 3,
retry_delay: float = 1.0,
show_progress_bar: bool = False,
) -> Any:
pass
|
__init__()
Model provider.
Source code in llmfy/llmfy_core/embeddings/base_embedding_model.py
| def __init__(self):
"""Model provider."""
self.provider: ServiceProvider
self.model: str
|