Reqxtract-v2 / schemas.py
om4r932's picture
Add packages PIP + add schema file
ded7ed5
raw
history blame
1.04 kB
from pydantic import BaseModel
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 RequirementsRequest(BaseModel):
class DocInfo(BaseModel):
document: str
url: str
documents: List[DocInfo]
class DocRequirements(BaseModel):
document: str
context: str
requirements: List[str]
class RequirementsResponse(BaseModel):
requirements: List[DocRequirements]
# --------------------------------------
class SingleRequirement(BaseModel):
document: str
context: str
requirement: str
class ReqSearchRequest(BaseModel):
query: str
requirements: List[SingleRequirement]
class ReqSearchResponse(BaseModel):
requirements: List[SingleRequirement]