File size: 668 Bytes
325b94c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
from typing import List, Optional
from pydantic import BaseModel


class PartModel(BaseModel):
    page_url: Optional[str] = None
    title: Optional[str] = None
    content: Optional[str] = None

    img_url: Optional[List[str]] = None
    video_url: Optional[List[str]] = None
    audio_url: Optional[List[str]] = None
    pdf_url: Optional[List[str]] = None

    agent_recommendation_rank: Optional[float] = None
    agent_recommendation_notes: Optional[str] = None


class UnitSubtopicOutputModel(BaseModel):
    unit_title: Optional[str] = None
    subtopic_title: Optional[str] = None
    query: Optional[str] = None

    parts: Optional[List[PartModel]] = None