From 11146b6baeba1fd1566704c4e92af02127adee3b Mon Sep 17 00:00:00 2001
From: heyszt <270985384@qq.com>
Date: Mon, 19 May 2025 10:09:21 +0800
Subject: [PATCH] fix create_tracing_app_config error (#19884)

---
 api/core/ops/ops_trace_manager.py | 4 +++-
 api/services/ops_service.py       | 8 ++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/api/core/ops/ops_trace_manager.py b/api/core/ops/ops_trace_manager.py
index 2bcca6ccea..8ddeb1b846 100644
--- a/api/core/ops/ops_trace_manager.py
+++ b/api/core/ops/ops_trace_manager.py
@@ -287,7 +287,9 @@ class OpsTraceManager:
         :return:
         """
         # auth check
-        if tracing_provider not in provider_config_map and tracing_provider is not None:
+        try:
+            provider_config_map[tracing_provider]
+        except KeyError:
             raise ValueError(f"Invalid tracing provider: {tracing_provider}")
 
         app_config: Optional[App] = db.session.query(App).filter(App.id == app_id).first()
diff --git a/api/services/ops_service.py b/api/services/ops_service.py
index 6b317212d1..a9c2b28476 100644
--- a/api/services/ops_service.py
+++ b/api/services/ops_service.py
@@ -87,7 +87,9 @@ class OpsService:
         :param tracing_config: tracing config
         :return:
         """
-        if tracing_provider not in provider_config_map and tracing_provider:
+        try:
+            provider_config_map[tracing_provider]
+        except KeyError:
             return {"error": f"Invalid tracing provider: {tracing_provider}"}
 
         config_class, other_keys = (
@@ -150,7 +152,9 @@ class OpsService:
         :param tracing_config: tracing config
         :return:
         """
-        if tracing_provider not in provider_config_map:
+        try:
+            provider_config_map[tracing_provider]
+        except KeyError:
             raise ValueError(f"Invalid tracing provider: {tracing_provider}")
 
         # check if trace config already exists