dto.py

from datetime import datetime
from typing import List
from typing import Optional

from pydantic import BaseModel
from pydantic import Field


class SummarizeRequest(BaseModel):
    """Text Summarize request model."""
    text: str = Field(..., description="The text you want to summarize", examples=["ከተለያዩ የትግራይ አካባቢዎች በመምጣት ኑሯቸውን ..."])
    model: Optional[int] = Field(None, description="model index 0 = ExtractiveModel, 1 = Cosine Similarity, 3 = Abstractive - News Headline", examples=[0,1,2])