File size: 569 Bytes
db774b8 | 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 | from pydantic import BaseModel
class Chunk(BaseModel):
id: str
text: str
source: str
metadata: dict = {}
class Entity(BaseModel):
id: str
name: str
type: str
chunk_id: str
properties: dict = {}
class Relationship(BaseModel):
id: str
source_entity: str
target_entity: str
relation_type: str
chunk_id: str
properties: dict = {}
class QueryResult(BaseModel):
query: str
answer: str
vector_context: list[str]
graph_context: list[str]
entities: list[dict]
relationships: list[dict]
|