ContiAI / schemas /extractor_schema.py
ziadsameh32's picture
Add login page
325b94c
Raw
History Blame Contribute Delete
668 Bytes
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