diff --git a/app/schemas/skill_gap.py b/app/schemas/skill_gap.py index c7a76e3..20e4e3a 100644 --- a/app/schemas/skill_gap.py +++ b/app/schemas/skill_gap.py @@ -4,8 +4,13 @@ 字段命名使用 camelCase alias,与前端 JSON 对齐。 """ +from typing import Literal + from pydantic import BaseModel, Field +# 定制简历可优化模块 +OptimizeModule = Literal["summary", "skills", "experience"] + # ===== 请求参数 ===== @@ -16,7 +21,7 @@ class SkillGapParam(BaseModel): class CustomizeResumeParam(BaseModel): job_id: int = Field(..., alias="jobId") resume_id: int = Field(..., alias="resumeId") - optimize_modules: list[str] = Field(..., alias="optimizeModules") + optimize_modules: list[OptimizeModule] = Field(..., alias="optimizeModules") add_skills: list[str] = Field(default_factory=list, alias="addSkills")