From c0f0eb43e3c8a58b6d1e82e89baabeefa45a4493 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Wed, 3 Dec 2025 18:49:02 +0800 Subject: [PATCH] feat: update SQL queries to support edition-based logic in application stats --- apps/application/serializers/application_stats.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/apps/application/serializers/application_stats.py b/apps/application/serializers/application_stats.py index 8d45a5311..e1cb5e190 100644 --- a/apps/application/serializers/application_stats.py +++ b/apps/application/serializers/application_stats.py @@ -21,6 +21,7 @@ from common.db.search import native_search, get_dynamics_model from common.exception.app_exception import AppApiException from common.utils.common import get_file_content from maxkb.conf import PROJECT_DIR +from maxkb.settings import edition class ApplicationStatsSerializer(serializers.Serializer): @@ -133,7 +134,9 @@ class ApplicationStatisticsSerializer(serializers.Serializer): 'application_chat_record.create_time__lte': end_time} )}, select_string=get_file_content( - os.path.join(PROJECT_DIR, "apps", "application", 'sql', 'get_token_usage.sql'))) + os.path.join(PROJECT_DIR, "apps", "application", 'sql', + ('get_token_usage_ee.sql' if ['PE', 'EE'].__contains__( + edition) else 'get_token_usage.sql')))) return get_token_usage def get_top_questions_statistics(self, with_valid=True): @@ -150,5 +153,6 @@ class ApplicationStatisticsSerializer(serializers.Serializer): 'application_chat_record.create_time__lte': end_time} )}, select_string=get_file_content( - os.path.join(PROJECT_DIR, "apps", "application", 'sql', 'top_questions.sql'))) - return get_top_questions \ No newline at end of file + os.path.join(PROJECT_DIR, "apps", "application", 'sql', ( + 'top_questions_ee.sql' if ['PE', 'EE'].__contains__(edition) else 'top_questions.sql')))) + return get_top_questions