Spaces:

ABAO77
/

Triventure-AI

Build error

Triventure-AI / src /apis /models /api_models.py

Upload 37 files

5ce8318 verified over 1 year ago

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")

	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")