Skip to content

Commit ccaf6aa

Browse files
authored
chore(python-deps): replace ibm_watson_machine_learning with ibm_watsonx_ai (#3302)
# What does this PR do? This PR updates the Watsonx provider dependencies from `ibm_watson_machine_learning` to `ibm_watsonx_ai`. The old package `ibm_watson_machine_learning` is in **deprecation mode** ([[PyPI link](https://pypi.org/project/ibm-watson-machine-learning/)](https://pypi.org/project/ibm-watson-machine-learning/)) and relies on older versions of dependencies such as `pandas`. Updating to `ibm_watsonx_ai` ensures compatibility with current dependency versions and ongoing support. ## Test Plan I verified the update by running an inference using a model provided by Watsonx. The model ran successfully, confirming that the new dependency works as expected. Co-authored-by: are-ces <[email protected]>
1 parent c59d8c5 commit ccaf6aa

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

llama_stack/providers/registry/inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ def available_providers() -> list[ProviderSpec]:
292292
api=Api.inference,
293293
adapter=AdapterSpec(
294294
adapter_type="watsonx",
295-
pip_packages=["ibm_watson_machine_learning"],
295+
pip_packages=["ibm_watsonx_ai"],
296296
module="llama_stack.providers.remote.inference.watsonx",
297297
config_class="llama_stack.providers.remote.inference.watsonx.WatsonXConfig",
298298
provider_data_validator="llama_stack.providers.remote.inference.watsonx.WatsonXProviderDataValidator",

llama_stack/providers/remote/inference/watsonx/watsonx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
from collections.abc import AsyncGenerator, AsyncIterator
88
from typing import Any
99

10-
from ibm_watson_machine_learning.foundation_models import Model
11-
from ibm_watson_machine_learning.metanames import GenTextParamsMetaNames as GenParams
10+
from ibm_watsonx_ai.foundation_models import Model
11+
from ibm_watsonx_ai.metanames import GenTextParamsMetaNames as GenParams
1212
from openai import AsyncOpenAI
1313

1414
from llama_stack.apis.common.content_types import InterleavedContent, InterleavedContentItem

0 commit comments

Comments
 (0)