diff --git a/.stats.yml b/.stats.yml
index 7cdabf4..f5c30e9 100644
--- a/.stats.yml
+++ b/.stats.yml
@@ -1,2 +1,2 @@
 configured_endpoints: 50
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-43adb46012be95b03aec84cd642acd07342b7e9bb79077256f4827c98846c644.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/julep-ai-inc-dash%2Fjulep-44f4c6ee81a60ead4d0d9cbc4eba836ce7c8ecbd827f62bfd7bf17ce113fe63f.yml
diff --git a/src/julep/types/task.py b/src/julep/types/task.py
index 8b7df29..8d7cded 100644
--- a/src/julep/types/task.py
+++ b/src/julep/types/task.py
@@ -43,11 +43,11 @@
     "MainSearchStepSearchVectorDocSearchRequest",
     "MainSearchStepSearchTextOnlyDocSearchRequest",
     "MainSearchStepSearchHybridDocSearchRequest",
+    "MainYieldStep",
     "MainReturnStep",
     "MainSleepStep",
     "MainSleepStepSleep",
     "MainErrorWorkflowStep",
-    "MainYieldStep",
     "MainWaitForInputStep",
     "MainWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepOutput",
@@ -82,11 +82,11 @@
     "MainIfElseWorkflowStepOutputThenSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepOutputThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepOutputThenSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepOutputThenYieldStep",
     "MainIfElseWorkflowStepOutputThenReturnStep",
     "MainIfElseWorkflowStepOutputThenSleepStep",
     "MainIfElseWorkflowStepOutputThenSleepStepSleep",
     "MainIfElseWorkflowStepOutputThenErrorWorkflowStep",
-    "MainIfElseWorkflowStepOutputThenYieldStep",
     "MainIfElseWorkflowStepOutputThenWaitForInputStep",
     "MainIfElseWorkflowStepOutputThenWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepOutputElse",
@@ -120,11 +120,11 @@
     "MainIfElseWorkflowStepOutputElseSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepOutputElseSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepOutputElseSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepOutputElseYieldStep",
     "MainIfElseWorkflowStepOutputElseReturnStep",
     "MainIfElseWorkflowStepOutputElseSleepStep",
     "MainIfElseWorkflowStepOutputElseSleepStepSleep",
     "MainIfElseWorkflowStepOutputElseErrorWorkflowStep",
-    "MainIfElseWorkflowStepOutputElseYieldStep",
     "MainIfElseWorkflowStepOutputElseWaitForInputStep",
     "MainIfElseWorkflowStepOutputElseWaitForInputStepWaitForInput",
     "MainSwitchStepOutput",
@@ -160,11 +160,11 @@
     "MainSwitchStepOutputSwitchThenSearchStepSearchVectorDocSearchRequest",
     "MainSwitchStepOutputSwitchThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainSwitchStepOutputSwitchThenSearchStepSearchHybridDocSearchRequest",
+    "MainSwitchStepOutputSwitchThenYieldStep",
     "MainSwitchStepOutputSwitchThenReturnStep",
     "MainSwitchStepOutputSwitchThenSleepStep",
     "MainSwitchStepOutputSwitchThenSleepStepSleep",
     "MainSwitchStepOutputSwitchThenErrorWorkflowStep",
-    "MainSwitchStepOutputSwitchThenYieldStep",
     "MainSwitchStepOutputSwitchThenWaitForInputStep",
     "MainSwitchStepOutputSwitchThenWaitForInputStepWaitForInput",
     "MainForeachStepOutput",
@@ -200,6 +200,7 @@
     "MainForeachStepOutputForeachDoSearchStepSearchVectorDocSearchRequest",
     "MainForeachStepOutputForeachDoSearchStepSearchTextOnlyDocSearchRequest",
     "MainForeachStepOutputForeachDoSearchStepSearchHybridDocSearchRequest",
+    "MainForeachStepOutputForeachDoYieldStep",
     "MainParallelStepOutput",
     "MainParallelStepOutputParallel",
     "MainParallelStepOutputParallelEvaluateStep",
@@ -232,6 +233,7 @@
     "MainParallelStepOutputParallelSearchStepSearchVectorDocSearchRequest",
     "MainParallelStepOutputParallelSearchStepSearchTextOnlyDocSearchRequest",
     "MainParallelStepOutputParallelSearchStepSearchHybridDocSearchRequest",
+    "MainParallelStepOutputParallelYieldStep",
     "MainMainOutput",
     "MainMainOutputMap",
     "MainMainOutputMapEvaluateStep",
@@ -264,6 +266,7 @@
     "MainMainOutputMapSearchStepSearchVectorDocSearchRequest",
     "MainMainOutputMapSearchStepSearchTextOnlyDocSearchRequest",
     "MainMainOutputMapSearchStepSearchHybridDocSearchRequest",
+    "MainMainOutputMapYieldStep",
 ]
 
 
@@ -487,6 +490,14 @@ class MainSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 class MainReturnStep(BaseModel):
     return_: Dict[str, str] = FieldInfo(alias="return")
 
@@ -515,14 +526,6 @@ class MainErrorWorkflowStep(BaseModel):
     kind: Optional[Literal["error"]] = FieldInfo(alias="kind_", default=None)
 
 
-class MainYieldStep(BaseModel):
-    workflow: str
-
-    arguments: Union[Dict[str, str], Literal["_"], None] = None
-
-    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
-
-
 class MainWaitForInputStepWaitForInput(BaseModel):
     info: Dict[str, str]
 
@@ -762,6 +765,14 @@ class MainIfElseWorkflowStepOutputThenSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainIfElseWorkflowStepOutputThenYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 class MainIfElseWorkflowStepOutputThenReturnStep(BaseModel):
     return_: Dict[str, str] = FieldInfo(alias="return")
 
@@ -790,14 +801,6 @@ class MainIfElseWorkflowStepOutputThenErrorWorkflowStep(BaseModel):
     kind: Optional[Literal["error"]] = FieldInfo(alias="kind_", default=None)
 
 
-class MainIfElseWorkflowStepOutputThenYieldStep(BaseModel):
-    workflow: str
-
-    arguments: Union[Dict[str, str], Literal["_"], None] = None
-
-    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
-
-
 class MainIfElseWorkflowStepOutputThenWaitForInputStepWaitForInput(BaseModel):
     info: Dict[str, str]
 
@@ -817,10 +820,10 @@ class MainIfElseWorkflowStepOutputThenWaitForInputStep(BaseModel):
     MainIfElseWorkflowStepOutputThenLogStep,
     MainIfElseWorkflowStepOutputThenEmbedStep,
     MainIfElseWorkflowStepOutputThenSearchStep,
+    MainIfElseWorkflowStepOutputThenYieldStep,
     MainIfElseWorkflowStepOutputThenReturnStep,
     MainIfElseWorkflowStepOutputThenSleepStep,
     MainIfElseWorkflowStepOutputThenErrorWorkflowStep,
-    MainIfElseWorkflowStepOutputThenYieldStep,
     MainIfElseWorkflowStepOutputThenWaitForInputStep,
 ]
 
@@ -1054,6 +1057,14 @@ class MainIfElseWorkflowStepOutputElseSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainIfElseWorkflowStepOutputElseYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 class MainIfElseWorkflowStepOutputElseReturnStep(BaseModel):
     return_: Dict[str, str] = FieldInfo(alias="return")
 
@@ -1082,14 +1093,6 @@ class MainIfElseWorkflowStepOutputElseErrorWorkflowStep(BaseModel):
     kind: Optional[Literal["error"]] = FieldInfo(alias="kind_", default=None)
 
 
-class MainIfElseWorkflowStepOutputElseYieldStep(BaseModel):
-    workflow: str
-
-    arguments: Union[Dict[str, str], Literal["_"], None] = None
-
-    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
-
-
 class MainIfElseWorkflowStepOutputElseWaitForInputStepWaitForInput(BaseModel):
     info: Dict[str, str]
 
@@ -1109,10 +1112,10 @@ class MainIfElseWorkflowStepOutputElseWaitForInputStep(BaseModel):
     MainIfElseWorkflowStepOutputElseLogStep,
     MainIfElseWorkflowStepOutputElseEmbedStep,
     MainIfElseWorkflowStepOutputElseSearchStep,
+    MainIfElseWorkflowStepOutputElseYieldStep,
     MainIfElseWorkflowStepOutputElseReturnStep,
     MainIfElseWorkflowStepOutputElseSleepStep,
     MainIfElseWorkflowStepOutputElseErrorWorkflowStep,
-    MainIfElseWorkflowStepOutputElseYieldStep,
     MainIfElseWorkflowStepOutputElseWaitForInputStep,
     None,
 ]
@@ -1355,6 +1358,14 @@ class MainSwitchStepOutputSwitchThenSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainSwitchStepOutputSwitchThenYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 class MainSwitchStepOutputSwitchThenReturnStep(BaseModel):
     return_: Dict[str, str] = FieldInfo(alias="return")
 
@@ -1383,14 +1394,6 @@ class MainSwitchStepOutputSwitchThenErrorWorkflowStep(BaseModel):
     kind: Optional[Literal["error"]] = FieldInfo(alias="kind_", default=None)
 
 
-class MainSwitchStepOutputSwitchThenYieldStep(BaseModel):
-    workflow: str
-
-    arguments: Union[Dict[str, str], Literal["_"], None] = None
-
-    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
-
-
 class MainSwitchStepOutputSwitchThenWaitForInputStepWaitForInput(BaseModel):
     info: Dict[str, str]
 
@@ -1410,10 +1413,10 @@ class MainSwitchStepOutputSwitchThenWaitForInputStep(BaseModel):
     MainSwitchStepOutputSwitchThenLogStep,
     MainSwitchStepOutputSwitchThenEmbedStep,
     MainSwitchStepOutputSwitchThenSearchStep,
+    MainSwitchStepOutputSwitchThenYieldStep,
     MainSwitchStepOutputSwitchThenReturnStep,
     MainSwitchStepOutputSwitchThenSleepStep,
     MainSwitchStepOutputSwitchThenErrorWorkflowStep,
-    MainSwitchStepOutputSwitchThenYieldStep,
     MainSwitchStepOutputSwitchThenWaitForInputStep,
 ]
 
@@ -1657,6 +1660,14 @@ class MainForeachStepOutputForeachDoSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainForeachStepOutputForeachDoYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 MainForeachStepOutputForeachDo: TypeAlias = Union[
     MainForeachStepOutputForeachDoEvaluateStep,
     MainForeachStepOutputForeachDoToolCallStep,
@@ -1666,6 +1677,7 @@ class MainForeachStepOutputForeachDoSearchStep(BaseModel):
     MainForeachStepOutputForeachDoLogStep,
     MainForeachStepOutputForeachDoEmbedStep,
     MainForeachStepOutputForeachDoSearchStep,
+    MainForeachStepOutputForeachDoYieldStep,
 ]
 
 
@@ -1908,6 +1920,14 @@ class MainParallelStepOutputParallelSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainParallelStepOutputParallelYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 MainParallelStepOutputParallel: TypeAlias = Union[
     MainParallelStepOutputParallelEvaluateStep,
     MainParallelStepOutputParallelToolCallStep,
@@ -1917,6 +1937,7 @@ class MainParallelStepOutputParallelSearchStep(BaseModel):
     MainParallelStepOutputParallelLogStep,
     MainParallelStepOutputParallelEmbedStep,
     MainParallelStepOutputParallelSearchStep,
+    MainParallelStepOutputParallelYieldStep,
 ]
 
 
@@ -2147,6 +2168,14 @@ class MainMainOutputMapSearchStep(BaseModel):
     kind: Optional[Literal["search"]] = FieldInfo(alias="kind_", default=None)
 
 
+class MainMainOutputMapYieldStep(BaseModel):
+    workflow: str
+
+    arguments: Union[Dict[str, str], Literal["_"], None] = None
+
+    kind: Optional[Literal["yield"]] = FieldInfo(alias="kind_", default=None)
+
+
 MainMainOutputMap: TypeAlias = Union[
     MainMainOutputMapEvaluateStep,
     MainMainOutputMapToolCallStep,
@@ -2156,6 +2185,7 @@ class MainMainOutputMapSearchStep(BaseModel):
     MainMainOutputMapLogStep,
     MainMainOutputMapEmbedStep,
     MainMainOutputMapSearchStep,
+    MainMainOutputMapYieldStep,
 ]
 
 
@@ -2182,10 +2212,10 @@ class MainMainOutput(BaseModel):
     MainLogStep,
     MainEmbedStep,
     MainSearchStep,
+    MainYieldStep,
     MainReturnStep,
     MainSleepStep,
     MainErrorWorkflowStep,
-    MainYieldStep,
     MainWaitForInputStep,
     MainIfElseWorkflowStepOutput,
     MainSwitchStepOutput,
diff --git a/src/julep/types/task_create_or_update_params.py b/src/julep/types/task_create_or_update_params.py
index c476a08..0dc1e16 100644
--- a/src/julep/types/task_create_or_update_params.py
+++ b/src/julep/types/task_create_or_update_params.py
@@ -41,11 +41,11 @@
     "MainSearchStepSearchVectorDocSearchRequest",
     "MainSearchStepSearchTextOnlyDocSearchRequest",
     "MainSearchStepSearchHybridDocSearchRequest",
+    "MainYieldStep",
     "MainReturnStep",
     "MainSleepStep",
     "MainSleepStepSleep",
     "MainErrorWorkflowStep",
-    "MainYieldStep",
     "MainWaitForInputStep",
     "MainWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepInput",
@@ -80,11 +80,11 @@
     "MainIfElseWorkflowStepInputThenSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepInputThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepInputThenSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepInputThenYieldStep",
     "MainIfElseWorkflowStepInputThenReturnStep",
     "MainIfElseWorkflowStepInputThenSleepStep",
     "MainIfElseWorkflowStepInputThenSleepStepSleep",
     "MainIfElseWorkflowStepInputThenErrorWorkflowStep",
-    "MainIfElseWorkflowStepInputThenYieldStep",
     "MainIfElseWorkflowStepInputThenWaitForInputStep",
     "MainIfElseWorkflowStepInputThenWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepInputElse",
@@ -118,11 +118,11 @@
     "MainIfElseWorkflowStepInputElseSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepInputElseSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepInputElseSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepInputElseYieldStep",
     "MainIfElseWorkflowStepInputElseReturnStep",
     "MainIfElseWorkflowStepInputElseSleepStep",
     "MainIfElseWorkflowStepInputElseSleepStepSleep",
     "MainIfElseWorkflowStepInputElseErrorWorkflowStep",
-    "MainIfElseWorkflowStepInputElseYieldStep",
     "MainIfElseWorkflowStepInputElseWaitForInputStep",
     "MainIfElseWorkflowStepInputElseWaitForInputStepWaitForInput",
     "MainSwitchStepInput",
@@ -158,11 +158,11 @@
     "MainSwitchStepInputSwitchThenSearchStepSearchVectorDocSearchRequest",
     "MainSwitchStepInputSwitchThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainSwitchStepInputSwitchThenSearchStepSearchHybridDocSearchRequest",
+    "MainSwitchStepInputSwitchThenYieldStep",
     "MainSwitchStepInputSwitchThenReturnStep",
     "MainSwitchStepInputSwitchThenSleepStep",
     "MainSwitchStepInputSwitchThenSleepStepSleep",
     "MainSwitchStepInputSwitchThenErrorWorkflowStep",
-    "MainSwitchStepInputSwitchThenYieldStep",
     "MainSwitchStepInputSwitchThenWaitForInputStep",
     "MainSwitchStepInputSwitchThenWaitForInputStepWaitForInput",
     "MainForeachStepInput",
@@ -198,6 +198,7 @@
     "MainForeachStepInputForeachDoSearchStepSearchVectorDocSearchRequest",
     "MainForeachStepInputForeachDoSearchStepSearchTextOnlyDocSearchRequest",
     "MainForeachStepInputForeachDoSearchStepSearchHybridDocSearchRequest",
+    "MainForeachStepInputForeachDoYieldStep",
     "MainParallelStepInput",
     "MainParallelStepInputParallel",
     "MainParallelStepInputParallelEvaluateStep",
@@ -230,6 +231,7 @@
     "MainParallelStepInputParallelSearchStepSearchVectorDocSearchRequest",
     "MainParallelStepInputParallelSearchStepSearchTextOnlyDocSearchRequest",
     "MainParallelStepInputParallelSearchStepSearchHybridDocSearchRequest",
+    "MainParallelStepInputParallelYieldStep",
     "MainMainInput",
     "MainMainInputMap",
     "MainMainInputMapEvaluateStep",
@@ -262,6 +264,7 @@
     "MainMainInputMapSearchStepSearchVectorDocSearchRequest",
     "MainMainInputMapSearchStepSearchTextOnlyDocSearchRequest",
     "MainMainInputMapSearchStepSearchHybridDocSearchRequest",
+    "MainMainInputMapYieldStep",
 ]
 
 
@@ -493,6 +496,12 @@ class MainSearchStep(TypedDict, total=False):
     search: Required[MainSearchStepSearch]
 
 
+class MainYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainReturnStepReservedKeywords = TypedDict(
     "_MainReturnStepReservedKeywords",
     {
@@ -524,12 +533,6 @@ class MainErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -769,6 +772,12 @@ class MainIfElseWorkflowStepInputThenSearchStep(TypedDict, total=False):
     search: Required[MainIfElseWorkflowStepInputThenSearchStepSearch]
 
 
+class MainIfElseWorkflowStepInputThenYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainIfElseWorkflowStepInputThenReturnStepReservedKeywords = TypedDict(
     "_MainIfElseWorkflowStepInputThenReturnStepReservedKeywords",
     {
@@ -802,12 +811,6 @@ class MainIfElseWorkflowStepInputThenErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainIfElseWorkflowStepInputThenYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainIfElseWorkflowStepInputThenWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -825,10 +828,10 @@ class MainIfElseWorkflowStepInputThenWaitForInputStep(TypedDict, total=False):
     MainIfElseWorkflowStepInputThenLogStep,
     MainIfElseWorkflowStepInputThenEmbedStep,
     MainIfElseWorkflowStepInputThenSearchStep,
+    MainIfElseWorkflowStepInputThenYieldStep,
     MainIfElseWorkflowStepInputThenReturnStep,
     MainIfElseWorkflowStepInputThenSleepStep,
     MainIfElseWorkflowStepInputThenErrorWorkflowStep,
-    MainIfElseWorkflowStepInputThenYieldStep,
     MainIfElseWorkflowStepInputThenWaitForInputStep,
 ]
 
@@ -1064,6 +1067,12 @@ class MainIfElseWorkflowStepInputElseSearchStep(TypedDict, total=False):
     search: Required[MainIfElseWorkflowStepInputElseSearchStepSearch]
 
 
+class MainIfElseWorkflowStepInputElseYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainIfElseWorkflowStepInputElseReturnStepReservedKeywords = TypedDict(
     "_MainIfElseWorkflowStepInputElseReturnStepReservedKeywords",
     {
@@ -1097,12 +1106,6 @@ class MainIfElseWorkflowStepInputElseErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainIfElseWorkflowStepInputElseYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainIfElseWorkflowStepInputElseWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -1120,10 +1123,10 @@ class MainIfElseWorkflowStepInputElseWaitForInputStep(TypedDict, total=False):
     MainIfElseWorkflowStepInputElseLogStep,
     MainIfElseWorkflowStepInputElseEmbedStep,
     MainIfElseWorkflowStepInputElseSearchStep,
+    MainIfElseWorkflowStepInputElseYieldStep,
     MainIfElseWorkflowStepInputElseReturnStep,
     MainIfElseWorkflowStepInputElseSleepStep,
     MainIfElseWorkflowStepInputElseErrorWorkflowStep,
-    MainIfElseWorkflowStepInputElseYieldStep,
     MainIfElseWorkflowStepInputElseWaitForInputStep,
 ]
 
@@ -1366,6 +1369,12 @@ class MainSwitchStepInputSwitchThenSearchStep(TypedDict, total=False):
     search: Required[MainSwitchStepInputSwitchThenSearchStepSearch]
 
 
+class MainSwitchStepInputSwitchThenYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainSwitchStepInputSwitchThenReturnStepReservedKeywords = TypedDict(
     "_MainSwitchStepInputSwitchThenReturnStepReservedKeywords",
     {
@@ -1397,12 +1406,6 @@ class MainSwitchStepInputSwitchThenErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainSwitchStepInputSwitchThenYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainSwitchStepInputSwitchThenWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -1420,10 +1423,10 @@ class MainSwitchStepInputSwitchThenWaitForInputStep(TypedDict, total=False):
     MainSwitchStepInputSwitchThenLogStep,
     MainSwitchStepInputSwitchThenEmbedStep,
     MainSwitchStepInputSwitchThenSearchStep,
+    MainSwitchStepInputSwitchThenYieldStep,
     MainSwitchStepInputSwitchThenReturnStep,
     MainSwitchStepInputSwitchThenSleepStep,
     MainSwitchStepInputSwitchThenErrorWorkflowStep,
-    MainSwitchStepInputSwitchThenYieldStep,
     MainSwitchStepInputSwitchThenWaitForInputStep,
 ]
 
@@ -1663,6 +1666,12 @@ class MainForeachStepInputForeachDoSearchStep(TypedDict, total=False):
     search: Required[MainForeachStepInputForeachDoSearchStepSearch]
 
 
+class MainForeachStepInputForeachDoYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainForeachStepInputForeachDo: TypeAlias = Union[
     MainForeachStepInputForeachDoEvaluateStep,
     MainForeachStepInputForeachDoToolCallStep,
@@ -1672,6 +1681,7 @@ class MainForeachStepInputForeachDoSearchStep(TypedDict, total=False):
     MainForeachStepInputForeachDoLogStep,
     MainForeachStepInputForeachDoEmbedStep,
     MainForeachStepInputForeachDoSearchStep,
+    MainForeachStepInputForeachDoYieldStep,
 ]
 
 _MainForeachStepInputForeachReservedKeywords = TypedDict(
@@ -1916,6 +1926,12 @@ class MainParallelStepInputParallelSearchStep(TypedDict, total=False):
     search: Required[MainParallelStepInputParallelSearchStepSearch]
 
 
+class MainParallelStepInputParallelYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainParallelStepInputParallel: TypeAlias = Union[
     MainParallelStepInputParallelEvaluateStep,
     MainParallelStepInputParallelToolCallStep,
@@ -1925,6 +1941,7 @@ class MainParallelStepInputParallelSearchStep(TypedDict, total=False):
     MainParallelStepInputParallelLogStep,
     MainParallelStepInputParallelEmbedStep,
     MainParallelStepInputParallelSearchStep,
+    MainParallelStepInputParallelYieldStep,
 ]
 
 
@@ -2149,6 +2166,12 @@ class MainMainInputMapSearchStep(TypedDict, total=False):
     search: Required[MainMainInputMapSearchStepSearch]
 
 
+class MainMainInputMapYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainMainInputMap: TypeAlias = Union[
     MainMainInputMapEvaluateStep,
     MainMainInputMapToolCallStep,
@@ -2158,6 +2181,7 @@ class MainMainInputMapSearchStep(TypedDict, total=False):
     MainMainInputMapLogStep,
     MainMainInputMapEmbedStep,
     MainMainInputMapSearchStep,
+    MainMainInputMapYieldStep,
 ]
 
 
@@ -2182,10 +2206,10 @@ class MainMainInput(TypedDict, total=False):
     MainLogStep,
     MainEmbedStep,
     MainSearchStep,
+    MainYieldStep,
     MainReturnStep,
     MainSleepStep,
     MainErrorWorkflowStep,
-    MainYieldStep,
     MainWaitForInputStep,
     MainIfElseWorkflowStepInput,
     MainSwitchStepInput,
diff --git a/src/julep/types/task_create_params.py b/src/julep/types/task_create_params.py
index d23993f..8acd8bd 100644
--- a/src/julep/types/task_create_params.py
+++ b/src/julep/types/task_create_params.py
@@ -41,11 +41,11 @@
     "MainSearchStepSearchVectorDocSearchRequest",
     "MainSearchStepSearchTextOnlyDocSearchRequest",
     "MainSearchStepSearchHybridDocSearchRequest",
+    "MainYieldStep",
     "MainReturnStep",
     "MainSleepStep",
     "MainSleepStepSleep",
     "MainErrorWorkflowStep",
-    "MainYieldStep",
     "MainWaitForInputStep",
     "MainWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepInput",
@@ -80,11 +80,11 @@
     "MainIfElseWorkflowStepInputThenSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepInputThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepInputThenSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepInputThenYieldStep",
     "MainIfElseWorkflowStepInputThenReturnStep",
     "MainIfElseWorkflowStepInputThenSleepStep",
     "MainIfElseWorkflowStepInputThenSleepStepSleep",
     "MainIfElseWorkflowStepInputThenErrorWorkflowStep",
-    "MainIfElseWorkflowStepInputThenYieldStep",
     "MainIfElseWorkflowStepInputThenWaitForInputStep",
     "MainIfElseWorkflowStepInputThenWaitForInputStepWaitForInput",
     "MainIfElseWorkflowStepInputElse",
@@ -118,11 +118,11 @@
     "MainIfElseWorkflowStepInputElseSearchStepSearchVectorDocSearchRequest",
     "MainIfElseWorkflowStepInputElseSearchStepSearchTextOnlyDocSearchRequest",
     "MainIfElseWorkflowStepInputElseSearchStepSearchHybridDocSearchRequest",
+    "MainIfElseWorkflowStepInputElseYieldStep",
     "MainIfElseWorkflowStepInputElseReturnStep",
     "MainIfElseWorkflowStepInputElseSleepStep",
     "MainIfElseWorkflowStepInputElseSleepStepSleep",
     "MainIfElseWorkflowStepInputElseErrorWorkflowStep",
-    "MainIfElseWorkflowStepInputElseYieldStep",
     "MainIfElseWorkflowStepInputElseWaitForInputStep",
     "MainIfElseWorkflowStepInputElseWaitForInputStepWaitForInput",
     "MainSwitchStepInput",
@@ -158,11 +158,11 @@
     "MainSwitchStepInputSwitchThenSearchStepSearchVectorDocSearchRequest",
     "MainSwitchStepInputSwitchThenSearchStepSearchTextOnlyDocSearchRequest",
     "MainSwitchStepInputSwitchThenSearchStepSearchHybridDocSearchRequest",
+    "MainSwitchStepInputSwitchThenYieldStep",
     "MainSwitchStepInputSwitchThenReturnStep",
     "MainSwitchStepInputSwitchThenSleepStep",
     "MainSwitchStepInputSwitchThenSleepStepSleep",
     "MainSwitchStepInputSwitchThenErrorWorkflowStep",
-    "MainSwitchStepInputSwitchThenYieldStep",
     "MainSwitchStepInputSwitchThenWaitForInputStep",
     "MainSwitchStepInputSwitchThenWaitForInputStepWaitForInput",
     "MainForeachStepInput",
@@ -198,6 +198,7 @@
     "MainForeachStepInputForeachDoSearchStepSearchVectorDocSearchRequest",
     "MainForeachStepInputForeachDoSearchStepSearchTextOnlyDocSearchRequest",
     "MainForeachStepInputForeachDoSearchStepSearchHybridDocSearchRequest",
+    "MainForeachStepInputForeachDoYieldStep",
     "MainParallelStepInput",
     "MainParallelStepInputParallel",
     "MainParallelStepInputParallelEvaluateStep",
@@ -230,6 +231,7 @@
     "MainParallelStepInputParallelSearchStepSearchVectorDocSearchRequest",
     "MainParallelStepInputParallelSearchStepSearchTextOnlyDocSearchRequest",
     "MainParallelStepInputParallelSearchStepSearchHybridDocSearchRequest",
+    "MainParallelStepInputParallelYieldStep",
     "MainMainInput",
     "MainMainInputMap",
     "MainMainInputMapEvaluateStep",
@@ -262,6 +264,7 @@
     "MainMainInputMapSearchStepSearchVectorDocSearchRequest",
     "MainMainInputMapSearchStepSearchTextOnlyDocSearchRequest",
     "MainMainInputMapSearchStepSearchHybridDocSearchRequest",
+    "MainMainInputMapYieldStep",
 ]
 
 
@@ -491,6 +494,12 @@ class MainSearchStep(TypedDict, total=False):
     search: Required[MainSearchStepSearch]
 
 
+class MainYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainReturnStepReservedKeywords = TypedDict(
     "_MainReturnStepReservedKeywords",
     {
@@ -522,12 +531,6 @@ class MainErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -767,6 +770,12 @@ class MainIfElseWorkflowStepInputThenSearchStep(TypedDict, total=False):
     search: Required[MainIfElseWorkflowStepInputThenSearchStepSearch]
 
 
+class MainIfElseWorkflowStepInputThenYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainIfElseWorkflowStepInputThenReturnStepReservedKeywords = TypedDict(
     "_MainIfElseWorkflowStepInputThenReturnStepReservedKeywords",
     {
@@ -800,12 +809,6 @@ class MainIfElseWorkflowStepInputThenErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainIfElseWorkflowStepInputThenYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainIfElseWorkflowStepInputThenWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -823,10 +826,10 @@ class MainIfElseWorkflowStepInputThenWaitForInputStep(TypedDict, total=False):
     MainIfElseWorkflowStepInputThenLogStep,
     MainIfElseWorkflowStepInputThenEmbedStep,
     MainIfElseWorkflowStepInputThenSearchStep,
+    MainIfElseWorkflowStepInputThenYieldStep,
     MainIfElseWorkflowStepInputThenReturnStep,
     MainIfElseWorkflowStepInputThenSleepStep,
     MainIfElseWorkflowStepInputThenErrorWorkflowStep,
-    MainIfElseWorkflowStepInputThenYieldStep,
     MainIfElseWorkflowStepInputThenWaitForInputStep,
 ]
 
@@ -1062,6 +1065,12 @@ class MainIfElseWorkflowStepInputElseSearchStep(TypedDict, total=False):
     search: Required[MainIfElseWorkflowStepInputElseSearchStepSearch]
 
 
+class MainIfElseWorkflowStepInputElseYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainIfElseWorkflowStepInputElseReturnStepReservedKeywords = TypedDict(
     "_MainIfElseWorkflowStepInputElseReturnStepReservedKeywords",
     {
@@ -1095,12 +1104,6 @@ class MainIfElseWorkflowStepInputElseErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainIfElseWorkflowStepInputElseYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainIfElseWorkflowStepInputElseWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -1118,10 +1121,10 @@ class MainIfElseWorkflowStepInputElseWaitForInputStep(TypedDict, total=False):
     MainIfElseWorkflowStepInputElseLogStep,
     MainIfElseWorkflowStepInputElseEmbedStep,
     MainIfElseWorkflowStepInputElseSearchStep,
+    MainIfElseWorkflowStepInputElseYieldStep,
     MainIfElseWorkflowStepInputElseReturnStep,
     MainIfElseWorkflowStepInputElseSleepStep,
     MainIfElseWorkflowStepInputElseErrorWorkflowStep,
-    MainIfElseWorkflowStepInputElseYieldStep,
     MainIfElseWorkflowStepInputElseWaitForInputStep,
 ]
 
@@ -1364,6 +1367,12 @@ class MainSwitchStepInputSwitchThenSearchStep(TypedDict, total=False):
     search: Required[MainSwitchStepInputSwitchThenSearchStepSearch]
 
 
+class MainSwitchStepInputSwitchThenYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 _MainSwitchStepInputSwitchThenReturnStepReservedKeywords = TypedDict(
     "_MainSwitchStepInputSwitchThenReturnStepReservedKeywords",
     {
@@ -1395,12 +1404,6 @@ class MainSwitchStepInputSwitchThenErrorWorkflowStep(TypedDict, total=False):
     error: Required[str]
 
 
-class MainSwitchStepInputSwitchThenYieldStep(TypedDict, total=False):
-    workflow: Required[str]
-
-    arguments: Union[Dict[str, str], Literal["_"]]
-
-
 class MainSwitchStepInputSwitchThenWaitForInputStepWaitForInput(TypedDict, total=False):
     info: Required[Dict[str, str]]
 
@@ -1418,10 +1421,10 @@ class MainSwitchStepInputSwitchThenWaitForInputStep(TypedDict, total=False):
     MainSwitchStepInputSwitchThenLogStep,
     MainSwitchStepInputSwitchThenEmbedStep,
     MainSwitchStepInputSwitchThenSearchStep,
+    MainSwitchStepInputSwitchThenYieldStep,
     MainSwitchStepInputSwitchThenReturnStep,
     MainSwitchStepInputSwitchThenSleepStep,
     MainSwitchStepInputSwitchThenErrorWorkflowStep,
-    MainSwitchStepInputSwitchThenYieldStep,
     MainSwitchStepInputSwitchThenWaitForInputStep,
 ]
 
@@ -1661,6 +1664,12 @@ class MainForeachStepInputForeachDoSearchStep(TypedDict, total=False):
     search: Required[MainForeachStepInputForeachDoSearchStepSearch]
 
 
+class MainForeachStepInputForeachDoYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainForeachStepInputForeachDo: TypeAlias = Union[
     MainForeachStepInputForeachDoEvaluateStep,
     MainForeachStepInputForeachDoToolCallStep,
@@ -1670,6 +1679,7 @@ class MainForeachStepInputForeachDoSearchStep(TypedDict, total=False):
     MainForeachStepInputForeachDoLogStep,
     MainForeachStepInputForeachDoEmbedStep,
     MainForeachStepInputForeachDoSearchStep,
+    MainForeachStepInputForeachDoYieldStep,
 ]
 
 _MainForeachStepInputForeachReservedKeywords = TypedDict(
@@ -1914,6 +1924,12 @@ class MainParallelStepInputParallelSearchStep(TypedDict, total=False):
     search: Required[MainParallelStepInputParallelSearchStepSearch]
 
 
+class MainParallelStepInputParallelYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainParallelStepInputParallel: TypeAlias = Union[
     MainParallelStepInputParallelEvaluateStep,
     MainParallelStepInputParallelToolCallStep,
@@ -1923,6 +1939,7 @@ class MainParallelStepInputParallelSearchStep(TypedDict, total=False):
     MainParallelStepInputParallelLogStep,
     MainParallelStepInputParallelEmbedStep,
     MainParallelStepInputParallelSearchStep,
+    MainParallelStepInputParallelYieldStep,
 ]
 
 
@@ -2147,6 +2164,12 @@ class MainMainInputMapSearchStep(TypedDict, total=False):
     search: Required[MainMainInputMapSearchStepSearch]
 
 
+class MainMainInputMapYieldStep(TypedDict, total=False):
+    workflow: Required[str]
+
+    arguments: Union[Dict[str, str], Literal["_"]]
+
+
 MainMainInputMap: TypeAlias = Union[
     MainMainInputMapEvaluateStep,
     MainMainInputMapToolCallStep,
@@ -2156,6 +2179,7 @@ class MainMainInputMapSearchStep(TypedDict, total=False):
     MainMainInputMapLogStep,
     MainMainInputMapEmbedStep,
     MainMainInputMapSearchStep,
+    MainMainInputMapYieldStep,
 ]
 
 
@@ -2180,10 +2204,10 @@ class MainMainInput(TypedDict, total=False):
     MainLogStep,
     MainEmbedStep,
     MainSearchStep,
+    MainYieldStep,
     MainReturnStep,
     MainSleepStep,
     MainErrorWorkflowStep,
-    MainYieldStep,
     MainWaitForInputStep,
     MainIfElseWorkflowStepInput,
     MainSwitchStepInput,