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

fix task v1 generation in run_task api (#2074)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-04-02 13:49:06 -04:00
committed by GitHub
parent c7e62f45d3
commit a45e630b35
3 changed files with 19 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1470,16 +1470,15 @@ async def run_task(
data_extraction_schema = run_request.data_extraction_schema
navigation_goal = run_request.prompt
navigation_payload = None
if not url:
task_generation = await task_v1_service.generate_task(
user_prompt=run_request.prompt,
organization=current_org,
)
url = task_generation.url
navigation_goal = task_generation.navigation_goal or run_request.prompt
navigation_payload = task_generation.navigation_payload
data_extraction_goal = task_generation.data_extraction_goal
data_extraction_schema = data_extraction_schema or task_generation.extracted_information_schema
task_generation = await task_v1_service.generate_task(
user_prompt=run_request.prompt,
organization=current_org,
)
url = url or task_generation.url
navigation_goal = task_generation.navigation_goal or run_request.prompt
navigation_payload = task_generation.navigation_payload
data_extraction_goal = task_generation.data_extraction_goal
data_extraction_schema = data_extraction_schema or task_generation.extracted_information_schema
task_v1_request = TaskRequest(
title=run_request.title,