Credentials page UX improvement (#3941)

skyvern/forge/sdk/db/client.py

@@ -3290,7 +3290,7 @@ class AgentDB:
         self,
         organization_id: str,
         limit: int = 50,
-        valid_lifespan_minutes: int = settings.TOTP_LIFESPAN_MINUTES,
+        valid_lifespan_minutes: int | None = None,
         otp_type: OTPType | None = None,
         workflow_run_id: str | None = None,
     ) -> list[TOTPCode]:
@@ -3304,11 +3304,13 @@ class AgentDB:
             TOTPCodeModel.workflow_run_id.is_(None),
         )
         async with self.Session() as session:
-            query = (
-                select(TOTPCodeModel)
-                .filter_by(organization_id=organization_id)
-                .filter(TOTPCodeModel.created_at > datetime.utcnow() - timedelta(minutes=valid_lifespan_minutes))
-            )
+            query = select(TOTPCodeModel).filter_by(organization_id=organization_id)
+
+            if valid_lifespan_minutes is not None:
+                query = query.filter(
+                    TOTPCodeModel.created_at > datetime.utcnow() - timedelta(minutes=valid_lifespan_minutes)
+                )
+
             if otp_type:
                 query = query.filter(TOTPCodeModel.otp_type == otp_type)
             if workflow_run_id is not None:

skyvern/forge/sdk/routes/credentials.py

@@ -177,6 +177,7 @@ async def get_totp_codes(
         codes = await app.DATABASE.get_recent_otp_codes(
             organization_id=curr_org.organization_id,
             limit=limit,
+            valid_lifespan_minutes=None,
             otp_type=otp_type,
             workflow_run_id=workflow_run_id,
         )