Reapply logging configuration after AsyncLlamaStackAsLibraryClient

During initialization of AsyncLlamaStackAsLibraryClient, a logging configartion
is generated and applied. We want the lightspeed-stack logging config to always
be used.

Author: samdoran

src/client.py

@@ -18,7 +18,7 @@
     enrich_byok_rag,
     enrich_solr,
 )
-from log import get_logger
+from log import get_logger, setup_logging
 from models.api.responses.error import ServiceUnavailableResponse
 from models.config import LlamaStackConfiguration
 from utils.types import Singleton
@@ -66,6 +66,11 @@ async def _load_library_client(self, config: LlamaStackConfiguration) -> None:
         await client.initialize()
         self._lsc = client
 
+        # Re-apply logging configuration after ogx's setup_logging() is called.
+        # This ensures the desired logging configuration is applied when
+        # using AsyncLlamaStackAsLibraryClient.
+        setup_logging()
+
     def _load_service_client(self, config: LlamaStackConfiguration) -> None:
         """Initialize client in service mode (remote HTTP)."""
         logger.info("Using Llama stack running as a service")
@@ -151,6 +156,11 @@ async def reload_library_client(self) -> AsyncLlamaStackClient:
             )
             raise HTTPException(**error_response.model_dump()) from e
         self._lsc = client
+        # Re-apply logging configuration after ogx's setup_logging() is called.
+        # This ensures the desired logging configuration is applied when
+        # using AsyncLlamaStackAsLibraryClient.
+        setup_logging()
+
         return client
 
     async def check_model_available(self, model_id: str) -> tuple[bool, str]:
@@ -247,6 +257,11 @@ async def update_azure_token(self) -> AsyncLlamaStackClient:
             )
             await client.initialize()
             self._lsc = client
+            # Re-apply logging configuration after ogx's setup_logging() is called.
+            # This ensures the desired logging configuration is applied when
+            # using AsyncLlamaStackAsLibraryClient.
+            setup_logging()
+
             return client
 
         # Service client mode