deploy

main.py

@@ -6,3 +6,7 @@ app = FastAPI()
 
 # Include the router for SMS endpoints
 app.include_router(sms_router)
+@app.get("/")
+def home():
+    
+    return {"message": "Welcome to embedding sms API, use /docs to post SMS text and get demenstions"}

routes/sms_router.py

@@ -15,10 +15,28 @@ async def get_embeddings(sms_request: SMSRequest):
     # Generate embeddings
     embeddings = generate_embeddings(sms_request.messages)
 
-    # Structure the response
-    response = [
-        EmbeddingResponse(message=msg, embedding=emb) 
-        for msg, emb in zip(sms_request.messages, embeddings)
-    ]
+    from fastapi import APIRouter, HTTPException
+from schemas.schema import SMSRequest, EmbeddingResponse
+from service.embedded_service import generate_embeddings
+
+# Initialize Router
+router = APIRouter()
 
-    return {"embeddings": response}
+@router.post("/get_embeddings/", response_model=EmbeddingResponse)
+async def get_embeddings(sms_request: SMSRequest):
+    # Check if the input list is not empty
+    if not sms_request.messages:
+        raise HTTPException(status_code=400, detail="No messages provided.")
+    
+    # Generate embeddings
+    embeddings = generate_embeddings(sms_request.messages)
+    
+    # Check if embeddings are generated and are in the correct format
+    if not embeddings or not all(isinstance(emb, list) for emb in embeddings):
+        raise HTTPException(status_code=500, detail="Failed to generate embeddings.")
+    
+    # Get the dimensions from the first embedding (assuming all are the same)
+    dimensions = len(embeddings[0]) if embeddings else 0
+    
+    # Return the response as per the EmbeddingResponse model
+    return EmbeddingResponse(dimensions=dimensions, embeddings=embeddings)

schemas/schema.py

@@ -13,8 +13,8 @@ class EmbeddingResponse(BaseModel):
             "example": {
                 "dimensions": 768,
                 "embeddings": [
-                    [0.1, 0.2, 0.3, 0.4, 0.5, 0.6, ...],  # 768 values for each message
-                    [0.7, 0.8, 0.9, 1.0, 1.1, 1.2, ...]
+                    [0.1, 0.2, 0.3, 0.4, 0.5, 0.6],  # Example values for first message
+                    [0.7, 0.8, 0.9, 1.0, 1.1, 1.2]   # Example values for second message
                 ]
             }
         }