routes/sms_routes.py

from fastapi import APIRouter, HTTPException
from schema.pydantic_schema import SMSRequest, EmbeddingsResponse, EmbeddingResponse
# from service.embedding_service import generate_embeddings
from service.embedding_service import generate_embeddings


# Initialize Router
router = APIRouter()

@router.post("/get_embeddings/", response_model=EmbeddingsResponse)
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}