fix linting

Author: ishaan-jaff

litellm/llms/OpenAI/o1_reasoning.py

@@ -65,7 +65,9 @@ def get_supported_openai_params(self, model: str) -> list:
             param for param in all_openai_params if param not in non_supported_params
         ]
 
-    def map_openai_params(self, non_default_params: dict, optional_params: dict):
+    def map_openai_params(
+        self, non_default_params: dict, optional_params: dict, model: str
+    ):
         for param, value in non_default_params.items():
             if param == "max_tokens":
                 optional_params["max_completion_tokens"] = value

litellm/llms/OpenAI/openai.py

@@ -573,6 +573,7 @@ def map_openai_params(
             return litellm.OpenAIO1Config().map_openai_params(
                 non_default_params=non_default_params,
                 optional_params=optional_params,
+                model=model,
             )
         supported_openai_params = self.get_supported_openai_params(model)
         for param, value in non_default_params.items():

litellm/tests/test_openai_o1.py

@@ -50,4 +50,3 @@ async def test_o1_handle_system_role(respx_mock: MockRouter):
 
     print(f"response: {response}")
     assert isinstance(response, ModelResponse)
-    assert response.choices[0].message.content == "Mocked response"