from typing import Any, Dict, List, Optional

from pydantic import BaseModel


class ArchRequest(BaseModel):
    repo_id: str


class AskRequest(BaseModel):
    repo_id: str
    question: str
    chat_history: Optional[List[Dict[str, Any]]] = []


class DocsRequest(BaseModel):
    repo_id: str
    doc_type: str = "onboarding"