fix bug (#1387)

Signed-off-by: XuhuiRen <[email protected]>
intel · Mar 18, 2024 · 8797cfe · 8797cfe
1 parent 4db5cb6
commit 8797cfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/intel_extension_for_transformers/langchain/retrievers/child_parent_retriever.py b/intel_extension_for_transformers/langchain/retrievers/child_parent_retriever.py
@@ -67,7 +67,7 @@ def _get_relevant_documents(self, query: str, *, run_manager: CallbackManagerFor
                 ids.append(d.metadata['identify_id'])
 
         retrieved_documents = self.parentstore.get(ids)
-        for i in range(len(retrieved_documents)):
+        for i in range(len(retrieved_documents['ids'])):
             metadata = retrieved_documents['metadatas'][i]
             context = retrieved_documents['documents'][i]
             instance = Document(page_content=context, metadata=metadata)