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

Implement get_latest_screenshots, add action_screenshots to TaskResponse (#148)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-04-02 14:43:29 -07:00
committed by GitHub
parent 5c0f482b05
commit 69c458bd7c
4 changed files with 65 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
skyvern/forge/sdk/schemas/tasks.py
View File

@@ -166,7 +166,11 @@ class Task(TaskRequest):
raise ValueError(f"cant_override_failure_reason({self.task_id})")
def to_task_response(
self, screenshot_url: str | None = None, recording_url: str | None = None, failure_reason: str | None = None
self,
action_screenshot_urls: list[str] | None = None,
screenshot_url: str | None = None,
recording_url: str | None = None,
failure_reason: str | None = None,
) -> TaskResponse:
return TaskResponse(
request=self,
@@ -176,6 +180,7 @@ class Task(TaskRequest):
modified_at=self.modified_at,
extracted_information=self.extracted_information,
failure_reason=failure_reason or self.failure_reason,
action_screenshot_urls=action_screenshot_urls,
screenshot_url=screenshot_url,
recording_url=recording_url,
errors=self.errors,
@@ -189,6 +194,7 @@ class TaskResponse(BaseModel):
created_at: datetime
modified_at: datetime
extracted_information: list | dict[str, Any] | str | None = None
action_screenshot_urls: list[str] | None = None
screenshot_url: str | None = None
recording_url: str | None = None
failure_reason: str | None = None