gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix https://github.com/SengokuCola/MaiMBot/issues/23

Browse Source
This commit is contained in:
NepPure
2025-03-02 11:03:11 +08:00
parent 86b76496f6
commit bd2f0b258a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/schedule/schedule_llm_module.py
View File

@@ -6,8 +6,8 @@ class LLMModel:
# def __init__(self, model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", **kwargs): # def __init__(self, model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", **kwargs):
def __init__(self, model_name="Pro/deepseek-ai/DeepSeek-R1",api_using=None, **kwargs): def __init__(self, model_name="Pro/deepseek-ai/DeepSeek-R1",api_using=None, **kwargs):
if api_using == "deepseek": if api_using == "deepseek":
self.api_key = os.getenv("DEEPSEEK_API_KEY") self.api_key = os.getenv("DEEP_SEEK_KEY")
self.base_url = os.getenv("DEEPSEEK_BASE_URL") self.base_url = os.getenv("DEEP_SEEK_BASE_URL")
if model_name != "Pro/deepseek-ai/DeepSeek-R1": if model_name != "Pro/deepseek-ai/DeepSeek-R1":
self.model_name = model_name self.model_name = model_name
else: else: