feat: Add default thinking parameter handling for specific Gemini models

src/rotator_library/client.py

@@ -27,6 +27,7 @@ class RotatingClient:
     def __init__(self, api_keys: Dict[str, List[str]], max_retries: int = 2, usage_file_path: str = "key_usage.json"):
         os.environ["LITELLM_LOG"] = "ERROR"
         litellm.set_verbose = False
+        litellm.drop_params = True
         if not api_keys:
             raise ValueError("API keys dictionary cannot be empty.")
         self.api_keys = api_keys
@@ -139,6 +140,10 @@ class RotatingClient:
                     else:
                         # If conversion returns None, remove it to avoid sending empty settings
                         del litellm_kwargs["safety_settings"]
+                
+                if provider == "gemini":
+                    provider_instance = self._provider_instances[provider]
+                    provider_instance.handle_thinking_parameter(litellm_kwargs, model)
 
                 if "gemma-3" in model and "messages" in litellm_kwargs:
                     new_messages = [

src/rotator_library/providers/gemini_provider.py

@@ -50,3 +50,10 @@ class GeminiProvider(ProviderInterface):
                 })
         
         return gemini_settings
+
+    def handle_thinking_parameter(self, payload: Dict[str, Any], model: str):
+        """
+        Adds a default thinking parameter for specific Gemini models if not already present (To show reasoning output).
+        """
+        if model in ["gemini/gemini-2.5-pro", "gemini/gemini-2.5-flash"] and "thinking" not in payload:
+            payload["thinking"] = {"type": "enabled", "budget_tokens": -1}