fix(main.py): response_format typing for acompletion

Fixes #5239
BerriAI · Aug 20, 2024 · 89791d9 · 89791d9
1 parent 8b23a23
commit 89791d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/litellm/main.py b/litellm/main.py
@@ -253,7 +253,7 @@ async def acompletion(
     logit_bias: Optional[dict] = None,
     user: Optional[str] = None,
     # openai v1.0+ new params
-    response_format: Optional[dict] = None,
+    response_format: Optional[Union[dict, Type[BaseModel]]] = None,
     seed: Optional[int] = None,
     tools: Optional[List] = None,
     tool_choice: Optional[str] = None,