From 89791d9285fb92813db5644ababfcf09cdd0dcd3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 20 Aug 2024 08:13:56 -0700
Subject: [PATCH] fix(main.py): response_format typing for acompletion

Fixes https://github.com/BerriAI/litellm/issues/5239
---
 litellm/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/main.py b/litellm/main.py
index 24ae12631d2e..96496c7177e4 100644
--- 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,