File size: 434 Bytes
5ce8318
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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")