models.py

from pydantic import BaseModel

class RequestModel(BaseModel):
    originId: int
    sequence: int
    assetCode: str
    contentType: str
    source: str


class ResponseModel(BaseModel):
    originId: int
    sequence: int
    assetCode: str
    contentType: str
    similarity: float
    similarityOrb: float
    similarityMobileNet: float
    source: str