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

fix SkyvernClient (#2055)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-31 22:35:16 -04:00
committed by GitHub
parent ead1e4ba59
commit 59d2eb9df7
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
skyvern/agent/client.py
View File

@@ -29,7 +29,7 @@ class SkyvernClient:
publish_workflow: bool = False, publish_workflow: bool = False,
) -> TaskRunResponse: ) -> TaskRunResponse:
task_run_obj = await self.client.agent.run_task( task_run_obj = await self.client.agent.run_task(
goal=prompt, prompt=prompt,
url=url, url=url,
title=title, title=title,
engine=engine, engine=engine,
@@ -42,7 +42,7 @@ class SkyvernClient:
browser_session_id=browser_session_id, browser_session_id=browser_session_id,
publish_workflow=publish_workflow, publish_workflow=publish_workflow,
) )
return TaskRunResponse.model_validate(task_run_obj) return TaskRunResponse.model_validate(task_run_obj.dict())
async def run_workflow( async def run_workflow(
self, self,
@@ -65,7 +65,7 @@ class SkyvernClient:
browser_session_id=browser_session_id, browser_session_id=browser_session_id,
template=template, template=template,
) )
return WorkflowRunResponse.model_validate(workflow_run_obj) return WorkflowRunResponse.model_validate(workflow_run_obj.dict())
async def get_run( async def get_run(
self, self,
@@ -73,7 +73,7 @@ class SkyvernClient:
) -> RunResponse: ) -> RunResponse:
run_obj = await self.client.agent.get_run(run_id=run_id) run_obj = await self.client.agent.get_run(run_id=run_id)
if run_obj.run_type in [RunType.task_v1, RunType.task_v2]: if run_obj.run_type in [RunType.task_v1, RunType.task_v2]:
return TaskRunResponse.model_validate(run_obj) return TaskRunResponse.model_validate(run_obj.dict())
elif run_obj.run_type == RunType.workflow_run: elif run_obj.run_type == RunType.workflow_run:
return WorkflowRunResponse.model_validate(run_obj) return WorkflowRunResponse.model_validate(run_obj.dict())
raise ValueError(f"Invalid run type: {run_obj.run_type}") raise ValueError(f"Invalid run type: {run_obj.run_type}")

4
skyvern/forge/sdk/routes/agent_protocol.py
View File

@@ -1475,7 +1475,7 @@ async def run_task(
try: try:
task_v2 = await task_v2_service.initialize_task_v2( task_v2 = await task_v2_service.initialize_task_v2(
organization=current_org, organization=current_org,
user_prompt=run_request.goal, user_prompt=run_request.prompt,
user_url=run_request.url, user_url=run_request.url,
totp_identifier=run_request.totp_identifier, totp_identifier=run_request.totp_identifier,
totp_verification_url=run_request.totp_url, totp_verification_url=run_request.totp_url,
@@ -1504,7 +1504,7 @@ async def run_task(
run_type=RunType.task_v2, run_type=RunType.task_v2,
status=str(task_v2.status), status=str(task_v2.status),
output=task_v2.output, output=task_v2.output,
failure_reason=task_v2.failure_reason, failure_reason=None,
created_at=task_v2.created_at, created_at=task_v2.created_at,
modified_at=task_v2.modified_at, modified_at=task_v2.modified_at,
run_request=TaskRunRequest( run_request=TaskRunRequest(