Spaces:
Sleeping
Sleeping
Update src/rag_engine.py
Browse files- 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 |
-
|
| 343 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 344 |
source = meta.get('source', 'unknown_restored.txt')
|
| 345 |
-
|
|
|
|
|
|
|
| 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] = []
|