File size: 1,269 Bytes
50d44e1 ded7ed5 03e159d ded7ed5 fe9ab1e ded7ed5 1538533 ded7ed5 03e159d 50d44e1 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
from pydantic import BaseModel, Field
from typing import Any, List, Dict, Optional
class MeetingsRequest(BaseModel):
working_group: str
class MeetingsResponse(BaseModel):
meetings: Dict[str, str]
# --------------------------------------
class DataRequest(BaseModel):
working_group: str
meeting: str
class DataResponse(BaseModel):
data: List[Dict[Any, Any]]
# --------------------------------------
class DocInfo(BaseModel):
document: str
url: str
class RequirementsRequest(BaseModel):
documents: List[DocInfo]
class DocRequirements(BaseModel):
document: str
context: str
requirements: List[str]
class RequirementsResponse(BaseModel):
requirements: List[DocRequirements]
# --------------------------------------
class SingleRequirement(BaseModel):
req_id: int
document: str
context: str
requirement: str
class ReqSearchLLMResponse(BaseModel):
selected: List[int]
class ReqSearchRequest(BaseModel):
query: str
requirements: List[SingleRequirement]
class ReqSearchResponse(BaseModel):
requirements: List[SingleRequirement]
# --------------------------------------
class DownloadRequest(BaseModel):
documents: List[str] = Field(description="List of document IDs to download") |