From c0ef2e9dd0e681bd094e989f43586cdea2e16b23 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 7 Aug 2024 13:21:35 -0700 Subject: [PATCH] fix(main.py): fix linting error for python3.8 --- litellm/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/main.py b/litellm/main.py index 0e281b5ed..e1660b21a 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -31,6 +31,7 @@ from typing import ( Literal, Mapping, Optional, + Type, Union, ) @@ -608,7 +609,7 @@ def completion( logit_bias: Optional[dict] = None, user: Optional[str] = None, # openai v1.0+ new params - response_format: Optional[Union[dict, type[BaseModel]]] = None, + response_format: Optional[Union[dict, Type[BaseModel]]] = None, seed: Optional[int] = None, tools: Optional[List] = None, tool_choice: Optional[Union[str, dict]] = None,