Sync cloud streamlit to open-source streamlit (#54)

This commit is contained in:
Kerem Yilmaz 2024-03-12 22:03:32 -07:00 committed by GitHub
parent 59a4a528e0
commit 647ea2ac0f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3,10 +3,13 @@ from typing import Any
import curlify
import requests
import structlog
from requests import PreparedRequest
from skyvern.forge.sdk.schemas.tasks import TaskRequest
LOG = structlog.get_logger()
class SkyvernClient:
def __init__(self, base_url: str, credentials: str):
@ -28,6 +31,7 @@ class SkyvernClient:
response = requests.post(url, headers=headers, data=json.dumps(payload))
if "task_id" not in response.json():
LOG.info(f"Failed to create task: {response.text}")
return None
return response.json()["task_id"]