fix: hotfix for LI version change

This commit is contained in:
trducng
2024-11-11 02:50:41 +00:00
parent f1330ced26
commit f701a5eefd

View File

@@ -100,7 +100,9 @@ class KnetRetrievalPipeline(BaseFileIndexRetriever):
# Load YAML content from the response content
chunks = yaml.safe_load(response.content)
for chunk in chunks:
metadata = chunk["node"]["metadata"]
metadata = chunk["node"].get("metadata")
if not metadata:
metadata = chunk["node"].get("extra_info")
content_type = metadata.pop("content_type", "")
metadata["page_label"] = metadata.get(