fix(main.py): response_format typing for acompletion

Fixes https://github.com/BerriAI/litellm/issues/5239
2024-08-20 08:13:56 -07:00 · 2024-08-20 08:13:56 -07:00 · 89791d9285
commit 89791d9285
parent 8b23a2334f
1 changed files with 1 additions and 1 deletions
--- 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,