From 4a990060a137b7405e90d674667b774c2ab3d9df Mon Sep 17 00:00:00 2001 From: Sachin Padmanabhan Date: Wed, 11 Dec 2024 16:34:43 -0800 Subject: [PATCH] generate correct OpenAPI for nullable zod unions (#480) --- py/src/braintrust/_types.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/src/braintrust/_types.py b/py/src/braintrust/_types.py index 1009e855..24c86f25 100644 --- a/py/src/braintrust/_types.py +++ b/py/src/braintrust/_types.py @@ -274,7 +274,7 @@ class ModelParams1(TypedDict): max_tokens: NotRequired[Optional[float]] frequency_penalty: NotRequired[Optional[float]] presence_penalty: NotRequired[Optional[float]] - response_format: NotRequired[Optional[Union[ResponseFormat, ResponseFormat1, ResponseFormat2, Mapping[str, Any]]]] + response_format: NotRequired[Optional[Union[ResponseFormat, ResponseFormat1, ResponseFormat2]]] tool_choice: NotRequired[Optional[Union[Literal["auto"], Literal["none"], Literal["required"], ToolChoice]]] function_call: NotRequired[Optional[Union[Literal["auto"], Literal["none"], FunctionCall]]] n: NotRequired[Optional[float]]