phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 11:43:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix fireworks ai config

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-26 06:40:44 -07:00
parent 493a737787
commit dcdf266f36
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/fireworks_ai.py
View file

@ -92,16 +92,15 @@ class FireworksAIConfig:
non_default_params: dict, non_default_params: dict,
optional_params: dict, optional_params: dict,
model: str, model: str,
drop_params: bool,
) -> dict: ) -> dict:
supported_openai_params = self.get_supported_openai_params() supported_openai_params = self.get_supported_openai_params()
for param, value in non_default_params.items(): for param, value in non_default_params.items():
if param == "tool_choice": if param == "tool_choice":
if value == "required": if value == "required":
# relevant issue: https://github.com/BerriAI/litellm/issues/4416 # relevant issue: https://github.com/BerriAI/litellm/issues/4416
optional_params["tools"] = "any" optional_params["tool_choice"] = "any"
if param in supported_openai_params: elif param in supported_openai_params:
if value is not None: if value is not None:
optional_params[param] = value optional_params[param] = value
return optional_params return optional_params