Store user_id in vectordb (#2466)

This commit is contained in:
Dev Khant
2025-04-11 13:37:34 +05:30
committed by GitHub
parent 19d7beef43
commit 15a3e20371
6 changed files with 57 additions and 20 deletions

View File

@@ -3,6 +3,7 @@ import concurrent
import hashlib
import json
import logging
import os
import uuid
import warnings
from datetime import datetime
@@ -18,7 +19,7 @@ from mem0.configs.prompts import (
get_update_memory_messages,
)
from mem0.memory.base import MemoryBase
from mem0.memory.setup import setup_config
from mem0.memory.setup import mem0_dir, setup_config
from mem0.memory.storage import SQLiteManager
from mem0.memory.telemetry import capture_event
from mem0.memory.utils import (
@@ -62,6 +63,15 @@ class Memory(MemoryBase):
self.graph = MemoryGraph(self.config)
self.enable_graph = True
self.config.vector_store.config.collection_name = "mem0_migrations"
if self.config.vector_store.provider in ["faiss", "qdrant"]:
provider_path = f"migrations_{self.config.vector_store.provider}"
self.config.vector_store.config.path = os.path.join(mem0_dir, provider_path)
os.makedirs(self.config.vector_store.config.path, exist_ok=True)
self._telemetry_vector_store = VectorStoreFactory.create(
self.config.vector_store.provider, self.config.vector_store.config
)
capture_event("mem0.init", self)
@classmethod