marvin.memory.providers.chroma

Classes

ChromaMemory

class ChromaMemory(client: chromadb.ClientAPI = lambda: chromadb.PersistentClient(path=str(marvin.settings.home_path / 'memory/chroma'))(), collection_name: str = 'memory-{key}')

Methods:

  • add
    def add(self, memory_key: str, content: str) -> str
    
  • delete
    def delete(self, memory_key: str, memory_id: str)
    
  • get_collection
    def get_collection(self, memory_key: str) -> chromadb.Collection
    
  • search
    def search(self, memory_key: str, query: str, n: int = 20) -> dict[str, str]
    

Functions

ChromaCloudMemory

def ChromaCloudMemory(tenant: str | None = None, database: str | None = None, api_key: str | None = None, **kwargs: Any = {}) -> ChromaMemory

ChromaEphemeralMemory

def ChromaEphemeralMemory(**kwargs: Any = {}) -> ChromaMemory

ChromaPersistentMemory

def ChromaPersistentMemory(path: str | None = None, **kwargs: Any = {}) -> ChromaMemory

Parent Module: providers