From 17e938bc87693eaaeb53f0aa14b79efc9a797150 Mon Sep 17 00:00:00 2001 From: Guy Lichtenfeld Date: Wed, 8 Jan 2025 14:41:10 +0200 Subject: [PATCH] fix: do not require all fields for RequiredAction --- ai21/models/assistant/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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