修改模型

This commit is contained in:
zk
2026-07-30 18:00:09 +08:00
parent ca227ebef7
commit c5440eefa9
6 changed files with 27 additions and 7 deletions
+4 -4
View File
@@ -9,14 +9,14 @@ from app.ai.models import LLM
class JobCleanModel:
"""岗位清洗模块"""
# 第一次AI:结构化提取岗位信息
STRUCTURE = LLM.CLAUDE_OPUS.create(temperature=0)
STRUCTURE = LLM.GPT_5_4_MINI.create(temperature=0)
# 第二次AI:专业匹配
MAJOR_MATCH = LLM.CLAUDE_OPUS.create(temperature=0)
MAJOR_MATCH = LLM.GPT_5_4_MINI.create(temperature=0)
# 第三次AI:技能提取
SKILL_EXTRACT = LLM.CLAUDE_OPUS.create(temperature=0)
SKILL_EXTRACT = LLM.GPT_5_4_MINI.create(temperature=0)
class CompanyCleanModel:
"""公司补充模块"""
# 公司信息补充
ENRICH = LLM.CLAUDE_OPUS.create(temperature=0)
ENRICH = LLM.GPT_5_4_MINI.create(temperature=0)
+4
View File
@@ -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):
# ClaudeAnthropic 风格)
CLAUDE_OPUS = ("claude-opus-4-6", ChatAnthropic, *_ANTHROPIC)
# OpenAIOpenAI 兼容)
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