hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 11:56:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: update load_stream method to directly yield file chunks (#9806)

Browse Source
This commit is contained in:
zhuhao
2024-10-25 10:11:25 +08:00
committed by GitHub
parent dd17506078
commit 5bf31e7a86
11 changed files with 48 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/extensions/storage/local_fs_storage.py
View File

@@ -40,15 +40,11 @@ class LocalFsStorage(BaseStorage):
def load_stream(self, filename: str) -> Generator:
filepath = self._build_filepath(filename)
def generate() -> Generator:
if not os.path.exists(filepath):
raise FileNotFoundError("File not found")
with open(filepath, "rb") as f:
while chunk := f.read(4096): # Read in chunks of 4KB
yield chunk
return generate()
if not os.path.exists(filepath):
raise FileNotFoundError("File not found")
with open(filepath, "rb") as f:
while chunk := f.read(4096): # Read in chunks of 4KB
yield chunk
def download(self, filename, target_filepath):
filepath = self._build_filepath(filename)