from typing import List

from pydantic import BaseModel


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


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


class IndexInsertInput(BaseModel):
    idx: str
    id: int
    embedding: List[float]


class IndexSearchInput(BaseModel):
    idx: str
    embedding: List[float]
    k: int = 5


class IndexDeleteInput(BaseModel):
    idx: str