TOTP code db + agent support for fetching totp_code from db (#784)

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

skyvern/forge/sdk/schemas/tasks.py

@@ -39,6 +39,7 @@ class TaskRequest(BaseModel):
         examples=["https://my-webhook.com"],
     )
     totp_verification_url: str | None = None
+    totp_identifier: str | None = None
     navigation_goal: str | None = Field(
         default=None,
         description="The user's goal for the task.",

skyvern/forge/sdk/schemas/totp_codes.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+
+from pydantic import BaseModel, ConfigDict
+
+
+class TOTPCodeBase(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    totp_identifier: str | None = None
+    organization_id: str | None = None
+    task_id: str | None = None
+    workflow_id: str | None = None
+    source: str | None = None
+    content: str | None = None
+
+    expired_at: datetime | None = None
+
+
+class TOTPCodeCreate(TOTPCodeBase):
+    totp_identifier: str
+    organization_id: str
+    content: str
+
+
+class TOTPCode(TOTPCodeCreate):
+    totp_code_id: str
+    code: str
+    created_at: datetime
+    modified_at: datetime