AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 04:28:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve error handling for file retrieval in AwsS3Storage (#12002)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2024-12-23 16:28:54 +08:00 committed by GitHub
parent ef95b1268e
commit c4091c4c66
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/extensions/storage/aws_s3_storage.py
View File

@ -67,7 +67,9 @@ class AwsS3Storage(BaseStorage):
yield from response["Body"].iter_chunks() yield from response["Body"].iter_chunks()
except ClientError as ex: except ClientError as ex:
if ex.response["Error"]["Code"] == "NoSuchKey": if ex.response["Error"]["Code"] == "NoSuchKey":
raise FileNotFoundError("File not found") raise FileNotFoundError("file not found")
elif "reached max retries" in str(ex):
raise ValueError("please do not request the same file too frequently")
else: else:
raise raise