File size: 641 Bytes
ab2012f ce3e778 ab2012f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from pydantic import BaseModel
from typing import List, Optional
class WardrobeItem(BaseModel):
id: Optional[int] = None
category: str
style: str
color: Optional[str] = None
brand: Optional[str] = None
name: Optional[str] = None
class ChatRequest(BaseModel):
message: str
session_id: Optional[str] = "default"
wardrobe: Optional[List[WardrobeItem]] = None
wardrobe_description: Optional[str] = None
images: Optional[List[str]] = None
class ChatResponse(BaseModel):
response: str
session_id: str
class TextRequest(BaseModel):
message: str
session_id: Optional[str] = "default"
|