migrate Organization data model from skyvern/forge/sdk/models.py to skyvern/forge/sdk/schemas/organizations.py (#1343)

skyvern/forge/sdk/schemas/organizations.py

@@ -1,6 +1,34 @@
-from pydantic import BaseModel
+from datetime import datetime
 
-from skyvern.forge.sdk.models import Organization, OrganizationAuthToken
+from pydantic import BaseModel, ConfigDict
+
+from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType
+
+
+class Organization(BaseModel):
+    model_config = ConfigDict(from_attributes=True)
+
+    organization_id: str
+    organization_name: str
+    webhook_callback_url: str | None = None
+    max_steps_per_run: int | None = None
+    max_retries_per_step: int | None = None
+    domain: str | None = None
+    bw_organization_id: str | None = None
+    bw_collection_ids: list[str] | None = None
+
+    created_at: datetime
+    modified_at: datetime
+
+
+class OrganizationAuthToken(BaseModel):
+    id: str
+    organization_id: str
+    token_type: OrganizationAuthTokenType
+    token: str
+    valid: bool
+    created_at: datetime
+    modified_at: datetime
 
 
 class GetOrganizationsResponse(BaseModel):