from pydantic import BaseModel
from typing import Dict, List, Any

class CreateCollectionSerializer(BaseModel):
    collection_name: str
    vector: List[float]
    vector_size: int
    payload: Dict[str, Any]
    id: int

class QueryCollectionSerializer(BaseModel):
    collection_name: str
    query_vector: List[float]

class UpdateCollectionSerializer(BaseModel):
    collection_name: str
    vector: List[float]
    payload: Dict[str, Any]
    id: int

class DeleteCollectionSerializer(BaseModel):
    collection_name: str