token统计部分

2025-03-16 23:11:32 +08:00
parent 71132315d8
commit 4bc222ba6f
7 changed files with 43 additions and 13 deletions
--- a/src/plugins/schedule/schedule_generator.py
+++ b/src/plugins/schedule/schedule_generator.py
@@ -23,7 +23,7 @@ class ScheduleGenerator:
    def __init__(self):
        # 根据global_config.llm_normal这一字典配置指定模型
        # self.llm_scheduler = LLMModel(model = global_config.llm_normal,temperature=0.9)
-        self.llm_scheduler = LLM_request(model=global_config.llm_normal, temperature=0.9)
+        self.llm_scheduler = LLM_request(model=global_config.llm_normal, temperature=0.9,request_type = 'scheduler')
        self.today_schedule_text = ""
        self.today_schedule = {}
        self.tomorrow_schedule_text = ""