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