vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-01 18:20:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats 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 import typer
from skyvern.forge import app 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 INTERVAL = 1
LOG = structlog.get_logger() LOG = structlog.get_logger()
@ -13,7 +13,7 @@ LOG = structlog.get_logger()
async def run() -> None: async def run() -> None:
file_name = "skyvern_screenshot.png" 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: while True:
# run subprocess to take screenshot # 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"} 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: try:
await app.STORAGE.save_streaming_file("placeholder_org", file_name) await app.STORAGE.save_streaming_file("placeholder_org", file_name)
except Exception: except Exception: