Make otp type also selectable in the totp endpoint (#4529)

skyvern/forge/sdk/routes/credentials.py

@@ -113,10 +113,10 @@ async def send_totp_code(
         if not workflow_run:
             raise HTTPException(status_code=400, detail=f"Invalid workflow run id: {data.workflow_run_id}")
     content = data.content.strip()
-    otp_value: OTPValue | None = OTPValue(value=content, type=OTPType.TOTP)
+    otp_value: OTPValue | None = OTPValue(value=content, type=data.type or OTPType.TOTP)
     # We assume the user is sending the code directly when the length of code is less than or equal to 10
     if len(content) > 10:
-        otp_value = await parse_otp_login(content, curr_org.organization_id)
+        otp_value = await parse_otp_login(content, curr_org.organization_id, enforced_otp_type=data.type)
 
     if not otp_value:
         LOG.error(

skyvern/forge/sdk/schemas/totp_codes.py

@@ -47,6 +47,11 @@ class TOTPCodeBase(BaseModel):
     )
 
 
+class OTPType(StrEnum):
+    TOTP = "totp"
+    MAGIC_LINK = "magic_link"
+
+
 class TOTPCodeCreate(TOTPCodeBase):
     totp_identifier: str = Field(
         ...,
@@ -58,6 +63,11 @@ class TOTPCodeCreate(TOTPCodeBase):
         description="The content of the TOTP code. It can be the email content that contains the TOTP code, or the sms message that contains the TOTP code. Skyvern will automatically extract the TOTP code from the content.",
         examples=["Hello, your verification code is 123456"],
     )
+    type: OTPType | None = Field(
+        default=None,
+        description="Optional. If provided, forces extraction of this specific OTP type (totp or magic_link). Use this when the content contains multiple OTP types and you want to specify which one to extract.",
+        examples=["totp", "magic_link"],
+    )
 
     @field_validator("content")
     @classmethod
@@ -66,11 +76,6 @@ class TOTPCodeCreate(TOTPCodeBase):
         return sanitize_postgres_text(value)
 
 
-class OTPType(StrEnum):
-    TOTP = "totp"
-    MAGIC_LINK = "magic_link"
-
-
 class TOTPCode(TOTPCodeCreate):
     totp_code_id: str = Field(..., description="The skyvern ID of the TOTP code.")
     code: str = Field(..., description="The TOTP code extracted from the content.")