If initialization fails for library client, error the test

This commit is contained in:
Ashwin Bharambe 2025-01-22 18:11:42 -08:00
parent 72a1b27d01
commit f4b0f2af8b

View file

@ -64,7 +64,9 @@ def llama_stack_client(provider_data):
provider_data=provider_data,
skip_logger_removal=True,
)
client.initialize()
if not client.initialize():
raise RuntimeError("Initialization failed")
elif os.environ.get("LLAMA_STACK_BASE_URL"):
client = LlamaStackClient(
base_url=get_env_or_fail("LLAMA_STACK_BASE_URL"),