From dd677132fe2380382b9543de96e0fd879cf177e8 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 18 Oct 2024 11:09:32 -0700 Subject: [PATCH] fix python style for webhook url and totp url when creating a task (#1003) --- 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 7ba7e02e..dcb5d63c 100644 --- a/skyvern/forge/agent.py +++ b/skyvern/forge/agent.py @@ -184,11 +184,13 @@ class ForgeAgent: 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,