From 123ee24f7e6c2933e7dd6a70d5ea1963eac0cd72 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Sat, 25 May 2024 18:01:16 +0800 Subject: [PATCH] fix: stat perf --- model/provider/openai_official.py | 1 + util/general_utils.py | 6 ++++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/model/provider/openai_official.py b/model/provider/openai_official.py index 39bd7d3ac..d270cc201 100644 --- a/model/provider/openai_official.py +++ b/model/provider/openai_official.py @@ -73,6 +73,7 @@ class ProviderOpenAIOfficial(Provider): base_url=self.base_url ) self.model_configs: Dict = cfg['chatGPTConfigs'] + super().set_curr_model(self.model_configs['model']) self.image_generator_model_configs: Dict = self.cc.get('openai_image_generate', None) self.session_memory: Dict[str, List] = {} # 会话记忆 self.session_memory_lock = threading.Lock() diff --git a/util/general_utils.py b/util/general_utils.py index e97f00e66..931b4314d 100644 --- a/util/general_utils.py +++ b/util/general_utils.py @@ -470,6 +470,7 @@ def upload(_global_object: GlobalObject): ''' 上传相关非敏感统计数据 ''' + time.sleep(10) while True: platform_stats = {} llm_stats = {} @@ -481,8 +482,9 @@ def upload(_global_object: GlobalObject): } for llm in _global_object.llms: - for k, v in llm.llm_instance.model_stat: - llm_stats[llm.llm_name + "_" + k] = v + stat = llm.llm_instance.model_stat + for k in stat: + llm_stats[llm.llm_name + "#" + k] = stat[k] llm.llm_instance.reset_model_stat() for plugin in _global_object.cached_plugins: