vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-10 15:35:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

better detect dropdown menu (#778)

Browse source
This commit is contained in:
LawyZheng 2024-09-07 09:34:33 +08:00 • committed by GitHub
parent 95b2e53c46
commit 692ffb6d43
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 144 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

8
streamlit_app/visualizer/api.py
View file

@ -16,18 +16,20 @@ class SkyvernClient:
self.base_url = base_url
self.credentials = credentials
def generate_curl_params(self, task_request_body: TaskRequest) -> PreparedRequest:
def generate_curl_params(self, task_request_body: TaskRequest, max_steps: int | None = None) -> PreparedRequest:
url = f"{self.base_url}/tasks"
payload = task_request_body.model_dump()
headers = {
"Content-Type": "application/json",
"x-api-key": self.credentials,
}
if max_steps is not None:
headers["x-max-steps-override"] = str(max_steps)
return url, payload, headers
def create_task(self, task_request_body: TaskRequest) -> str | None:
url, payload, headers = self.generate_curl_params(task_request_body)
def create_task(self, task_request_body: TaskRequest, max_steps: int | None = None) -> str | None:
url, payload, headers = self.generate_curl_params(task_request_body, max_steps=max_steps)
response = requests.post(url, headers=headers, data=json.dumps(payload))
if "task_id" not in response.json():