fix tencent_cos_storage image-preview error is not a byte (#6652)

2024-07-25 11:20:20 +08:00
parent a876baf0a9
commit 349ec0db77
1 changed files with 1 additions and 2 deletions
--- a/api/extensions/storage/tencent_storage.py
+++ b/api/extensions/storage/tencent_storage.py
@@ -32,8 +32,7 @@ class TencentStorage(BaseStorage):
    def load_stream(self, filename: str) -> Generator:
        def generate(filename: str = filename) -> Generator:
            response = self.client.get_object(Bucket=self.bucket_name, Key=filename)
-            while chunk := response['Body'].get_stream(chunk_size=4096):
-                yield chunk
+            yield from response['Body'].get_stream(chunk_size=4096)

        return generate()