修改模型

This commit is contained in:
zk
2026-08-05 17:30:35 +08:00
parent c5440eefa9
commit bdbbf32b8f
6 changed files with 22 additions and 4 deletions
+4 -4
View File
@@ -9,14 +9,14 @@ from app.ai.models import LLM
class JobCleanModel:
"""岗位清洗模块"""
# 第一次AI:结构化提取岗位信息
STRUCTURE = LLM.GPT_5_4_MINI.create(temperature=0)
STRUCTURE = LLM.CLAUDE_SONNET_4_5.create(temperature=0)
# 第二次AI:专业匹配
MAJOR_MATCH = LLM.GPT_5_4_MINI.create(temperature=0)
MAJOR_MATCH = LLM.CLAUDE_SONNET_4_5.create(temperature=0)
# 第三次AI:技能提取
SKILL_EXTRACT = LLM.GPT_5_4_MINI.create(temperature=0)
SKILL_EXTRACT = LLM.CLAUDE_SONNET_4_5.create(temperature=0)
class CompanyCleanModel:
"""公司补充模块"""
# 公司信息补充
ENRICH = LLM.GPT_5_4_MINI.create(temperature=0)
ENRICH = LLM.CLAUDE_SONNET_4_5.create(temperature=0)
+2
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)
_ANTHROPIC_ALT = (lambda: settings.anthropic_alt_api_key, lambda: settings.anthropic_alt_base_url)
_OPENAI = (lambda: settings.openai_api_key, lambda: settings.openai_base_url)
@@ -33,6 +34,7 @@ class LLM(Enum):
# ClaudeAnthropic 风格)
CLAUDE_OPUS = ("claude-opus-4-6", ChatAnthropic, *_ANTHROPIC)
CLAUDE_SONNET_4_5 = ("claude-sonnet-4-5-20250929", ChatAnthropic, *_ANTHROPIC_ALT)
# OpenAIOpenAI 兼容)
GPT_5_4_MINI = ("gpt-5.4-mini", ChatOpenAI, *_OPENAI)