# -*- coding: utf-8 -*-
|
from typing import List, Optional
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
class KeywordConfig(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
storefront: List[str] = Field(default_factory=lambda: ["到店", "到达", "门头"])
|
handover: List[str] = Field(default_factory=lambda: ["交付", "交货", "签收"])
|
|
|
class AnalyzeRequest(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
media_id: int = Field(alias="mediaId")
|
video_url: Optional[str] = Field(default=None, alias="videoUrl")
|
sample_fps: float = Field(default=0.5, alias="sampleFps")
|
enable_asr: bool = Field(default=True, alias="enableAsr")
|
keywords: Optional[KeywordConfig] = None
|
duration_sec: Optional[float] = Field(default=None, alias="durationSec")
|
|
|
class SnapshotHit(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
time_sec: float = Field(alias="timeSec")
|
confidence: float
|
source: str
|
|
|
class AsrHit(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
keyword: str
|
time_sec: float = Field(alias="timeSec")
|
|
|
class AnalyzeResponse(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
success: bool = True
|
model_version: str = Field(default="1.0.0", alias="modelVersion")
|
duration_sec: float = Field(default=0.0, alias="durationSec")
|
storefront: Optional[SnapshotHit] = None
|
handover: Optional[SnapshotHit] = None
|
asr_hits: List[AsrHit] = Field(default_factory=list, alias="asrHits")
|
message: Optional[str] = None
|
|
|
class HealthResponse(BaseModel):
|
model_config = ConfigDict(populate_by_name=True)
|
status: str
|
model_version: str = Field(alias="modelVersion")
|
onnx_storefront_loaded: bool = Field(default=False, alias="onnxStorefrontLoaded")
|
onnx_handover_loaded: bool = Field(default=False, alias="onnxHandoverLoaded")
|
asr_available: bool = Field(default=False, alias="asrAvailable")
|
