Skip to content

Commit a2b1c2c

Browse files
fix: update health checks to use new ModelFacade client API (#470)
The ModelFacade constructor was refactored in #373 to accept a ModelClient instead of a SecretResolver. The health_checks.py script was not updated, causing a TypeError (exit code 2) on every run since March 9.
1 parent 0a2d372 commit a2b1c2c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

scripts/health_checks.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
PREDEFINED_PROVIDERS_MODEL_MAP,
3232
)
3333
from data_designer.engine.model_provider import ModelProviderRegistry
34+
from data_designer.engine.models.clients.factory import create_model_client
3435
from data_designer.engine.models.facade import ModelFacade
3536
from data_designer.engine.secret_resolver import EnvironmentResolver
3637

@@ -67,7 +68,8 @@ def _check_model(provider_name: str, model_type: str) -> None:
6768
provider=provider_name,
6869
)
6970

70-
facade = ModelFacade(model_config, secret_resolver, provider_registry)
71+
client = create_model_client(model_config, secret_resolver, provider_registry)
72+
facade = ModelFacade(model_config, provider_registry, client=client)
7173

7274
if model_type == "embedding":
7375
result = facade.generate_text_embeddings(

0 commit comments

Comments
 (0)