From 35c4763bba29ac36004f5f794091968c001f74e5 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 16 Apr 2025 18:37:08 -0700 Subject: [PATCH] support totp verification url and totp identifier in run_task (#2175) --- skyvern/agent/agent.py | 2 ++ skyvern/forge/sdk/routes/agent_protocol.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/skyvern/agent/agent.py b/skyvern/agent/agent.py index a9fdf48b..63c897bc 100644 --- a/skyvern/agent/agent.py +++ b/skyvern/agent/agent.py @@ -297,6 +297,8 @@ class SkyvernAgent: extracted_information_schema=data_extraction_schema, error_code_mapping=error_code_mapping, proxy_location=proxy_location, + totp_identifier=totp_identifier, + totp_verification_url=totp_url, ) created_task = await app.agent.create_task(task_request, organization.organization_id) diff --git a/skyvern/forge/sdk/routes/agent_protocol.py b/skyvern/forge/sdk/routes/agent_protocol.py index 15bbf5ac..10e7046c 100644 --- a/skyvern/forge/sdk/routes/agent_protocol.py +++ b/skyvern/forge/sdk/routes/agent_protocol.py @@ -1495,6 +1495,8 @@ async def run_task( error_code_mapping=run_request.error_code_mapping, proxy_location=run_request.proxy_location, browser_session_id=run_request.browser_session_id, + totp_verification_url=run_request.totp_url, + totp_identifier=run_request.totp_identifier, ) task_v1_response = await task_v1_service.run_task( task=task_v1_request,