| 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 | |