"""Pydantic request bodies for the evaluation FastAPI app."""
  
  from __future__ import annotations
  
  from typing import List, Optional
  
  from pydantic import BaseModel, Field
  
  
  class SearchEvalRequest(BaseModel):
      query: str

      dataset_id: Optional[str] = None

      top_k: int = Field(default=100, ge=1, le=500)
      auto_annotate: bool = False

      language: Optional[str] = None

  
  
  class BatchEvalRequest(BaseModel):

      dataset_id: Optional[str] = None

      queries: Optional[List[str]] = None
      top_k: int = Field(default=100, ge=1, le=500)
      auto_annotate: bool = False

      language: Optional[str] = None

      force_refresh_labels: bool = False