120 lines
2.8 KiB
Python
120 lines
2.8 KiB
Python
from pydantic import BaseModel, Field, validator
|
|
from typing import Optional, List
|
|
from datetime import datetime
|
|
|
|
|
|
class MeetingBase(BaseModel):
|
|
title: str = Field(..., max_length=255)
|
|
tags: Optional[str] = None
|
|
meeting_time: Optional[datetime] = None
|
|
status: str = "draft"
|
|
asr_model_id: Optional[int] = None
|
|
summary_model_id: Optional[int] = None
|
|
type: str = "upload" # live, upload
|
|
|
|
@validator('meeting_time', pre=True)
|
|
def parse_meeting_time(cls, v):
|
|
if v == "":
|
|
return None
|
|
return v
|
|
|
|
|
|
class MeetingCreate(MeetingBase):
|
|
participants: Optional[List[int]] = None
|
|
file_path: Optional[str] = None
|
|
prompt_id: Optional[int] = None
|
|
|
|
|
|
class MeetingUpdate(BaseModel):
|
|
title: Optional[str] = None
|
|
tags: Optional[str] = None
|
|
status: Optional[str] = None
|
|
summary: Optional[str] = None
|
|
|
|
|
|
class AttendeeOut(BaseModel):
|
|
attendee_id: int
|
|
user_id: Optional[int]
|
|
display_name: Optional[str] = None
|
|
avatar: Optional[str] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class MeetingOut(MeetingBase):
|
|
meeting_id: int
|
|
user_id: Optional[int]
|
|
asr_model_id: Optional[int] = None
|
|
summary_model_id: Optional[int] = None
|
|
summary_prompt_id: Optional[int] = None
|
|
creator_name: Optional[str] = None
|
|
creator_avatar: Optional[str] = None
|
|
summary: Optional[str] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
attendees: List[AttendeeOut] = []
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TranscriptSegmentOut(BaseModel):
|
|
segment_id: int
|
|
speaker_id: Optional[int]
|
|
speaker_tag: Optional[str]
|
|
start_time_ms: int
|
|
end_time_ms: int
|
|
text_content: str
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class TranscriptTaskOut(BaseModel):
|
|
task_id: str
|
|
status: str
|
|
progress: int
|
|
error_message: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class SummarizeTaskOut(BaseModel):
|
|
task_id: str
|
|
status: str
|
|
progress: int
|
|
error_message: Optional[str] = None
|
|
created_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class MeetingAudioOut(BaseModel):
|
|
audio_id: int
|
|
file_name: Optional[str] = None
|
|
file_size: Optional[float] = None
|
|
duration: int
|
|
processing_status: str
|
|
upload_time: datetime
|
|
error_message: Optional[str] = None
|
|
audio_url: Optional[str] = None
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class MeetingDetailOut(MeetingOut):
|
|
segments: List[TranscriptSegmentOut] = []
|
|
latest_transcript_task: Optional[TranscriptTaskOut] = None
|
|
latest_summarize_task: Optional[SummarizeTaskOut] = None
|
|
audio_files: List[MeetingAudioOut] = []
|
|
|
|
|
|
class MeetingListOut(BaseModel):
|
|
items: List[MeetingOut]
|
|
total: int
|