Lars
a024487aae
All checks were successful
Deploy mindnet to llm-node / deploy (push) Successful in 3s
23 lines
649 B
Python
23 lines
649 B
Python
|
|
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}
|