from typing import Any, Optional

from pydantic import BaseModel



class RetrievalResponse(BaseModel):
    """Simple response container for tool results"""
    results: list[dict[str, Any]]
    took_ms: Optional[int] = None
    total_count: Optional[int] = None