22 lines
450 B
Python
22 lines
450 B
Python
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
from models.dataset import DocumentSegment
|
||
|
|
|
||
|
|
|
||
|
|
class RetrievalChildChunk(BaseModel):
|
||
|
|
"""Retrieval segments."""
|
||
|
|
|
||
|
|
id: str
|
||
|
|
content: str
|
||
|
|
score: float
|
||
|
|
position: int
|
||
|
|
|
||
|
|
|
||
|
|
class RetrievalSegments(BaseModel):
|
||
|
|
"""Retrieval segments."""
|
||
|
|
|
||
|
|
model_config = {"arbitrary_types_allowed": True}
|
||
|
|
segment: DocumentSegment
|
||
|
|
child_chunks: list[RetrievalChildChunk] | None = None
|
||
|
|
score: float | None = None
|