"""Record数据结构"""
import uuid
from typing import Any, Literal
from pydantic import BaseModel, Field
from apps.models import StepStatus
class RecordDocument(BaseModel):
"""GET /api/record/{conversation_id} Result中的document数据结构"""
id: str = Field(alias="_id", default="")
order: int = Field(default=0, description="文档顺序")
abstract: str = Field(default="", description="文档摘要")
user_id: None = None
author: str = Field(default="", description="文档作者")
associated: Literal["question", "answer"]
class Config:
"""配置"""
populate_by_name = True
class RecordFlowStep(BaseModel):
"""Record表子项：flow的单步数据结构"""
step_id: str = Field(alias="stepId")
step_status: StepStatus = Field(alias="stepStatus")
input: dict[str, Any]
output: dict[str, Any]
ex_data: dict[str, Any] | None = Field(default=None, alias="exData")
class RecordExecutor(BaseModel):
"""Flow的执行信息"""
id: str
record_id: str = Field(alias="recordId")
flow_id: str = Field(alias="flowId")
flow_name: str = Field(alias="flowName", default="")
flow_status: StepStatus = Field(alias="flowStatus", default=StepStatus.SUCCESS)
step_num: int = Field(alias="stepNum")
steps: list[RecordFlowStep]
class RecordContent(BaseModel):
"""Record表子项：Record加密前的数据结构"""
question: str
answer: str
data: dict[str, Any] = {}
facts: list[str] = Field(description="[运行后修改]与Record关联的事实信息", default=[])
class RecordMetadata(BaseModel):
"""Record表子项：Record的元信息"""
input_tokens: int = Field(default=0, alias="inputTokens")
output_tokens: int = Field(default=0, alias="outputTokens")
time_cost: float = Field(default=0, alias="timeCost")
class RecordData(BaseModel):
"""GET /api/record/{conversation_id} Result内元素数据结构"""
id: uuid.UUID
conversation_id: uuid.UUID = Field(alias="conversationId")
task_id: uuid.UUID | None = Field(alias="taskId", default=None)
document: list[RecordDocument] = []
flow: RecordExecutor | None = None
content: RecordContent
metadata: RecordMetadata
created_at: float = Field(alias="createdAt")