add 2fa type tracking to credentials (#3647)

skyvern/forge/sdk/db/client.py

@@ -3623,6 +3623,7 @@ class AgentDB:
         credential_type: CredentialType,
         organization_id: str,
         item_id: str,
+        totp_type: str = "none",
     ) -> Credential:
         async with self.Session() as session:
             credential = CredentialModel(
@@ -3630,6 +3631,7 @@ class AgentDB:
                 name=name,
                 credential_type=credential_type,
                 item_id=item_id,
+                totp_type=totp_type,
             )
             session.add(credential)
             await session.commit()

skyvern/forge/sdk/db/models.py

@@ -818,6 +818,7 @@ class CredentialModel(Base):
 
     name = Column(String, nullable=False)
     credential_type = Column(String, nullable=False)
+    totp_type = Column(String, nullable=False, default="none")
 
     created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
     modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)