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="Pro/deepseek-ai/DeepSeek-R1",api_using=None, **kwargs):
if api_using == "deepseek":
self.api_key = os.getenv("DEEPSEEK_API_KEY")
self.base_url = os.getenv("DEEPSEEK_BASE_URL")
self.api_key = os.getenv("DEEP_SEEK_KEY")
self.base_url = os.getenv("DEEP_SEEK_BASE_URL")
if model_name != "Pro/deepseek-ai/DeepSeek-R1":
self.model_name = model_name
else: