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

Fix: Resolve JSON download errors in Document.download() (#8084)

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

An exception is thrown only when the json file has only two keys, `code`
and `message`. In other cases, response.content is returned normally.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Liu An
2025-06-05 18:03:51 +08:00
committed by GitHub
parent 841291dda0
commit f007c1c772
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
sdk/python/ragflow_sdk/modules/document.py
View File

@ -63,9 +63,14 @@ class Document(Base):
def download(self): def download(self):
res = self.get(f"/datasets/{self.dataset_id}/documents/{self.id}") res = self.get(f"/datasets/{self.dataset_id}/documents/{self.id}")
error_keys = set(["code", "message"])
try: try:
res = res.json() response = res.json()
raise Exception(res.get("message")) actual_keys = set(response.keys())
if actual_keys == error_keys:
raise Exception(res.get("message"))
else:
return res.content
except json.JSONDecodeError: except json.JSONDecodeError:
return res.content return res.content