app/schemas/chat.py

from pydantic import BaseModel
from typing import List, Optional
from datetime import datetime

class Message(BaseModel):
    content: str
    role: str

class ChatRequest(BaseModel):
    question: str
    user_id: Optional[int] = None

class ChatResponse(BaseModel):
    answer: str
    sources: List[str] = []

class ChatSelectionRequest(BaseModel):
    question: str
    selected_text: str
    user_id: Optional[int] = None

class ChatSelectionResponse(BaseModel):
    response: str