From 9d57aa8f997ec10cd7e53ec2337ba25989926b60 Mon Sep 17 00:00:00 2001
From: Deon Sanchez <69873175+deon-sanchez@users.noreply.github.com>
Date: Thu, 11 Dec 2025 14:08:24 -0700
Subject: [PATCH] bug: Fix watsonX embedding model selection (#10980)

* fix watson x

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
---
 src/lfx/src/lfx/base/models/unified_models.py    | 2 +-
 src/lfx/src/lfx/base/models/watsonx_constants.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/lfx/src/lfx/base/models/unified_models.py b/src/lfx/src/lfx/base/models/unified_models.py
index ed0c3c0233..77a5175045 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 6a76435acb..1b92767c47 100644
--- a/src/lfx/src/lfx/base/models/watsonx_constants.py
+++ b/src/lfx/src/lfx/base/models/watsonx_constants.py
@@ -5,21 +5,25 @@ WATSONX_DEFAULT_EMBEDDING_MODELS = [
         provider="IBM Watsonx",
         name="sentence-transformers/all-minilm-l12-v2",
         icon="WatsonxAI",
+        model_type="embeddings",
     ),
     create_model_metadata(
         provider="IBM Watsonx",
         name="ibm/slate-125m-english-rtrvr-v2",
         icon="WatsonxAI",
+        model_type="embeddings",
     ),
     create_model_metadata(
         provider="IBM Watsonx",
         name="ibm/slate-30m-english-rtrvr-v2",
         icon="WatsonxAI",
+        model_type="embeddings",
     ),
     create_model_metadata(
         provider="IBM Watsonx",
         name="intfloat/multilingual-e5-large",
         icon="WatsonxAI",
+        model_type="embeddings",
     ),
 ]