diff --git a/ai21/models/assistant/run.py b/ai21/models/assistant/run.py
index 715488e3..3d29eaa0 100644
--- a/ai21/models/assistant/run.py
+++ b/ai21/models/assistant/run.py
@@ -45,7 +45,7 @@ class SubmitInput(TypedDict):
     data: Dict[str, Any]
 
 
-class RequiredAction(TypedDict):
+class RequiredAction(TypedDict, total=False):
     type: Literal["submit_tool_outputs", "submit_input"]
     submit_tool_outputs: Optional[SubmitToolCallOutputs] = None
     submit_input: Optional[SubmitInput] = None