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