from typing import Optional
from pydantic import BaseModel


class CreateVectorStoreRequest(BaseModel):
    name: str


class SearchSimilarityRequest(BaseModel):
    name_database: str
    query: str
    fuente: Optional[str] = None


class ListSourcesRequest(BaseModel):
    nombre_db_vectorial: str


class SaveTempRequest(BaseModel):
    nombre_db_vectorial: str
    fuente: str


class AddFilesRequest(BaseModel):
    nombre_db_vectorial: str


class DeleteVectorStoreRequest(BaseModel):
    nombre_db_vectorial: str


class DownloadVectorStoreRequest(BaseModel):
    nombre_db_vectorial: str