修一下漏改的deepseek API支持

2025-03-01 18:52:40 +08:00
parent 0ed3b2bf9c
commit 2554f8a931
2 changed files with 16 additions and 5 deletions
--- a/src/plugins/schedule/schedule_generator.py
+++ b/src/plugins/schedule/schedule_generator.py
@@ -29,7 +29,10 @@ Database.initialize(

 class ScheduleGenerator:
    def __init__(self):
-        self.llm_scheduler = LLMModel(model_name="Pro/deepseek-ai/DeepSeek-V3")
+        if global_config.API_USING == "siliconflow":
+            self.llm_scheduler = LLMModel(model_name="Pro/deepseek-ai/DeepSeek-V3")
+        elif global_config.API_USING == "deepseek":
+            self.llm_scheduler = LLMModel(model_name="deepseek-chat",api_using="deepseek")
        self.db = Database.get_instance()
        
        today = datetime.datetime.now()