Add ollama embeddings (#1634)

2024-08-03 10:55:40 +05:30
parent 1c46fddce1
commit 024089d33e
7 changed files with 57 additions and 16 deletions
--- a/mem0/vector_stores/configs.py
+++ b/mem0/vector_stores/configs.py
@@ -73,8 +73,12 @@ class VectorStoreConfig(BaseModel):
        
        if isinstance(v, dict):
            if provider == "qdrant":
+                if "path" not in v:
+                    v["path"] = "/tmp/qdrant"
                return QdrantConfig(**v)
            elif provider == "chromadb":
+                if "path" not in v:
+                    v["path"] = "/tmp/chromadb"
                return ChromaDbConfig(**v)
        
        return v