修改模型
This commit is contained in:
+1
-1
@@ -27,7 +27,7 @@ class LLM(Enum):
|
|||||||
DOUBAO_LITE_128K = ("doubao-lite-128k", *_VOLCENGINE)
|
DOUBAO_LITE_128K = ("doubao-lite-128k", *_VOLCENGINE)
|
||||||
DEEPSEEK_V3 = ("deepseek-v3-250324", *_VOLCENGINE)
|
DEEPSEEK_V3 = ("deepseek-v3-250324", *_VOLCENGINE)
|
||||||
DEEPSEEK_R1 = ("deepseek-r1-250528", *_VOLCENGINE)
|
DEEPSEEK_R1 = ("deepseek-r1-250528", *_VOLCENGINE)
|
||||||
DOUBAO_SEED_LIST = ("doubao-seed-2-0-lite-260215", *_VOLCENGINE)
|
DOUBAO_SEED_LITE = ("doubao-seed-2-0-lite-260215", *_VOLCENGINE)
|
||||||
DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE)
|
DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE)
|
||||||
|
|
||||||
# 心缘
|
# 心缘
|
||||||
|
|||||||
@@ -116,7 +116,7 @@ class ResumeParseService:
|
|||||||
|
|
||||||
async def _ai_parse(self, text: str) -> dict:
|
async def _ai_parse(self, text: str) -> dict:
|
||||||
"""调用 AI 将纯文本解析为结构化 JSON"""
|
"""调用 AI 将纯文本解析为结构化 JSON"""
|
||||||
llm = LLM.DOUBAO_SEED_PRO.create(temperature=0)
|
llm = LLM.DOUBAO_SEED_LITE.create(temperature=0)
|
||||||
messages = [SystemMessage(content=_SYSTEM_PROMPT), HumanMessage(content=text)]
|
messages = [SystemMessage(content=_SYSTEM_PROMPT), HumanMessage(content=text)]
|
||||||
response = await llm.ainvoke(messages)
|
response = await llm.ainvoke(messages)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user