From 99e4cc16045862deccd37be3b95b9ff47f954e17 Mon Sep 17 00:00:00 2001
From: lidia9 <lidia@aliasrobotics.com>
Date: Mon, 5 May 2025 13:31:15 +0200
Subject: [PATCH] FIX litellm.drop_params=true for gemini

---
 src/cai/sdk/agents/models/openai_chatcompletions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cai/sdk/agents/models/openai_chatcompletions.py b/src/cai/sdk/agents/models/openai_chatcompletions.py
index 8ef8df3a..9b4579d8 100644
--- a/src/cai/sdk/agents/models/openai_chatcompletions.py
+++ b/src/cai/sdk/agents/models/openai_chatcompletions.py
@@ -93,7 +93,7 @@ if TYPE_CHECKING:
 # Suppress debug info from litellm
 litellm.suppress_debug_info = True
 
-if os.getenv('CAI_MODEL') == "o3-mini": 
+if os.getenv('CAI_MODEL') == "o3-mini" or os.getenv('CAI_MODEL') == "gemini-1.5-pro": 
     litellm.drop_params = True
 
 _USER_AGENT = f"Agents/Python {__version__}"