修改模型

2026-07-30 18:00:09 +08:00
parent ca227ebef7
commit c5440eefa9
6 changed files with 27 additions and 7 deletions
@@ -18,6 +18,7 @@ from app.config import settings
 # 供应商连接配置 = (api_key函数, base_url函数)
 _VOLCENGINE = (lambda: settings.volcengine_api_key, lambda: settings.volcengine_base_url)
 _ANTHROPIC = (lambda: settings.anthropic_api_key, lambda: settings.anthropic_base_url)
+_OPENAI = (lambda: settings.openai_api_key, lambda: settings.openai_base_url)


 class LLM(Enum):
@@ -33,6 +34,9 @@ class LLM(Enum):
    # Claude（Anthropic 风格）
    CLAUDE_OPUS = ("claude-opus-4-6", ChatAnthropic, *_ANTHROPIC)

+    # OpenAI（OpenAI 兼容）
+    GPT_5_4_MINI = ("gpt-5.4-mini", ChatOpenAI, *_OPENAI)
+
    def __init__(self, model_name: str, cls, api_key_fn, base_url_fn):
        self.model_name = model_name
        self._cls = cls