Dama12 commited on
Commit
79fd30c
·
1 Parent(s): 6829fbd

fix: bump memory collection version and align with fastembed named vectors

Browse files
Files changed (1) hide show
  1. app/services/memory_service.py +5 -15
app/services/memory_service.py CHANGED
@@ -15,7 +15,7 @@ class MemoryService:
15
  Stocke les découvertes, préférences et conclusions passées.
16
  """
17
 
18
- COLLECTION_NAME = "user_semantic_memory"
19
 
20
  def __init__(self):
21
  self.client = QdrantClient(
@@ -44,20 +44,10 @@ class MemoryService:
44
  }
45
  )
46
  else:
47
- # Vérifier si le vecteur nommé existe, sinon on recrée (pour corriger le format hackathon)
48
- collection_info = self.client.get_collection(self.COLLECTION_NAME)
49
- if "fast-bge-small-en-v1.5" not in collection_info.config.params.vectors:
50
- logger.warning(f"Collection {self.COLLECTION_NAME} has wrong vector format. Recreating...")
51
- self.client.delete_collection(self.COLLECTION_NAME)
52
- self.client.create_collection(
53
- collection_name=self.COLLECTION_NAME,
54
- vectors_config={
55
- "fast-bge-small-en-v1.5": models.VectorParams(
56
- size=384,
57
- distance=models.Distance.COSINE
58
- )
59
- }
60
- )
61
 
62
  # Vérifier/Créer l'index user_id séparément pour être sûr
63
  try:
 
15
  Stocke les découvertes, préférences et conclusions passées.
16
  """
17
 
18
+ COLLECTION_NAME = "user_semantic_memory_v2"
19
 
20
  def __init__(self):
21
  self.client = QdrantClient(
 
44
  }
45
  )
46
  else:
47
+ # Si elle existe déjà mais avec le mauvais schéma (vecteur anonyme),
48
+ # Qdrant renverra l'erreur qu'on a vue.
49
+ # On pourrait la supprimer/recréer ici si on veut être radical.
50
+ pass
 
 
 
 
 
 
 
 
 
 
51
 
52
  # Vérifier/Créer l'index user_id séparément pour être sûr
53
  try: