From 61cd8540de7358c616b64b1339fa32602a9b0bac Mon Sep 17 00:00:00 2001 From: Sumit Jaiswal Date: Wed, 4 Jun 2025 20:47:58 +0530 Subject: [PATCH] fix review comment against health check env var naming --- llama_stack/providers/remote/inference/watsonx/config.py | 2 +- llama_stack/providers/remote/inference/watsonx/watsonx.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/inference/watsonx/config.py b/llama_stack/providers/remote/inference/watsonx/config.py index 8486f22e5..0bffae38f 100644 --- a/llama_stack/providers/remote/inference/watsonx/config.py +++ b/llama_stack/providers/remote/inference/watsonx/config.py @@ -33,7 +33,7 @@ class WatsonXConfig(BaseModel): description="The Project ID key, only needed of using the hosted service", ) model_id: str | None = Field( - default_factory=lambda: os.getenv("WATSONX_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", ) timeout: int = Field( diff --git a/llama_stack/providers/remote/inference/watsonx/watsonx.py b/llama_stack/providers/remote/inference/watsonx/watsonx.py index d13ee58fe..7567fb814 100644 --- a/llama_stack/providers/remote/inference/watsonx/watsonx.py +++ b/llama_stack/providers/remote/inference/watsonx/watsonx.py @@ -85,6 +85,8 @@ class WatsonXInferenceAdapter(Inference, ModelRegistryHelper): Performs a health check by verifying connectivity to the Watsonx server. This method is used by the Provider API to verify that the service is running correctly. + Specify the model ID as WATSONX_HEALTH_CHECK_MODEL_ID under env. var. + to use for the health check, which defaults to "ibm/granite-3-8b-instruct". Returns: HealthResponse: A dictionary containing the health status. """