From cb1645beee0d038ef6451bae3ad15726b48ae18f Mon Sep 17 00:00:00 2001
From: Yue Fei <luxun.fy@alibaba-inc.com>
Date: Fri, 6 Dec 2024 09:23:37 +0800
Subject: [PATCH] Don't init arms without license key (#298)

---
 src/pai_rag/app/web/webui.py | 2 +-
 src/pai_rag/main.py          | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/pai_rag/app/web/webui.py b/src/pai_rag/app/web/webui.py
index 3c91f34c..33150ac4 100644
--- a/src/pai_rag/app/web/webui.py
+++ b/src/pai_rag/app/web/webui.py
@@ -90,7 +90,7 @@ def make_homepage():
         with gr.Tab("\N{rocket} Agent"):
             agent_elements = create_agent_tab()
             elem_manager.add_elems(agent_elements)
-        with gr.Tab("\N{bar chart} Data Analysis"):
+        with gr.Tab("\N{bar chart} DBChat"):
             analysis_elements = create_data_analysis_tab()
             elem_manager.add_elems(analysis_elements)
 
diff --git a/src/pai_rag/main.py b/src/pai_rag/main.py
index b7eecea1..4d1fc0bf 100644
--- a/src/pai_rag/main.py
+++ b/src/pai_rag/main.py
@@ -1,7 +1,8 @@
 # init trace
-from aliyun.opentelemetry.instrumentation.auto_instrumentation import sitecustomize
-
 import os
+if os.getenv("ARMS_LICENSE_KEY"):
+    from aliyun.opentelemetry.instrumentation.auto_instrumentation import sitecustomize
+
 import click
 import uvicorn
 from fastapi import FastAPI