Don't init arms without license key (#298)

aigc-apps · Dec 6, 2024 · cb1645b · cb1645b
1 parent 51dc01e
commit cb1645b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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