From 879ea477c182f51dbf26cf718e5ee75da25800bd Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 18 Oct 2024 11:06:17 -0700 Subject: [PATCH] fix python style for webhook url and totp url when creating a task --- skyvern/forge/agent.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/agent.py b/skyvern/forge/agent.py index 7ba7e02e0..dcb5d63cc 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -184,11 +184,13 @@ async def create_task_and_step_from_block( return task, step async def create_task(self, task_request: TaskRequest, organization_id: str | None = None) -> Task: + webhook_callback_url = str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None + totp_verification_url = str(task_request.totp_verification_url) if task_request.totp_verification_url else None task = await app.DATABASE.create_task( url=str(task_request.url), title=task_request.title, - webhook_callback_url=str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None, - totp_verification_url=str(task_request.totp_verification_url) if task_request.totp_verification_url else None, + webhook_callback_url=webhook_callback_url, + totp_verification_url=totp_verification_url, totp_identifier=task_request.totp_identifier, navigation_goal=task_request.navigation_goal, data_extraction_goal=task_request.data_extraction_goal,