phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 10:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix var name

Browse source
This commit is contained in:
Sumit Jaiswal 2025-06-04 20:49:13 +05:30
parent 61cd8540de
commit 04825e854c
No known key found for this signature in database
GPG key ID: A4604B39D64D6AEC
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/watsonx/config.py
View file

@ -32,7 +32,7 @@ class WatsonXConfig(BaseModel):
default_factory=lambda: os.getenv("WATSONX_PROJECT_ID"), default_factory=lambda: os.getenv("WATSONX_PROJECT_ID"),
description="The Project ID key, only needed of using the hosted service", description="The Project ID key, only needed of using the hosted service",
) )
model_id: str | None = Field( health_check_model_id: str | None = Field(
default_factory=lambda: os.getenv("WATSONX_HEALTH_CHECK_MODEL_ID", "ibm/granite-3-8b-instruct"), default_factory=lambda: os.getenv("WATSONX_HEALTH_CHECK_MODEL_ID", "ibm/granite-3-8b-instruct"),
description="The Model ID key, only needed of using the hosted service", description="The Model ID key, only needed of using the hosted service",
) )

2
llama_stack/providers/remote/inference/watsonx/watsonx.py
View file

@ -91,7 +91,7 @@ class WatsonXInferenceAdapter(Inference, ModelRegistryHelper):
HealthResponse: A dictionary containing the health status. HealthResponse: A dictionary containing the health status.
""" """
try: try:
client_model = self._get_client(self._config.model_id) client_model = self._get_client(self._config.health_check_model_id)
client_model.get_details() client_model.get_details()
return HealthResponse(status=HealthStatus.OK) return HealthResponse(status=HealthStatus.OK)
except Exception as ex: except Exception as ex: