diff --git a/llama-index-core/llama_index/core/storage/docstore/keyval_docstore.py b/llama-index-core/llama_index/core/storage/docstore/keyval_docstore.py index 8681fa7b65..68f0d88dea 100644 --- a/llama-index-core/llama_index/core/storage/docstore/keyval_docstore.py +++ b/llama-index-core/llama_index/core/storage/docstore/keyval_docstore.py @@ -393,7 +393,10 @@ def _remove_legacy_info(self, ref_doc_info_dict: dict) -> RefDocInfo: ref_doc_info_dict["metadata"] = ref_doc_info_dict.get("extra_info", {}) ref_doc_info_dict.pop("extra_info") - return RefDocInfo(**ref_doc_info_dict) + return RefDocInfo( + metadata=ref_doc_info_dict.get("metadata", {}), + node_ids=ref_doc_info_dict.get("node_ids", []) + ) def get_ref_doc_info(self, ref_doc_id: str) -> Optional[RefDocInfo]: """Get the RefDocInfo for a given ref_doc_id."""