瀏覽代碼

Update dialogue_manager: change model

StrayWarrior 1 周之前
父節點
當前提交
4b5c9f4d43
共有 1 個文件被更改,包括 3 次插入1 次删除
  1. 3 1
      dialogue_manager.py

+ 3 - 1
dialogue_manager.py

@@ -7,6 +7,7 @@ from typing import Dict, List, Optional, Tuple, Any
 from datetime import datetime
 import time
 
+import chat_service
 import prompt_templates
 from logging_service import logger
 
@@ -619,7 +620,8 @@ class DialogueManager:
 
     def build_active_greeting_config(self, user_tags: List[str]):
         # FIXME: 这里的抽象不好,短期支持人为配置实验
-        chat_config = {'user_id': self.user_id}
+        # 由于产运要求,指定使用GPT-4o模型
+        chat_config = {'user_id': self.user_id, 'model_name': chat_service.OPENAI_MODEL_GPT_4o}
         prompt_context = self.get_prompt_context(None)
 
         current_time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')