wehub-resource-sync
14 lines
334 B
Python
14 lines
334 B
Python
from pydantic import BaseModel
|
|
|
|
from models.presentation_outline_model import SlideOutlineModel
|
|
|
|
|
|
class DocumentChunk(BaseModel):
|
|
heading: str
|
|
content: str
|
|
heading_index: int
|
|
score: float
|
|
|
|
def to_slide_outline(self) -> SlideOutlineModel:
|
|
return SlideOutlineModel(content=f"{self.heading}\n{self.content}")
|