Rename embedchain to mem0 and open sourcing code for long term memory (#1474)
Co-authored-by: Deshraj Yadav <deshrajdry@gmail.com>
This commit is contained in:
48
mem0/vector_stores/base.py
Normal file
48
mem0/vector_stores/base.py
Normal file
@@ -0,0 +1,48 @@
|
||||
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, name):
|
||||
"""Delete a collection."""
|
||||
pass
|
||||
|
||||
@abstractmethod
|
||||
def col_info(self, name):
|
||||
"""Get information about a collection."""
|
||||
pass
|
||||
Reference in New Issue
Block a user