routes/sms_router.py

from fastapi import APIRouter, HTTPException
from schemas.schema import SMSRequest
from schemas.schema import EmbeddingResponse
from service.embedded_service import generate_embeddings

# Initialize Router
router = APIRouter()

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

    # Structure the response
    response = [
        EmbeddingResponse(message=msg, embedding=emb) 
        for msg, emb in zip(sms_request.messages, embeddings)
    ]

    return {"embeddings": response}