From 281977f39525d24f155918cf1ab848bd27b4f19f Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Sun, 8 Sep 2024 17:57:03 -0700
Subject: [PATCH] update and fix totp schema (#785)

---
 skyvern/forge/sdk/schemas/totp_codes.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/skyvern/forge/sdk/schemas/totp_codes.py b/skyvern/forge/sdk/schemas/totp_codes.py
index 3ca1dd36..392fb34f 100644
--- a/skyvern/forge/sdk/schemas/totp_codes.py
+++ b/skyvern/forge/sdk/schemas/totp_codes.py
@@ -7,7 +7,6 @@ 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
@@ -18,12 +17,12 @@ class TOTPCodeBase(BaseModel):
 
 class TOTPCodeCreate(TOTPCodeBase):
     totp_identifier: str
-    organization_id: str
     content: str
 
 
 class TOTPCode(TOTPCodeCreate):
     totp_code_id: str
     code: str
+    organization_id: str
     created_at: datetime
     modified_at: datetime