22 lines
580 B
Python
22 lines
580 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class CollectedData(BaseModel):
|
||
|
|
client_name: str | None = None
|
||
|
|
date: str | None = None # YYYY-MM-DD
|
||
|
|
time_start: str | None = None # HH:MM
|
||
|
|
party_size: int | None = None
|
||
|
|
|
||
|
|
|
||
|
|
class ConversationContext(BaseModel):
|
||
|
|
phone: str
|
||
|
|
business_id: int
|
||
|
|
collected_data: CollectedData = CollectedData()
|
||
|
|
messages: list[dict] = [] # historial {role, content} para Claude
|
||
|
|
|
||
|
|
|
||
|
|
class BotResponse(BaseModel):
|
||
|
|
message: str
|
||
|
|
action: str # "collect_more" | "create_reservation" | "cancel"
|
||
|
|
collected_data: CollectedData
|