pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-23 06:46:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Python SDK retrieve document_name is empty (#12062)

Browse Source 
### What problem does this PR solve?

https://github.com/infiniflow/ragflow/issues/12056

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Stephen Hu
2025-12-22 11:08:39 +08:00
committed by GitHub
parent b49eb6826b
commit 2118bc2556
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
sdk/python/ragflow_sdk/modules/chunk.py
View File

@ -33,6 +33,7 @@ class Chunk(Base):
self.create_timestamp = 0.0 self.create_timestamp = 0.0
self.dataset_id = None self.dataset_id = None
self.document_name = "" self.document_name = ""
self.documnet_keyword = ""
self.document_id = "" self.document_id = ""
self.available = True self.available = True
# Additional fields for retrieval results # Additional fields for retrieval results
@ -46,6 +47,11 @@ class Chunk(Base):
res_dict.pop(k) res_dict.pop(k)
super().__init__(rag, res_dict) super().__init__(rag, res_dict)
#for backward compatibility
if not self.document_name:
self.document_name = self.documnet_keyword
def update(self, update_message: dict): def update(self, update_message: dict):
res = self.put(f"/datasets/{self.dataset_id}/documents/{self.document_id}/chunks/{self.id}", update_message) res = self.put(f"/datasets/{self.dataset_id}/documents/{self.document_id}/chunks/{self.id}", update_message)
res = res.json() res = res.json()