7aad8ebb3c
* Allow passing the collection name in each request to override the default * Allow getting the collection names in QdrantConnector * get vector size from model description * ruff format * add isort * apply pre-commit hooks --------- Co-authored-by: generall <andrey@vasnetsov.com>
27 lines
732 B
Python
27 lines
732 B
Python
from abc import ABC, abstractmethod
|
|
from typing import List
|
|
|
|
|
|
class EmbeddingProvider(ABC):
|
|
"""Abstract base class for embedding providers."""
|
|
|
|
@abstractmethod
|
|
async def embed_documents(self, documents: List[str]) -> List[List[float]]:
|
|
"""Embed a list of documents into vectors."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def embed_query(self, query: str) -> List[float]:
|
|
"""Embed a query into a vector."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_vector_name(self) -> str:
|
|
"""Get the name of the vector for the Qdrant collection."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_vector_size(self) -> int:
|
|
"""Get the size of the vector for the Qdrant collection."""
|
|
pass
|