From 365f8a6755dfb91b5aaa85833a423a7779bd5485 Mon Sep 17 00:00:00 2001 From: Yue Fei Date: Tue, 10 Sep 2024 15:07:39 +0800 Subject: [PATCH] fix default trace handler (#209) --- src/pai_rag/core/rag_trace.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/pai_rag/core/rag_trace.py b/src/pai_rag/core/rag_trace.py index fa881de9..0d551999 100644 --- 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.")