logging

app.py

@@ -88,9 +88,9 @@ CHATBOT_INSTRUCTIONS = (
 with open("config.yaml") as stream:
     try:
         demo_config = yaml.safe_load(stream)
-        logging.getLogger(__name__).info(f"Config: {demo_config}")
+        logging.info(f"Config: {demo_config}")
     except yaml.YAMLError as exc:
-        logging.getLogger(__name__).error(f"Error loading config: {exc}")
+        logging.error(f"Error loading config: {exc}")
         gr.Error("Error loading config: {exc}")
 
 try:
@@ -111,10 +111,10 @@ try:
             duration=-1
         )
 except Exception as exc:
-    logging.getLogger(__name__).error(f"Error processing config: {exc}")
+    logging.error(f"Error processing config: {exc}")
     gr.Error(f"Error processing config: {exc}")
 
-logging.getLogger(__name__).info(f"Reasoning guide expert model is {guide_kwargs['expert_model']}.")
+logging.info(f"Reasoning guide expert model is {guide_kwargs['expert_model']}.")
 
 
 
@@ -128,7 +128,7 @@ def setup_client_llm(**client_kwargs) -> LogitsModel | None:
     try:
         llm = create_logits_model(**client_kwargs)
     except Exception as e:
-        logging.getLogger(__name__).error(f"When setting up client llm: Error: {e}")
+        logging.error(f"When setting up client llm: Error: {e}")
         return False
     return llm
 
@@ -162,13 +162,14 @@ async def bot(
     if len(history_langchain_format) <= 1:
 
         # health check
+        gr.Info("Checking availability and health of inference endpoints ...", duration=6)
         health_check = await guide.health_check()
 
         if health_check.get("status", None) != "ok":
             health_msg = " | ".join([f"{k}: {v}" for k, v in health_check.items()])
-            logging.getLogger(__name__).error(f"Guide health check failed: {health_msg}")
+            logging.error(f"Guide health check failed: {health_msg}")
             gr.Error(f"LLM availability / health check failed: {health_msg}")
-        logging.getLogger(__name__).info(f"Health check: {health_check}")
+        logging.info(f"Health check: {health_check}")
 
         message = history[-1][0]
 
@@ -176,9 +177,9 @@ async def bot(
             artifacts = {}
             progress_step = 0
             async for otype, ovalue in guide.guide(message):
-                logging.getLogger(__name__).info(f"Guide output: {otype.value} - {ovalue}")
+                logging.info(f"Guide output: {otype.value} - {ovalue}")
                 if otype.value == "progress":
-                    logging.getLogger(__name__).info(f"Progress: {ovalue}")
+                    logging.info(f"Progress: {ovalue}")
                     gr.Info(ovalue, duration=12)
                     progress((progress_step,4))
                     progress_step += 1
@@ -186,7 +187,6 @@ async def bot(
                     artifacts[otype.value] = ovalue
                 else:
                     break
-                asyncio.sleep(0.1)
         except asyncio.TimeoutError:
             msg = "Guided reasoning process took too long. Please try again."
             raise gr.Error(msg)