# models.py
from pydantic import BaseModel
from typing import List

class EmbeddingInput(BaseModel):
    embedding: List[float]

class SearchInput(BaseModel):
    embedding: List[float]
    k: int = 5