54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
from abc import ABC, abstractmethod
|
|
|
|
|
|
class VectorStoreBase(ABC):
|
|
@abstractmethod
|
|
def create_col(self, name, vector_size, distance):
|
|
"""Create a new collection."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def insert(self, name, vectors, payloads=None, ids=None):
|
|
"""Insert vectors into a collection."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def search(self, name, query, limit=5, filters=None):
|
|
"""Search for similar vectors."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def delete(self, name, vector_id):
|
|
"""Delete a vector by ID."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def update(self, name, vector_id, vector=None, payload=None):
|
|
"""Update a vector and its payload."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get(self, name, vector_id):
|
|
"""Retrieve a vector by ID."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def list_cols(self):
|
|
"""List all collections."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def delete_col(self):
|
|
"""Delete a collection."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def col_info(self, name):
|
|
"""Get information about a collection."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def list(self, filters=None, limit=None):
|
|
"""List all memories."""
|
|
pass
|