"""
Version 0.1
"""

from __future__ import annotations
from fastapi import APIRouter
from pydantic import BaseModel, Field
from typing import List

from ..embeddings import embed_texts

router = APIRouter(prefix="/embed", tags=["embeddings"])

class EmbedRequest(BaseModel):
    texts: List[str] = Field(..., description="List of texts to embed")

class EmbedResponse(BaseModel):
    vectors: list[list[float]]
    dim: int

@router.post("", response_model=EmbedResponse, summary="Generate MiniLM embeddings for texts")
def post_embed(req: EmbedRequest):
    vecs = embed_texts(req.texts)
    dim = len(vecs[0]) if vecs else 0
    return {"vectors": vecs, "dim": dim}