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

20
skyvern/agent/agent.py
View File

@@ -268,20 +268,18 @@ class SkyvernAgent:
if not self.client:
if engine == RunEngine.skyvern_v1:
data_extraction_goal = None
data_extraction_schema = data_extraction_schema
navigation_goal = prompt
navigation_payload = None
organization = await self._get_organization()
if not url:
task_generation = await task_v1_service.generate_task(
user_prompt=prompt,
organization=organization,
)
url = task_generation.url
navigation_goal = task_generation.navigation_goal or 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=prompt,
organization=organization,
)
url = url or task_generation.url
navigation_goal = task_generation.navigation_goal or 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_request = TaskRequest(
title=title,

1
skyvern/cli/commands.py
View File

@@ -183,6 +183,7 @@ def init(
api_key = asyncio.run(_setup_local_organization())
# Generate .env file
with open(".env", "w") as env_file:
env_file.write("LITELLM_LOG=ERROR\n")
env_file.write("ENABLE_OPENAI=true\n")
env_file.write(f"OPENAI_API_KEY={openai_api_key}\n")
env_file.write(f"LOG_LEVEL={log_level}\n")

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,