Mirrors/kotaemon
1
0
Fork 0
mirror of https://github.com/Cinnamon/kotaemon.git synced 2026-02-23 19:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: page number in knet

Browse Source
This commit is contained in:
trducng
2024-10-02 06:16:20 +00:00
parent 6ac627cc29
commit 485c6a5510
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
libs/ktem/ktem/index/file/knet/pipelines.py
View File

@@ -91,6 +91,9 @@ class KnetRetrievalPipeline(BaseFileIndexRetriever):
chunks = yaml.safe_load(response.content)
for chunk in chunks:
metadata = chunk["node"]["metadata"]
metadata["page_label"] = metadata.get(
"pageIdx", metadata.get("parentPageIdx", "")
)
metadata["type"] = metadata_translation.get(
metadata.pop("content_type", ""), ""
)