Triventure-AI / src /apis /models /api_models.py
ABAO77's picture
Upload 37 files
5ce8318 verified
raw
history blame contribute delete
434 Bytes
from typing import Any, Dict, List
from pydantic import BaseModel, Field
from typing import Annotated
class WeightUpdateRequest(BaseModel):
tag_indices: List[int]
new_weights: List[float]
metadata: Dict[str, Any] = {}
class FeedbackRequest(BaseModel):
destination_id: int
tag_id: int
rating: int # 1-5 stars
class ImageSearchBody(BaseModel):
base64_image: str = Field(..., title="Base64 Image String")