From df79784e72409017ef1e2caeaa86b81c37db578c Mon Sep 17 00:00:00 2001 From: are-ces Date: Mon, 1 Sep 2025 10:30:56 +0200 Subject: [PATCH] Updated watsonx dependencies --- llama_stack/providers/registry/inference.py | 2 +- llama_stack/providers/remote/inference/watsonx/watsonx.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/registry/inference.py b/llama_stack/providers/registry/inference.py index 6264de7c7..fb841afdf 100644 --- a/llama_stack/providers/registry/inference.py +++ b/llama_stack/providers/registry/inference.py @@ -292,7 +292,7 @@ Available Models: api=Api.inference, adapter=AdapterSpec( adapter_type="watsonx", - pip_packages=["ibm_watson_machine_learning"], + pip_packages=["ibm_watsonx_ai"], module="llama_stack.providers.remote.inference.watsonx", config_class="llama_stack.providers.remote.inference.watsonx.WatsonXConfig", provider_data_validator="llama_stack.providers.remote.inference.watsonx.WatsonXProviderDataValidator", diff --git a/llama_stack/providers/remote/inference/watsonx/watsonx.py b/llama_stack/providers/remote/inference/watsonx/watsonx.py index 78161d1cb..cb7fc175f 100644 --- a/llama_stack/providers/remote/inference/watsonx/watsonx.py +++ b/llama_stack/providers/remote/inference/watsonx/watsonx.py @@ -7,8 +7,8 @@ from collections.abc import AsyncGenerator, AsyncIterator from typing import Any -from ibm_watson_machine_learning.foundation_models import Model -from ibm_watson_machine_learning.metanames import GenTextParamsMetaNames as GenParams +from ibm_watsonx_ai.foundation_models import Model +from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams from openai import AsyncOpenAI from llama_stack.apis.common.content_types import InterleavedContent, InterleavedContentItem