diff --git a/src/backend/base/langflow/services/adapters/deployment/watsonx_orchestrate/service.py b/src/backend/base/langflow/services/adapters/deployment/watsonx_orchestrate/service.py index 6738fbd314..33a83e599d 100644 --- a/src/backend/base/langflow/services/adapters/deployment/watsonx_orchestrate/service.py +++ b/src/backend/base/langflow/services/adapters/deployment/watsonx_orchestrate/service.py @@ -106,6 +106,7 @@ from langflow.services.adapters.deployment.watsonx_orchestrate.payloads import ( WatsonxDeploymentCreatePayload, WatsonxDeploymentCreateResultData, WatsonxDeploymentLlmListResultData, + WatsonxModelOut, WatsonxDeploymentUpdatePayload, WatsonxDeploymentUpdateResultData, WatsonxModelOut, @@ -305,7 +306,6 @@ class WatsonxOrchestrateDeploymentService(BaseDeploymentService): WatsonxModelOut(model_name="groq/openai/gpt-oss-120b"), WatsonxModelOut(model_name="bedrock/openai.gpt-oss-120b-1:0"), ] - hardcoded_names = {m.model_name for m in raw_models} try: diff --git a/src/backend/tests/unit/services/deployment/test_watsonx_orchestrate.py b/src/backend/tests/unit/services/deployment/test_watsonx_orchestrate.py index 9405d26f03..e9d0bce78b 100644 --- a/src/backend/tests/unit/services/deployment/test_watsonx_orchestrate.py +++ b/src/backend/tests/unit/services/deployment/test_watsonx_orchestrate.py @@ -5052,6 +5052,8 @@ async def test_list_llms_returns_normalized_model_names(monkeypatch): assert result.provider_result == { "models": [ + {"model_name": "groq/openai/gpt-oss-120b"}, + {"model_name": "bedrock/openai.gpt-oss-120b-1:0"}, {"model_name": "granite-3.1-8b"}, {"model_name": "granite-3.3-8b"}, {"model_name": "granite-3.1-8b"},