101 lines
2.4 KiB
Python
101 lines
2.4 KiB
Python
"""
|
||
Pydantic 数据模型
|
||
"""
|
||
from typing import Optional, List, Any
|
||
from pydantic import BaseModel, Field
|
||
from datetime import datetime
|
||
|
||
|
||
# ============ 通用响应 ============
|
||
|
||
class ApiResponse(BaseModel):
|
||
"""统一 API 响应格式"""
|
||
code: int = 0
|
||
message: str = "success"
|
||
data: Any = None
|
||
|
||
|
||
# ============ 候选人相关 ============
|
||
|
||
class SubmitCandidateResponse(BaseModel):
|
||
"""提交候选人信息响应"""
|
||
sessionId: str
|
||
fileId: str
|
||
|
||
|
||
class CreateRoomRequest(BaseModel):
|
||
"""创建房间请求"""
|
||
sessionId: Optional[str] = None # 方案A：可选，由后端生成
|
||
fileId: Optional[str] = None # 方案A：可选，由工作流收集
|
||
|
||
|
||
class CreateRoomResponse(BaseModel):
|
||
"""创建房间响应"""
|
||
roomId: str
|
||
token: str
|
||
appId: str
|
||
userId: str
|
||
sessionId: Optional[str] = None # 返回给前端用于后续操作
|
||
debugInfo: Optional[Any] = None # 调试信息（语音模式）
|
||
|
||
|
||
class EndInterviewResponse(BaseModel):
|
||
"""结束面试响应"""
|
||
success: bool
|
||
|
||
|
||
class ChatRequest(BaseModel):
|
||
"""文本对话请求（模拟语音）"""
|
||
sessionId: str
|
||
message: str
|
||
conversationId: Optional[str] = None
|
||
|
||
|
||
class ChatResponse(BaseModel):
|
||
"""文本对话响应"""
|
||
reply: str
|
||
conversationId: str
|
||
debugInfo: Optional[Any] = None # 调试信息（节点状态、消息列表等）
|
||
|
||
|
||
# ============ 管理后台相关 ============
|
||
|
||
class CandidateScores(BaseModel):
|
||
"""候选人评分"""
|
||
salesSkill: int = Field(0, ge=0, le=100)
|
||
salesMindset: int = Field(0, ge=0, le=100)
|
||
quality: int = Field(0, ge=0, le=100)
|
||
motivation: int = Field(0, ge=0, le=100)
|
||
total: float = Field(0, ge=0, le=100)
|
||
|
||
|
||
class CandidateListItem(BaseModel):
|
||
"""候选人列表项"""
|
||
sessionId: str
|
||
name: str
|
||
status: str # pending, ongoing, completed
|
||
score: Optional[float] = None
|
||
createdAt: str
|
||
|
||
|
||
class CandidateDetail(BaseModel):
|
||
"""候选人详情"""
|
||
sessionId: str
|
||
name: str
|
||
resume: Optional[str] = None
|
||
status: str
|
||
currentStage: int = 0
|
||
scores: Optional[CandidateScores] = None
|
||
analysis: Optional[str] = None
|
||
interviewLog: Optional[str] = None
|
||
createdAt: str
|
||
completedAt: Optional[str] = None
|
||
|
||
|
||
class CandidateListResponse(BaseModel):
|
||
"""候选人列表响应"""
|
||
list: List[CandidateListItem]
|
||
total: int
|
||
page: int
|
||
pageSize: int
|