forked from phoenix-oss/llama-stack-mirror
If initialization fails for library client, error the test
This commit is contained in:
parent
72a1b27d01
commit
f4b0f2af8b
1 changed files with 3 additions and 1 deletions
|
@ -64,7 +64,9 @@ def llama_stack_client(provider_data):
|
||||||
provider_data=provider_data,
|
provider_data=provider_data,
|
||||||
skip_logger_removal=True,
|
skip_logger_removal=True,
|
||||||
)
|
)
|
||||||
client.initialize()
|
if not client.initialize():
|
||||||
|
raise RuntimeError("Initialization failed")
|
||||||
|
|
||||||
elif os.environ.get("LLAMA_STACK_BASE_URL"):
|
elif os.environ.get("LLAMA_STACK_BASE_URL"):
|
||||||
client = LlamaStackClient(
|
client = LlamaStackClient(
|
||||||
base_url=get_env_or_fail("LLAMA_STACK_BASE_URL"),
|
base_url=get_env_or_fail("LLAMA_STACK_BASE_URL"),
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue