models.py

from pydantic import BaseModel
from typing import Optional

class RequestModel(BaseModel):
    originId: int
    sequence: int
    assetCode: Optional[str] = None    
    contentType: Optional[str] = None  
    source: str


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