api/schemas.py

# rag/api/schemas.py
from pydantic import BaseModel
from typing import List, Dict, Any

class SearchRequest(BaseModel):
    query: str

class SearchResponse(BaseModel):
    results: List[Dict[str, Any]]