from pydantic import BaseModel

class CosineSimilarity(BaseModel):
    message_1 : str
    message_2 : str