Updated watsonx dependencies

are-ces · are-ces · commit 82b3b1281fc8 · 2025-09-01T10:32:35.000+02:00
diff --git 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
diff --git a/pyproject.toml b/pyproject.toml
@@ -50,6 +50,7 @@ dependencies = [
     "opentelemetry-exporter-otlp-proto-http>=1.30.0", # server
     "aiosqlite>=0.21.0",                      # server - for metadata store
     "asyncpg",                                # for metadata store
+    "ibm_watsonx_ai",
 ]
 
 [project.optional-dependencies]
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -50,6 +50,7 @@ dependencies = [`
`50`	`50`	`"opentelemetry-exporter-otlp-proto-http>=1.30.0", # server`
`51`	`51`	`"aiosqlite>=0.21.0", # server - for metadata store`
`52`	`52`	`"asyncpg", # for metadata store`
	`53`	`+ "ibm_watsonx_ai",`
`53`	`54`	`]`
`54`	`55`
`55`	`56`	`[project.optional-dependencies]`