From 36eea7ee9db1ce6119dfd85364f870a219ea2f7c Mon Sep 17 00:00:00 2001 From: Himavarsha <40851462+HimavarshaVS@users.noreply.github.com> Date: Mon, 5 Jan 2026 13:27:09 -0500 Subject: [PATCH] fix: enable tool calling watson X (#11199) * enable tool calling watsonX * build component index * [autofix.ci] apply automated fixes * [autofix.ci] apply automated fixes --------- Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Eric Hare --- .../Image Sentiment Analysis.json | 12 ++++++------ .../Research Translation Loop.json | 6 +++--- .../starter_projects/Youtube Analysis.json | 12 ++++++------ src/frontend/src/constants/providerConstants.ts | 2 +- src/lfx/src/lfx/base/models/unified_models.py | 2 +- .../src/lfx/base/models/watsonx_constants.py | 17 ++++++++++------- 6 files changed, 27 insertions(+), 24 deletions(-) diff --git a/src/backend/base/langflow/initial_setup/starter_projects/Image Sentiment Analysis.json b/src/backend/base/langflow/initial_setup/starter_projects/Image Sentiment Analysis.json index a168e3c68c..a5c43c727b 100644 --- a/src/backend/base/langflow/initial_setup/starter_projects/Image Sentiment Analysis.json +++ b/src/backend/base/langflow/initial_setup/starter_projects/Image Sentiment Analysis.json @@ -1320,14 +1320,14 @@ "provider": "Google Generative AI" }, { - "category": "IBM Watsonx", + "category": "IBM WatsonX", "icon": "WatsonxAI", "metadata": { "is_disabled_provider": true, "variable_name": "WATSONX_APIKEY" }, - "name": "__enable_provider_IBM Watsonx__", - "provider": "IBM Watsonx" + "name": "__enable_provider_IBM WatsonX__", + "provider": "IBM WatsonX" } ], "override_skip": false, @@ -1840,14 +1840,14 @@ "provider": "Google Generative AI" }, { - "category": "IBM Watsonx", + "category": "IBM WatsonX", "icon": "WatsonxAI", "metadata": { "is_disabled_provider": true, "variable_name": "WATSONX_APIKEY" }, - "name": "__enable_provider_IBM Watsonx__", - "provider": "IBM Watsonx" + "name": "__enable_provider_IBM WatsonX__", + "provider": "IBM WatsonX" } ], "override_skip": false, diff --git a/src/backend/base/langflow/initial_setup/starter_projects/Research Translation Loop.json b/src/backend/base/langflow/initial_setup/starter_projects/Research Translation Loop.json index 76bd95c772..c1d0831c98 100644 --- a/src/backend/base/langflow/initial_setup/starter_projects/Research Translation Loop.json +++ b/src/backend/base/langflow/initial_setup/starter_projects/Research Translation Loop.json @@ -1254,14 +1254,14 @@ "provider": "Google Generative AI" }, { - "category": "IBM Watsonx", + "category": "IBM WatsonX", "icon": "WatsonxAI", "metadata": { "is_disabled_provider": true, "variable_name": "WATSONX_APIKEY" }, - "name": "__enable_provider_IBM Watsonx__", - "provider": "IBM Watsonx" + "name": "__enable_provider_IBM WatsonX__", + "provider": "IBM WatsonX" } ], "override_skip": false, diff --git a/src/backend/base/langflow/initial_setup/starter_projects/Youtube Analysis.json b/src/backend/base/langflow/initial_setup/starter_projects/Youtube Analysis.json index ce61fd5d95..f5cf5ca27c 100644 --- a/src/backend/base/langflow/initial_setup/starter_projects/Youtube Analysis.json +++ b/src/backend/base/langflow/initial_setup/starter_projects/Youtube Analysis.json @@ -871,14 +871,14 @@ "provider": "Ollama" }, { - "category": "IBM Watsonx", + "category": "IBM WatsonX", "icon": "WatsonxAI", "metadata": { "is_disabled_provider": true, "variable_name": "WATSONX_APIKEY" }, - "name": "__enable_provider_IBM Watsonx__", - "provider": "IBM Watsonx" + "name": "__enable_provider_IBM WatsonX__", + "provider": "IBM WatsonX" } ], "override_skip": false, @@ -2637,14 +2637,14 @@ "provider": "Ollama" }, { - "category": "IBM Watsonx", + "category": "IBM WatsonX", "icon": "WatsonxAI", "metadata": { "is_disabled_provider": true, "variable_name": "WATSONX_APIKEY" }, - "name": "__enable_provider_IBM Watsonx__", - "provider": "IBM Watsonx" + "name": "__enable_provider_IBM WatsonX__", + "provider": "IBM WatsonX" } ], "override_skip": false, diff --git a/src/frontend/src/constants/providerConstants.ts b/src/frontend/src/constants/providerConstants.ts index a72dea8233..31413d73b4 100644 --- a/src/frontend/src/constants/providerConstants.ts +++ b/src/frontend/src/constants/providerConstants.ts @@ -4,7 +4,7 @@ export const PROVIDER_VARIABLE_MAPPING: Record = { "Google Generative AI": "GOOGLE_API_KEY", Google: "GOOGLE_API_KEY", Ollama: "OLLAMA_BASE_URL", - "IBM Watsonx": "WATSONX_APIKEY", + "IBM WatsonX": "WATSONX_APIKEY", Cohere: "COHERE_API_KEY", HuggingFace: "HUGGINGFACEHUB_API_TOKEN", Groq: "GROQ_API_KEY", diff --git a/src/lfx/src/lfx/base/models/unified_models.py b/src/lfx/src/lfx/base/models/unified_models.py index a8814ba483..4e13bd4c25 100644 --- a/src/lfx/src/lfx/base/models/unified_models.py +++ b/src/lfx/src/lfx/base/models/unified_models.py @@ -75,7 +75,7 @@ def get_model_provider_metadata(): "icon": "Ollama", "variable_name": "OLLAMA_BASE_URL", # Ollama is local but can have custom URL }, - "IBM Watsonx": { + "IBM WatsonX": { "icon": "WatsonxAI", "variable_name": "WATSONX_APIKEY", }, diff --git a/src/lfx/src/lfx/base/models/watsonx_constants.py b/src/lfx/src/lfx/base/models/watsonx_constants.py index d84d5f3621..aef3590d76 100644 --- a/src/lfx/src/lfx/base/models/watsonx_constants.py +++ b/src/lfx/src/lfx/base/models/watsonx_constants.py @@ -2,52 +2,55 @@ from .model_metadata import create_model_metadata WATSONX_DEFAULT_LLM_MODELS = [ create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="ibm/granite-3-2b-instruct", icon="WatsonxAI", model_type="llm", + tool_calling=False, default=True, ), create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="ibm/granite-3-8b-instruct", icon="WatsonxAI", model_type="llm", + tool_calling=True, default=True, ), create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="ibm/granite-13b-instruct-v2", icon="WatsonxAI", model_type="llm", + tool_calling=False, default=True, ), ] WATSONX_DEFAULT_EMBEDDING_MODELS = [ create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="sentence-transformers/all-minilm-l12-v2", icon="WatsonxAI", model_type="embeddings", default=True, ), create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="ibm/slate-125m-english-rtrvr-v2", icon="WatsonxAI", model_type="embeddings", default=True, ), create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="ibm/slate-30m-english-rtrvr-v2", icon="WatsonxAI", model_type="embeddings", default=True, ), create_model_metadata( - provider="IBM Watsonx", + provider="IBM WatsonX", name="intfloat/multilingual-e5-large", icon="WatsonxAI", model_type="embeddings",