From 0548926e4d1b85c7d9b960d995ea9513662e8aaa Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Jun 2024 06:40:44 -0700 Subject: [PATCH] fix fireworks ai config --- litellm/llms/fireworks_ai.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/litellm/llms/fireworks_ai.py b/litellm/llms/fireworks_ai.py index 18309f4c2e0..7c2d3b72ad1 100644 --- a/litellm/llms/fireworks_ai.py +++ b/litellm/llms/fireworks_ai.py @@ -92,16 +92,15 @@ class FireworksAIConfig: non_default_params: dict, optional_params: dict, model: str, - drop_params: bool, ) -> dict: supported_openai_params = self.get_supported_openai_params() for param, value in non_default_params.items(): if param == "tool_choice": if value == "required": # 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: optional_params[param] = value return optional_params