Update src/get_medical_system.py

src/get_medical_system.py

@@ -42,6 +42,7 @@ def get_vector_store():
         
         # Create proper Document objects and InMemoryDocstore
         documents_dict = {}
+        documents = []
         for i, text in enumerate(texts):
             # Create Document objects with proper metadata
             doc = Document(
@@ -49,6 +50,7 @@ def get_vector_store():
                 metadata={"doc_id": i, "type": "medical_qa"}
             )
             documents_dict[str(i)] = doc
+            documents.append(doc)
         
         # Create proper docstore
         docstore = InMemoryDocstore(documents_dict)
@@ -64,7 +66,7 @@ def get_vector_store():
             index_to_docstore_id=index_to_docstore_id
         )
         
-        return vectorstore, embeddings_array
+        return vectorstore, documents
         
     except FileNotFoundError as e:
         print(f"❌ Pre-computed files not found: {e}")