Test / backend /models.py
UKielz's picture
Upload 14 files
0bbe8e9 verified
raw
history blame contribute delete
518 Bytes
from typing import Dict, Any, List, Optional
from pydantic import BaseModel
class ChatRequest(BaseModel):
message: str
feedback: Optional[str] = None
class NLUResult(BaseModel):
intent: str
confidence: float
class ActionResponse(BaseModel):
type: str
parameters: Optional[Dict[str, Any]] = {}
class ChatResponse(BaseModel):
response: str
context: str
nlu: NLUResult
action: Optional[ActionResponse] = None
images: Optional[List[str]] = None