secret 新增了人格测评两种模型

2025-03-20 21:19:43 +08:00
parent 8c178b80d7
commit f688362a0f
6 changed files with 362 additions and 51 deletions
--- a/src/plugins/personality/offline_llm.py
+++ b/src/plugins/personality/offline_llm.py
@@ -11,7 +11,7 @@ logger = get_module_logger("offline_llm")


 class LLMModel:
-    def __init__(self, model_name="deepseek-ai/DeepSeek-V3", **kwargs):
+    def __init__(self, model_name="Pro/deepseek-ai/DeepSeek-V3", **kwargs):
        self.model_name = model_name
        self.params = kwargs
        self.api_key = os.getenv("SILICONFLOW_KEY")