support workflow_run_id in totp code (#2199)

2025-04-21 08:24:29 +08:00
parent 8b9c7c17e3
commit 9038e6e047
6 changed files with 46 additions and 1 deletions
--- a/skyvern/forge/sdk/db/client.py
+++ b/skyvern/forge/sdk/db/client.py
@@ -2089,6 +2089,7 @@ class AgentDB:
        code: str,
        task_id: str | None = None,
        workflow_id: str | None = None,
+        workflow_run_id: str | None = None,
        source: str | None = None,
        expired_at: datetime | None = None,
    ) -> TOTPCode:
@@ -2100,6 +2101,7 @@ class AgentDB:
                code=code,
                task_id=task_id,
                workflow_id=workflow_id,
+                workflow_run_id=workflow_run_id,
                source=source,
                expired_at=expired_at,
            )