FIX litellm.drop_params=true for gemini

2025-05-05 13:31:15 +02:00 · 2025-05-05 13:31:15 +02:00 · 99e4cc1604
parent 696c79a049
commit 99e4cc1604
1 changed files with 1 additions and 1 deletions
--- 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__}"