Spaces:
Sleeping
Sleeping
File size: 1,692 Bytes
86c402d 744a170 86c402d |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from typing import List, Optional
from pydantic import BaseModel
class EntitySearchRequest(BaseModel):
"""Схема запроса для поиска сущностей."""
query: str
dataset_id: int
class EntitySearchResponse(BaseModel):
"""Схема ответа с результатами поиска сущностей."""
scores: List[float]
entity_ids: List[str]
class EntitySearchWithTextRequest(BaseModel):
"""Схема запроса для поиска сущностей с текстами."""
query: str
dataset_id: int
class ChunkInfo(BaseModel):
"""Информация о чанке."""
id: str
text: str
score: float
type: str
in_search_text: str
class EntitySearchWithTextResponse(BaseModel):
"""Схема ответа с результатами поиска сущностей и их текстами."""
chunks: List[ChunkInfo]
class EntityTextRequest(BaseModel):
"""Схема запроса для сборки текста из сущностей."""
entities: List[str]
chunk_scores: Optional[dict[str, float]] = None
include_tables: bool = True
max_documents: Optional[int] = None
class EntityTextResponse(BaseModel):
"""Схема ответа со сборкой текста из сущностей."""
text: str
class EntityNeighborsRequest(BaseModel):
"""Схема запроса для получения соседних чанков."""
entities: List[str]
max_distance: int = 1
class EntityNeighborsResponse(BaseModel):
"""Схема ответа с соседними чанками."""
entities: List[str]
|