from pydantic import BaseModel

# For Cosine Similarity input
class CosineSimilarityInput(BaseModel):
    text1: str
    text2: str

# For SMS classification input
class MessageInput(BaseModel):
    message: str

# For Embedding computation input
class EmbeddingInput(BaseModel):
    message: str

# For Cosine Similarity output response
class CosineSimilarityResponse(BaseModel):
    cosine_similarity: float

# For Embedding output response
class EmbeddingResponse(BaseModel):
    embeddings: list