shu/fix totp script gen (#3423)

2025-09-12 12:41:23 -07:00
parent 8df506660e
commit dde6837df8
4 changed files with 16 additions and 3 deletions
--- a/skyvern/forge/sdk/core/skyvern_context.py
+++ b/skyvern/forge/sdk/core/skyvern_context.py
@@ -39,6 +39,10 @@ class SkyvernContext:
    def __str__(self) -> str:
        return self.__repr__()

+    def pop_totp_code(self, task_id: str) -> None:
+        if task_id in self.totp_codes:
+            self.totp_codes.pop(task_id)
+

 _context: ContextVar[SkyvernContext | None] = ContextVar(
    "Global context",