From ce9b3ff588a3a1365e8636b9ba7eee39162c641f Mon Sep 17 00:00:00 2001 From: Esteban Zeller Date: Tue, 24 Feb 2026 10:44:52 -0300 Subject: [PATCH] Add tool calling support parameters on Gradient LLM Added basic gradient ai tool call logic --- litellm/llms/gradient_ai/chat/transformation.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/litellm/llms/gradient_ai/chat/transformation.py b/litellm/llms/gradient_ai/chat/transformation.py index d631affdef8..e115859c586 100644 --- a/litellm/llms/gradient_ai/chat/transformation.py +++ b/litellm/llms/gradient_ai/chat/transformation.py @@ -73,6 +73,11 @@ class GradientAIConfig(OpenAILikeChatConfig): "include_guardrails_info", "provide_citations", "retrieval_method", + # Tool calling support + "tools", + "tool_choice", + "parallel_tool_calls", + "response_format" ] return supported_params @@ -137,6 +142,9 @@ class GradientAIConfig(OpenAILikeChatConfig): for param, value in non_default_params.items(): if param in supported_openai_params: optional_params[param] = value + elif param in ["tools", "tool_choice", "parallel_tool_calls"]: + # Allow explicitly if the params definition changes + optional_params[param] = value elif not drop_params: from litellm.utils import UnsupportedParamsError raise UnsupportedParamsError(