Update retriever.py

retriever.py

@@ -32,8 +32,17 @@ def retrieve_documents(query: str, top_k=3):
     """Retrieve most relevant documents for a given query."""
     query_embedding = embed_model.encode(query).tolist()
     results = collection.query(query_embeddings=[query_embedding], n_results=top_k)
-    
-    return " ".join(results["documents"]) if results["documents"] else "No relevant documents found."
+
+    retrieved_texts = []
+    for doc in results["documents"]:
+        if isinstance(doc, list):
+            retrieved_texts.extend(doc)  # Unnest lists
+        else:
+            retrieved_texts.append(doc)
+
+    return " ".join(retrieved_texts) if retrieved_texts else "No relevant documents found."
+
+#    return " ".join(results["documents"]) if results["documents"] else "No relevant documents found."
 
 # Add some example data (You can replace this with actual text files)
 if collection.count() == 0: