Ling0925
/
suna
mirror of https://github.com/kortix-ai/suna.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor(sb_image_edit_tool): convert file operations to async for improved performance

This commit is contained in:
sharath 2025-07-05 11:00:00 +00:00
parent d6f7e5981a
commit 9cfaac080c
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/agent/tools/sb_image_edit_tool.py
View File

@ -139,13 +139,13 @@ class SandboxImageEditTool(SandboxToolsBase):
full_path = f"{self.workspace_path}/{cleaned_path}"
# Check if file exists and is not a directory
file_info = self.sandbox.fs.get_file_info(full_path)
file_info = await self.sandbox.fs.get_file_info(full_path)
if file_info.is_dir:
return self.fail_response(
f"Path '{cleaned_path}' is a directory, not an image file."
)
return self.sandbox.fs.download_file(full_path)
return await self.sandbox.fs.download_file(full_path)
except Exception as e:
return self.fail_response(
@ -167,7 +167,7 @@ class SandboxImageEditTool(SandboxToolsBase):
sandbox_path = f"{self.workspace_path}/{random_filename}"
# Save image to sandbox
self.sandbox.fs.upload_file(sandbox_path, img_response.content)
await self.sandbox.fs.upload_file(sandbox_path, img_response.content)
return random_filename
except Exception as e: