diff --git a/app/ai/models.py b/app/ai/models.py index 7184d48..18ec925 100644 --- a/app/ai/models.py +++ b/app/ai/models.py @@ -26,8 +26,10 @@ class LLM(Enum): # 火山引擎 DOUBAO_PRO_32K = ("doubao-1-5-pro-32k-250115", *_VOLCENGINE) + DOUBAO_LITE_32K = ("doubao-1-5-lite-32k-250115", *_VOLCENGINE) DEEPSEEK_V3 = ("deepseek-v3-250324", *_VOLCENGINE) DEEPSEEK_R1 = ("deepseek-r1-250528", *_VOLCENGINE) + DOUBAO_SEED_MINI = ("doubao-seed-2-0-mini-260215", *_VOLCENGINE) DOUBAO_SEED_LITE = ("doubao-seed-2-0-lite-260215", *_VOLCENGINE) DOUBAO_SEED_PRO = ("doubao-seed-2-0-pro-260215", *_VOLCENGINE) diff --git a/app/ai/resume_extractor/extractor.py b/app/ai/resume_extractor/extractor.py index ee6a148..29c1ed5 100644 --- a/app/ai/resume_extractor/extractor.py +++ b/app/ai/resume_extractor/extractor.py @@ -21,7 +21,7 @@ from app.ai.resume_extractor.prompts import ( from app.core.logger import log from app.tool.json_helper import parse_llm_json -_LLM_MODEL = LLM.DOUBAO_PRO_32K +_LLM_MODEL = LLM.DOUBAO_LITE_32K # ==================== 文本编号 ====================