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

app = FastAPI()  # ✅ This line is critical

model = SentenceTransformer("BAAI/bge-small-zh")

class InputText(BaseModel):
    texts: list[str]

@app.post("/embed")
def embed(texts: InputText):
    vectors = model.encode(texts.texts, normalize_embeddings=True).tolist()
    return {"embeddings": vectors}