test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-02 19:27:48 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
yanghx 2024-07-25 11:20:20 +08:00 committed by GitHub
parent a876baf0a9
commit 349ec0db77
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/extensions/storage/tencent_storage.py
View File

@ -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()