You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
584 B
25 lines
584 B
from pydantic import BaseModel
|
|
from typing import Dict, List, Optional
|
|
|
|
|
|
# 定义关系创建模型
|
|
class ChapterRelationCreate(BaseModel):
|
|
start_chapter: str # 起始章节的名称
|
|
end_chapter: str # 终止章节的名称
|
|
relation: str # 关系类型（如 "FOLLOWS", "RELATED_TO" 等）
|
|
|
|
|
|
# 定义关系查询响应模型
|
|
class ChapterRelationResponse(BaseModel):
|
|
# start_id: str
|
|
start_labels: List[str]
|
|
start_properties: Dict
|
|
|
|
relation: str
|
|
# relation_properties: Dict
|
|
#
|
|
# end_id: str
|
|
end_labels: List[str]
|
|
end_properties: Dict
|
|
|