NavyDevilDoc commited on
Commit
e4746b7
·
verified ·
1 Parent(s): 0a3adb1

Update src/rag_engine.py

Browse files
Files changed (1) hide show
  1. src/rag_engine.py +10 -3
src/rag_engine.py CHANGED
@@ -339,10 +339,17 @@ def rebuild_cache_from_pinecone(username: str, index_name: str) -> Tuple[bool, s
339
  batch_ids = ids[i : i + batch_size]
340
  response = pm.fetch_vectors(index_name, batch_ids, username)
341
 
342
- for vec_id, vec_data in response.get('vectors', {}).items():
343
- meta = vec_data.get('metadata', {})
 
 
 
 
 
344
  source = meta.get('source', 'unknown_restored.txt')
345
- text = meta.get('text', '') or vec_data.get('metadata', {}).get('page_content', '')
 
 
346
 
347
  if source not in reconstructed_files:
348
  reconstructed_files[source] = []
 
339
  batch_ids = ids[i : i + batch_size]
340
  response = pm.fetch_vectors(index_name, batch_ids, username)
341
 
342
+ vectors = response.vectors
343
+
344
+ for vec_id, vec_data in vectors.items():
345
+ # vec_data is also an object. Access .metadata via attribute.
346
+ meta = vec_data.metadata
347
+ if meta is None: meta = {}
348
+
349
  source = meta.get('source', 'unknown_restored.txt')
350
+
351
+ # Retrieve text (handle potential key variations)
352
+ text = meta.get('text') or meta.get('page_content') or ''
353
 
354
  if source not in reconstructed_files:
355
  reconstructed_files[source] = []