from pydantic import BaseModel
from typing import List, Optional, Dict, Any

class ChatRequest(BaseModel):
    message: str
    user_id: str
    context: Optional[List[str]] = []

class ChatResponse(BaseModel):
    response: str
    services: List[Dict[str, Any]] 