More not allowing org id to be None in some cases (mypy only) (#2620)

skyvern/forge/sdk/schemas/ai_suggestions.py

@@ -12,7 +12,7 @@ class AISuggestion(AISuggestionBase):
     model_config = ConfigDict(from_attributes=True)
     ai_suggestion_type: str
     ai_suggestion_id: str
-    organization_id: str | None = None
+    organization_id: str
 
     created_at: datetime
     modified_at: datetime

skyvern/forge/sdk/schemas/task_v2.py

@@ -30,7 +30,7 @@ class TaskV2(BaseModel):
 
     observer_cruise_id: str = Field(alias="task_id")
     status: TaskV2Status
-    organization_id: str | None = None
+    organization_id: str
     workflow_run_id: str | None = None
     workflow_id: str | None = None
     workflow_permanent_id: str | None = None
@@ -99,7 +99,7 @@ class Thought(BaseModel):
 
     observer_thought_id: str = Field(alias="thought_id")
     observer_cruise_id: str = Field(alias="task_id")
-    organization_id: str | None = None
+    organization_id: str
     workflow_run_id: str | None = None
     workflow_run_block_id: str | None = None
     workflow_id: str | None = None

skyvern/forge/sdk/schemas/workflow_runs.py

@@ -15,7 +15,7 @@ class WorkflowRunBlock(BaseModel):
     workflow_run_block_id: str
     block_workflow_run_id: str | None = None
     workflow_run_id: str
-    organization_id: str | None = None
+    organization_id: str
     description: str | None = None
     parent_workflow_run_block_id: str | None = None
     block_type: BlockType