From 8a7f2921f21dbb6679a20bc603b5372c10e462e7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 26 Jun 2024 06:57:04 -0700 Subject: [PATCH] fix + test fireworks ai param mapping for tools --- litellm/llms/fireworks_ai.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/llms/fireworks_ai.py b/litellm/llms/fireworks_ai.py index 7c2d3b72ad1..e9caf887ad4 100644 --- a/litellm/llms/fireworks_ai.py +++ b/litellm/llms/fireworks_ai.py @@ -99,7 +99,9 @@ class FireworksAIConfig: if value == "required": # relevant issue: https://github.com/BerriAI/litellm/issues/4416 optional_params["tool_choice"] = "any" - + else: + # pass through the value of tool choice + optional_params["tool_choice"] = value elif param in supported_openai_params: if value is not None: optional_params[param] = value