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

fix python style for webhook url and totp url when creating a task (#1003)

Browse Source
This commit is contained in:
Shuchang Zheng
2024-10-18 11:09:32 -07:00
committed by GitHub
parent c65795711a
commit dd677132fe
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
skyvern/forge/agent.py
View File

@@ -184,11 +184,13 @@ class ForgeAgent:
return task, step return task, step
async def create_task(self, task_request: TaskRequest, organization_id: str | None = None) -> Task: 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( task = await app.DATABASE.create_task(
url=str(task_request.url), url=str(task_request.url),
title=task_request.title, title=task_request.title,
webhook_callback_url=str(task_request.webhook_callback_url) if task_request.webhook_callback_url else None, webhook_callback_url=webhook_callback_url,
totp_verification_url=str(task_request.totp_verification_url) if task_request.totp_verification_url else None, totp_verification_url=totp_verification_url,
totp_identifier=task_request.totp_identifier, totp_identifier=task_request.totp_identifier,
navigation_goal=task_request.navigation_goal, navigation_goal=task_request.navigation_goal,
data_extraction_goal=task_request.data_extraction_goal, data_extraction_goal=task_request.data_extraction_goal,