models.py

from pydantic import BaseModel
from typing import List

class RequestModel(BaseModel):
    originId: int
    source: str

class TextSimilarityRequest(BaseModel):
    imageInfo: RequestModel
    keyTexts: List[str]
    similarityThreshold: float