schemas.py

from pydantic import BaseModel
from typing import *

class DocRequest(BaseModel):
    doc_id: str

class DocResponse(BaseModel):
    doc_id: str
    url: str
    release: Optional[str] = None
    scope: Optional[str] = None
    search_time: float
    
class BatchDocRequest(BaseModel):
    doc_ids: List[str]
    release: Optional[int] = None

class BatchDocResponse(BaseModel):
    results: Dict[str, str]
    missing: List[str]
    search_time: float

class BM25KeywordRequest(BaseModel):
    keywords: Optional[str] = ""
    threshold: Optional[int] = 60
    release: Optional[str] = None
    spec_type: Optional[Literal["TS", "TR"]] = None

class KeywordRequest(BaseModel):
    keywords: Optional[str] = ""
    search_mode: Literal["quick", "deep"]
    case_sensitive: Optional[bool] = False
    release: Optional[str] = None
    spec_type: Optional[Literal["TS", "TR"]] = None
    mode: Optional[Literal["and", "or"]] = "and"

class KeywordResponse(BaseModel):
    results: List[Dict[str, Any]]
    search_time: float