from pydantic import BaseModel
from typing import Any, Dict

class PageSpeedRequest(BaseModel):
    pagespeed_data: Dict[str, Any]

class PageSpeedURLRequest(BaseModel):
    target_url: str

class ReportIngestRequest(BaseModel):
    user_id: str
    report_text: str

class ReportQueryRequest(BaseModel):
    user_id: str
    query: str

class NewChatRequest(BaseModel):
    user_id: str


class ChatQueryRequest(BaseModel):
    user_id: str
    chat_id: str
    query: str