From ae3cf86ef669ddcce11cbdfb12692814c25bac17 Mon Sep 17 00:00:00 2001 From: kamilbenkirane Date: Wed, 25 Mar 2026 18:54:02 +0100 Subject: [PATCH] fix: add ToolSupport declarations to legacy OpenAI and xAI models gpt-4-turbo, gpt-4, gpt-3.5-turbo, and grok-2-vision-1212 all support tool calling but were missing TextParameter.TOOLS in their parameter constraints. Added ToolSupport(tools=[]) so they properly advertise tool support in model metadata. Fixes #225 --- src/celeste/modalities/text/providers/openai/models.py | 3 +++ src/celeste/modalities/text/providers/xai/models.py | 1 + 2 files changed, 4 insertions(+) diff --git a/src/celeste/modalities/text/providers/openai/models.py b/src/celeste/modalities/text/providers/openai/models.py index 3e53845..2c352b3 100644 --- a/src/celeste/modalities/text/providers/openai/models.py +++ b/src/celeste/modalities/text/providers/openai/models.py @@ -51,6 +51,7 @@ Parameter.TEMPERATURE: Range(min=0.0, max=2.0), Parameter.MAX_TOKENS: Range(min=1, max=4096), TextParameter.OUTPUT_SCHEMA: Schema(), + TextParameter.TOOLS: ToolSupport(tools=[]), TextParameter.IMAGE: ImagesConstraint(), }, ), @@ -63,6 +64,7 @@ parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0), Parameter.MAX_TOKENS: Range(min=1, max=8192), + TextParameter.TOOLS: ToolSupport(tools=[]), }, ), Model( @@ -74,6 +76,7 @@ parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0), Parameter.MAX_TOKENS: Range(min=1, max=4096), + TextParameter.TOOLS: ToolSupport(tools=[]), }, ), Model( diff --git a/src/celeste/modalities/text/providers/xai/models.py b/src/celeste/modalities/text/providers/xai/models.py index 76f124e..5d26ecb 100644 --- a/src/celeste/modalities/text/providers/xai/models.py +++ b/src/celeste/modalities/text/providers/xai/models.py @@ -128,6 +128,7 @@ parameter_constraints={ Parameter.TEMPERATURE: Range(min=0.0, max=2.0), Parameter.MAX_TOKENS: Range(min=1, max=32768), + TextParameter.TOOLS: ToolSupport(tools=[]), TextParameter.IMAGE: ImagesConstraint(), }, ),