email relay (#598)

Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>

skyvern/forge/sdk/workflow/models/workflow.py

@@ -14,6 +14,7 @@ class WorkflowRequestBody(BaseModel):
     data: dict[str, Any] | None = None
     proxy_location: ProxyLocation | None = None
     webhook_callback_url: str | None = None
+    totp_verification_url: str | None = None
 
 
 class RunWorkflowResponse(BaseModel):
@@ -49,6 +50,7 @@ class Workflow(BaseModel):
     workflow_definition: WorkflowDefinition
     proxy_location: ProxyLocation | None = None
     webhook_callback_url: str | None = None
+    totp_verification_url: str | None = None
 
     created_at: datetime
     modified_at: datetime
@@ -71,6 +73,7 @@ class WorkflowRun(BaseModel):
     status: WorkflowRunStatus
     proxy_location: ProxyLocation | None = None
     webhook_callback_url: str | None = None
+    totp_verification_url: str | None = None
 
     created_at: datetime
     modified_at: datetime
@@ -96,6 +99,7 @@ class WorkflowRunStatusResponse(BaseModel):
     status: WorkflowRunStatus
     proxy_location: ProxyLocation | None = None
     webhook_callback_url: str | None = None
+    totp_verification_url: str | None = None
     created_at: datetime
     modified_at: datetime
     parameters: dict[str, Any]

skyvern/forge/sdk/workflow/models/yaml.py

@@ -224,5 +224,6 @@ class WorkflowCreateYAMLRequest(BaseModel):
     description: str | None = None
     proxy_location: ProxyLocation | None = None
     webhook_callback_url: str | None = None
+    totp_verification_url: str | None = None
     workflow_definition: WorkflowDefinitionYAML
     is_saved_task: bool = False

skyvern/forge/sdk/workflow/service.py

@@ -285,6 +285,7 @@ class WorkflowService:
         description: str | None = None,
         proxy_location: ProxyLocation | None = None,
         webhook_callback_url: str | None = None,
+        totp_verification_url: str | None = None,
         workflow_permanent_id: str | None = None,
         version: int | None = None,
         is_saved_task: bool = False,
@@ -296,6 +297,7 @@ class WorkflowService:
             description=description,
             proxy_location=proxy_location,
             webhook_callback_url=webhook_callback_url,
+            totp_verification_url=totp_verification_url,
             workflow_permanent_id=workflow_permanent_id,
             version=version,
             is_saved_task=is_saved_task,
@@ -392,6 +394,7 @@ class WorkflowService:
             organization_id=organization_id,
             proxy_location=workflow_request.proxy_location,
             webhook_callback_url=workflow_request.webhook_callback_url,
+            totp_verification_url=workflow_request.totp_verification_url,
         )
 
     async def mark_workflow_run_as_completed(self, workflow_run_id: str) -> None:
@@ -634,6 +637,7 @@ class WorkflowService:
             status=workflow_run.status,
             proxy_location=workflow_run.proxy_location,
             webhook_callback_url=workflow_run.webhook_callback_url,
+            totp_verification_url=workflow_run.totp_verification_url,
             created_at=workflow_run.created_at,
             modified_at=workflow_run.modified_at,
             parameters=parameters_with_value,
@@ -821,6 +825,7 @@ class WorkflowService:
                     organization_id=organization_id,
                     proxy_location=request.proxy_location,
                     webhook_callback_url=request.webhook_callback_url,
+                    totp_verification_url=request.totp_verification_url,
                     workflow_permanent_id=workflow_permanent_id,
                     version=existing_version + 1,
                     is_saved_task=request.is_saved_task,
@@ -833,6 +838,7 @@ class WorkflowService:
                     organization_id=organization_id,
                     proxy_location=request.proxy_location,
                     webhook_callback_url=request.webhook_callback_url,
+                    totp_verification_url=request.totp_verification_url,
                     is_saved_task=request.is_saved_task,
                 )
             # Create parameters from the request