changed

schemas/input_schemas.py

@@ -1,19 +1,22 @@
-# app/schemas/input_schemas.py
 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
 
-
-class EmbeddingInput(BaseModel):
-    message: str
-
+# For Embedding output response
+class EmbeddingResponse(BaseModel):
+    embeddings: list

services/sms_service.py

@@ -6,6 +6,7 @@ from fastapi import HTTPException
 from schemas.input_schemas import CosineSimilarityResponse
 from schemas.input_schemas import EmbeddingResponse
 
+
 # Load the trained model and vectorizer
 def load_model():
     model_path = "models/sms_classifier_model.pkl"