修改模型
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user