utils/data_proto.py

from typing import List

from pydantic import BaseModel
from typing_extensions import TypedDict


class SummaryReq(BaseModel):
    key: str
    id: str
    text: str


class Page(TypedDict):
    id: str
    text: str


class SummariesReq(BaseModel):
    key: str
    pages: List[Page]