From f4b0f2af8bf91fcc109d17d55c235296c71aefd1 Mon Sep 17 00:00:00 2001
From: Ashwin Bharambe <ashwin.bharambe@gmail.com>
Date: Wed, 22 Jan 2025 18:11:42 -0800
Subject: [PATCH] If initialization fails for library client, error the test

---
 tests/client-sdk/conftest.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/client-sdk/conftest.py b/tests/client-sdk/conftest.py
index 0f0733010..779c10e21 100644
--- a/tests/client-sdk/conftest.py
+++ b/tests/client-sdk/conftest.py
@@ -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"),