Просмотр исходного кода

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

yanghx 9 месяцев назад
Родитель
Сommit
349ec0db77
1 измененных файлов с 1 добавлено и 2 удалено
  1. 1 2
      api/extensions/storage/tencent_storage.py

+ 1 - 2
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()