Skip to content

Commit df79784

Browse files
committed
Updated watsonx dependencies
1 parent 5c873d5 commit df79784

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)