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__}"