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

Signed-off-by: -LAN- <[email protected]>
langgenius · Dec 23, 2024 · c4091c4 · c4091c4
1 parent ef95b12
commit c4091c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/api/extensions/storage/aws_s3_storage.py b/api/extensions/storage/aws_s3_storage.py
@@ -67,7 +67,9 @@ def load_stream(self, filename: str) -> Generator:
             yield from response["Body"].iter_chunks()
         except ClientError as ex:
             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:
                 raise