diff --git a/skyvern/agent/agent.py b/skyvern/agent/agent.py index 3e88aa2a..4fff3956 100644 --- a/skyvern/agent/agent.py +++ b/skyvern/agent/agent.py @@ -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, diff --git a/skyvern/cli/commands.py b/skyvern/cli/commands.py index 8b7f85e0..ca4b0f6b 100644 --- a/skyvern/cli/commands.py +++ b/skyvern/cli/commands.py @@ -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") diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index c3aecc94..275f4560 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -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,