phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix linting

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-12 14:15:18 -07:00
parent a5a0773b19
commit 14dc7b3b54
3 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/OpenAI/o1_reasoning.py
View file

@ -65,7 +65,9 @@ class OpenAIO1Config(OpenAIConfig):
param for param in all_openai_params if param not in non_supported_params 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(): for param, value in non_default_params.items():
if param == "max_tokens": if param == "max_tokens":
optional_params["max_completion_tokens"] = value optional_params["max_completion_tokens"] = value

1
litellm/llms/OpenAI/openai.py
View file

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

1
litellm/tests/test_openai_o1.py
View file

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