Don't allow None for org id (partial mypy only change) more to come (#2613)

skyvern/forge/sdk/db/client.py

@@ -312,7 +312,7 @@ class AgentDB:
             LOG.error("UnexpectedError", exc_info=True)
             raise
 
-    async def get_task_steps(self, task_id: str, organization_id: str | None = None) -> list[Step]:
+    async def get_task_steps(self, task_id: str, organization_id: str) -> list[Step]:
         try:
             async with self.Session() as session:
                 if steps := (

skyvern/forge/sdk/models.py

@@ -49,7 +49,7 @@ class Step(BaseModel):
     order: int
     is_last: bool
     retry_index: int = 0
-    organization_id: str | None = None
+    organization_id: str
     input_token_count: int = 0
     output_token_count: int = 0
     reasoning_token_count: int | None = None

skyvern/forge/sdk/schemas/tasks.py

@@ -231,7 +231,7 @@ class Task(TaskBase):
         None,
         description="The reason for the task failure.",
     )
-    organization_id: str | None = None
+    organization_id: str
     workflow_run_id: str | None = None
     workflow_permanent_id: str | None = None
     order: int | None = None