Skip to content

Commit 3838b9c

Browse files
authored
refactor(watsonx): Drop WATSONX_SERVICE_URL getenv. (#469)
1 parent 5a3cd8e commit 3838b9c

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

src/any_llm/providers/watsonx/watsonx.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from __future__ import annotations
22

3-
import os
43
from typing import TYPE_CHECKING, Any
54

65
from pydantic import BaseModel
@@ -116,10 +115,7 @@ async def _acompletion(
116115

117116
model_inference = ModelInference(
118117
model_id=params.model_id,
119-
credentials=Credentials(
120-
api_key=self.config.api_key,
121-
url=self.config.api_base or os.getenv("WATSONX_SERVICE_URL"),
122-
),
118+
credentials=Credentials(api_key=self.config.api_key, url=self.config.api_base),
123119
**(self.config.client_args if self.config.client_args else {}),
124120
)
125121

@@ -148,10 +144,8 @@ async def _alist_models(self, **kwargs: Any) -> Sequence[Model]:
148144
Fetch available models from the /v1/models endpoint.
149145
"""
150146
client = WatsonxClient(
151-
url=self.config.api_base or os.getenv("WATSONX_SERVICE_URL"),
152-
credentials=Credentials(
153-
api_key=self.config.api_key, url=self.config.api_base or os.getenv("WATSONX_SERVICE_URL")
154-
),
147+
url=self.config.api_base,
148+
credentials=Credentials(api_key=self.config.api_key, url=self.config.api_base),
155149
**(self.config.client_args if self.config.client_args else {}),
156150
)
157151
models_response = client.foundation_models.get_model_specs(**kwargs)

0 commit comments

Comments
 (0)