File size: 1,047 Bytes
48e4b27 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from pydantic import BaseModel
from typing import *
class DocRequest(BaseModel):
doc_id: str
class DocResponse(BaseModel):
doc_id: str
url: str
version: str
scope: Optional[str] = None
search_time: float
class BatchDocRequest(BaseModel):
doc_ids: List[str]
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
working_group: 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
working_group: 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 |