23 lines
550 B
Python
Executable File
23 lines
550 B
Python
Executable File
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
|