from pydantic import BaseModel
from typing import Dict, Any

class ChatRequest(BaseModel):
    message: str
    user_id: str

class ChatResponse(BaseModel):
    response: str



class ToolCreate(BaseModel):
    name: str
    description: str
    parameters: Dict[str, Any]


class ToolResponse(BaseModel):
    id: int
    name: str
    description: str
    parameters: Dict[str, Any]

    class Config:
        from_attributes = True