Skip to content

Commit adc24de

Browse files
safely initialize RefDocInfo in Docstore (#20031)
1 parent 3c6590f commit adc24de

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

llama-index-core/llama_index/core/storage/docstore/keyval_docstore.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,10 @@ def _remove_legacy_info(self, ref_doc_info_dict: dict) -> RefDocInfo:
393393
ref_doc_info_dict["metadata"] = ref_doc_info_dict.get("extra_info", {})
394394
ref_doc_info_dict.pop("extra_info")
395395

396-
return RefDocInfo(**ref_doc_info_dict)
396+
return RefDocInfo(
397+
metadata=ref_doc_info_dict.get("metadata", {}),
398+
node_ids=ref_doc_info_dict.get("node_ids", [])
399+
)
397400

398401
def get_ref_doc_info(self, ref_doc_id: str) -> Optional[RefDocInfo]:
399402
"""Get the RefDocInfo for a given ref_doc_id."""

0 commit comments

Comments
 (0)