from fastapi import FastAPI
from pydantic import BaseModel
from sentence_transformers import SentenceTransformer

app = FastAPI()

model = SentenceTransformer('sentence-transformers/LaBSE')

class EmbedRequest(BaseModel):
    text: str

class EmbedResponse(BaseModel):
    embedding: list

@app.post("/vectorize", response_model=EmbedResponse)
async def embed_text(request: EmbedRequest):
    embedding = model.encode(request.text).tolist()
    return {"embedding": embedding}