vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix streaming bug (#1370)

Browse Source
This commit is contained in:
LawyZheng
2024-12-11 14:04:48 +08:00
committed by GitHub
parent 9ac37350ef
commit 3a626503b7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
run_streaming.py
View File

@@ -5,7 +5,7 @@ import structlog
import typer
from skyvern.forge import app
from skyvern.forge.sdk.settings_manager import SettingsManager
from skyvern.forge.sdk.api.files import get_skyvern_temp_dir
INTERVAL = 1
LOG = structlog.get_logger()
@@ -13,7 +13,7 @@ LOG = structlog.get_logger()
async def run() -> None:
file_name = "skyvern_screenshot.png"
png_file_path = f"{SettingsManager.get_settings().STREAMING_FILE_BASE_PATH}/{file_name}"
png_file_path = f"{get_skyvern_temp_dir()}/{file_name}"
while True:
# run subprocess to take screenshot
@@ -21,7 +21,7 @@ async def run() -> None:
f"xwd -root | xwdtopnm 2>/dev/null | pnmtopng > {png_file_path}", shell=True, env={"DISPLAY": ":99"}
)
# upload screenshot to S3
# FIXME: upload screenshot to S3 with correct organization id
try:
await app.STORAGE.save_streaming_file("placeholder_org", file_name)
except Exception: