fix VectorStoreBase abstract methods params (#2068)

Co-authored-by: Dev Khant <devkhant24@gmail.com>
This commit is contained in:
AkisAya
2025-01-09 20:00:16 +08:00
committed by GitHub
parent 2e782b0963
commit cbd845fe41

View File

@@ -8,27 +8,27 @@ class VectorStoreBase(ABC):
pass pass
@abstractmethod @abstractmethod
def insert(self, name, vectors, payloads=None, ids=None): def insert(self, vectors, payloads=None, ids=None):
"""Insert vectors into a collection.""" """Insert vectors into a collection."""
pass pass
@abstractmethod @abstractmethod
def search(self, name, query, limit=5, filters=None): def search(self, query, limit=5, filters=None):
"""Search for similar vectors.""" """Search for similar vectors."""
pass pass
@abstractmethod @abstractmethod
def delete(self, name, vector_id): def delete(self, vector_id):
"""Delete a vector by ID.""" """Delete a vector by ID."""
pass pass
@abstractmethod @abstractmethod
def update(self, name, vector_id, vector=None, payload=None): def update(self, vector_id, vector=None, payload=None):
"""Update a vector and its payload.""" """Update a vector and its payload."""
pass pass
@abstractmethod @abstractmethod
def get(self, name, vector_id): def get(self, vector_id):
"""Retrieve a vector by ID.""" """Retrieve a vector by ID."""
pass pass
@@ -43,7 +43,7 @@ class VectorStoreBase(ABC):
pass pass
@abstractmethod @abstractmethod
def col_info(self, name): def col_info(self):
"""Get information about a collection.""" """Get information about a collection."""
pass pass