fix default trace handler (#209)

aigc-apps · Sep 10, 2024 · 365f8a6 · 365f8a6
1 parent 2e9c796
commit 365f8a6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/pai_rag/core/rag_trace.py b/src/pai_rag/core/rag_trace.py
@@ -24,6 +24,8 @@ def init_trace(trace_config):
             service_owner_sub_id="",
         )
         logger.info(f"Pai-LLM-Trace enabled with endpoint: '{trace_config.endpoint}'.")
-    else:
+    elif trace_config is not None and trace_config.type == "arize_phoenix":
         set_global_handler("arize_phoenix")
         logger.info("Arize trace enabled.")
+    else:
+        logger.warning("No trace used.")