feat: Phase 2 交互与纠错工作台完成
- 新增实体编辑 API (名称、类型、定义、别名) - 新增实体删除和合并功能 - 新增关系管理 (创建、删除) - 新增转录文本编辑功能 - 新增划词创建实体功能 - 前端新增实体编辑器模态框 - 前端新增右键菜单和工具栏 - 文本与图谱双向联动优化
This commit is contained in:
OpenClaw Bot
264
backend/main.py
264
backend/main.py
@@ -71,10 +71,251 @@ class ProjectCreate(BaseModel):
|
||||
name: str
|
||||
description: str = ""
|
||||
|
||||
class EntityUpdate(BaseModel):
|
||||
name: Optional[str] = None
|
||||
type: Optional[str] = None
|
||||
definition: Optional[str] = None
|
||||
aliases: Optional[List[str]] = None
|
||||
|
||||
class RelationCreate(BaseModel):
|
||||
source_entity_id: str
|
||||
target_entity_id: str
|
||||
relation_type: str
|
||||
evidence: Optional[str] = ""
|
||||
|
||||
class TranscriptUpdate(BaseModel):
|
||||
full_text: str
|
||||
|
||||
class EntityMergeRequest(BaseModel):
|
||||
source_entity_id: str
|
||||
target_entity_id: str
|
||||
|
||||
# API Keys
|
||||
KIMI_API_KEY = os.getenv("KIMI_API_KEY", "")
|
||||
KIMI_BASE_URL = "https://api.kimi.com/coding"
|
||||
|
||||
# Phase 2: Entity Edit API
|
||||
@app.put("/api/v1/entities/{entity_id}")
|
||||
async def update_entity(entity_id: str, update: EntityUpdate):
|
||||
"""更新实体信息（名称、类型、定义、别名）"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
entity = db.get_entity(entity_id)
|
||||
if not entity:
|
||||
raise HTTPException(status_code=404, detail="Entity not found")
|
||||
|
||||
# 更新字段
|
||||
update_data = {k: v for k, v in update.dict().items() if v is not None}
|
||||
updated = db.update_entity(entity_id, **update_data)
|
||||
|
||||
return {
|
||||
"id": updated.id,
|
||||
"name": updated.name,
|
||||
"type": updated.type,
|
||||
"definition": updated.definition,
|
||||
"aliases": updated.aliases
|
||||
}
|
||||
|
||||
@app.delete("/api/v1/entities/{entity_id}")
|
||||
async def delete_entity(entity_id: str):
|
||||
"""删除实体"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
entity = db.get_entity(entity_id)
|
||||
if not entity:
|
||||
raise HTTPException(status_code=404, detail="Entity not found")
|
||||
|
||||
db.delete_entity(entity_id)
|
||||
return {"success": True, "message": f"Entity {entity_id} deleted"}
|
||||
|
||||
@app.post("/api/v1/entities/{entity_id}/merge")
|
||||
async def merge_entities_endpoint(entity_id: str, merge_req: EntityMergeRequest):
|
||||
"""合并两个实体"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
|
||||
# 验证两个实体都存在
|
||||
source = db.get_entity(merge_req.source_entity_id)
|
||||
target = db.get_entity(merge_req.target_entity_id)
|
||||
|
||||
if not source or not target:
|
||||
raise HTTPException(status_code=404, detail="Entity not found")
|
||||
|
||||
result = db.merge_entities(merge_req.target_entity_id, merge_req.source_entity_id)
|
||||
return {
|
||||
"success": True,
|
||||
"merged_entity": {
|
||||
"id": result.id,
|
||||
"name": result.name,
|
||||
"type": result.type,
|
||||
"definition": result.definition,
|
||||
"aliases": result.aliases
|
||||
}
|
||||
}
|
||||
|
||||
# Phase 2: Relation Edit API
|
||||
@app.post("/api/v1/projects/{project_id}/relations")
|
||||
async def create_relation_endpoint(project_id: str, relation: RelationCreate):
|
||||
"""创建新的实体关系"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
|
||||
# 验证实体存在
|
||||
source = db.get_entity(relation.source_entity_id)
|
||||
target = db.get_entity(relation.target_entity_id)
|
||||
|
||||
if not source or not target:
|
||||
raise HTTPException(status_code=404, detail="Source or target entity not found")
|
||||
|
||||
relation_id = db.create_relation(
|
||||
project_id=project_id,
|
||||
source_entity_id=relation.source_entity_id,
|
||||
target_entity_id=relation.target_entity_id,
|
||||
relation_type=relation.relation_type,
|
||||
evidence=relation.evidence
|
||||
)
|
||||
|
||||
return {
|
||||
"id": relation_id,
|
||||
"source_id": relation.source_entity_id,
|
||||
"target_id": relation.target_entity_id,
|
||||
"type": relation.relation_type,
|
||||
"success": True
|
||||
}
|
||||
|
||||
@app.delete("/api/v1/relations/{relation_id}")
|
||||
async def delete_relation(relation_id: str):
|
||||
"""删除关系"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
db.delete_relation(relation_id)
|
||||
return {"success": True, "message": f"Relation {relation_id} deleted"}
|
||||
|
||||
@app.put("/api/v1/relations/{relation_id}")
|
||||
async def update_relation(relation_id: str, relation: RelationCreate):
|
||||
"""更新关系"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
updated = db.update_relation(
|
||||
relation_id=relation_id,
|
||||
relation_type=relation.relation_type,
|
||||
evidence=relation.evidence
|
||||
)
|
||||
|
||||
return {
|
||||
"id": relation_id,
|
||||
"type": updated["relation_type"],
|
||||
"evidence": updated["evidence"],
|
||||
"success": True
|
||||
}
|
||||
|
||||
# Phase 2: Transcript Edit API
|
||||
@app.get("/api/v1/transcripts/{transcript_id}")
|
||||
async def get_transcript(transcript_id: str):
|
||||
"""获取转录详情"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
transcript = db.get_transcript(transcript_id)
|
||||
|
||||
if not transcript:
|
||||
raise HTTPException(status_code=404, detail="Transcript not found")
|
||||
|
||||
return transcript
|
||||
|
||||
@app.put("/api/v1/transcripts/{transcript_id}")
|
||||
async def update_transcript(transcript_id: str, update: TranscriptUpdate):
|
||||
"""更新转录文本（人工修正）"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
transcript = db.get_transcript(transcript_id)
|
||||
|
||||
if not transcript:
|
||||
raise HTTPException(status_code=404, detail="Transcript not found")
|
||||
|
||||
updated = db.update_transcript(transcript_id, update.full_text)
|
||||
return {
|
||||
"id": transcript_id,
|
||||
"full_text": updated["full_text"],
|
||||
"updated_at": updated["updated_at"],
|
||||
"success": True
|
||||
}
|
||||
|
||||
# Phase 2: Manual Entity Creation
|
||||
class ManualEntityCreate(BaseModel):
|
||||
name: str
|
||||
type: str = "OTHER"
|
||||
definition: str = ""
|
||||
transcript_id: Optional[str] = None
|
||||
start_pos: Optional[int] = None
|
||||
end_pos: Optional[int] = None
|
||||
|
||||
@app.post("/api/v1/projects/{project_id}/entities")
|
||||
async def create_manual_entity(project_id: str, entity: ManualEntityCreate):
|
||||
"""手动创建实体（划词新建）"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
|
||||
# 检查是否已存在
|
||||
existing = db.get_entity_by_name(project_id, entity.name)
|
||||
if existing:
|
||||
return {
|
||||
"id": existing.id,
|
||||
"name": existing.name,
|
||||
"type": existing.type,
|
||||
"existed": True
|
||||
}
|
||||
|
||||
entity_id = str(uuid.uuid4())[:8]
|
||||
new_entity = db.create_entity(Entity(
|
||||
id=entity_id,
|
||||
project_id=project_id,
|
||||
name=entity.name,
|
||||
type=entity.type,
|
||||
definition=entity.definition
|
||||
))
|
||||
|
||||
# 如果有提及位置信息，保存提及
|
||||
if entity.transcript_id and entity.start_pos is not None and entity.end_pos is not None:
|
||||
transcript = db.get_transcript(entity.transcript_id)
|
||||
if transcript:
|
||||
text = transcript["full_text"]
|
||||
mention = EntityMention(
|
||||
id=str(uuid.uuid4())[:8],
|
||||
entity_id=entity_id,
|
||||
transcript_id=entity.transcript_id,
|
||||
start_pos=entity.start_pos,
|
||||
end_pos=entity.end_pos,
|
||||
text_snippet=text[max(0, entity.start_pos-20):min(len(text), entity.end_pos+20)],
|
||||
confidence=1.0
|
||||
)
|
||||
db.add_mention(mention)
|
||||
|
||||
return {
|
||||
"id": new_entity.id,
|
||||
"name": new_entity.name,
|
||||
"type": new_entity.type,
|
||||
"definition": new_entity.definition,
|
||||
"success": True
|
||||
}
|
||||
|
||||
def transcribe_audio(audio_data: bytes, filename: str) -> dict:
|
||||
"""转录音频：OSS上传 + 听悟转录"""
|
||||
|
||||
@@ -379,14 +620,31 @@ async def get_entity_mentions(entity_id: str):
|
||||
} for m in mentions]
|
||||
|
||||
@app.post("/api/v1/entities/{entity_id}/merge")
|
||||
async def merge_entities(entity_id: str, target_entity_id: str):
|
||||
async def merge_entities_endpoint(entity_id: str, merge_req: EntityMergeRequest):
|
||||
"""合并两个实体"""
|
||||
if not DB_AVAILABLE:
|
||||
raise HTTPException(status_code=500, detail="Database not available")
|
||||
|
||||
db = get_db_manager()
|
||||
result = db.merge_entities(target_entity_id, entity_id)
|
||||
return {"success": True, "merged_entity": {"id": result.id, "name": result.name}}
|
||||
|
||||
# 验证两个实体都存在
|
||||
source = db.get_entity(merge_req.source_entity_id)
|
||||
target = db.get_entity(merge_req.target_entity_id)
|
||||
|
||||
if not source or not target:
|
||||
raise HTTPException(status_code=404, detail="Entity not found")
|
||||
|
||||
result = db.merge_entities(merge_req.target_entity_id, merge_req.source_entity_id)
|
||||
return {
|
||||
"success": True,
|
||||
"merged_entity": {
|
||||
"id": result.id,
|
||||
"name": result.name,
|
||||
"type": result.type,
|
||||
"definition": result.definition,
|
||||
"aliases": result.aliases
|
||||
}
|
||||
}
|
||||
|
||||
# Health check
|
||||
@app.get("/health")
|
||||
|
||||
Reference in New Issue
Block a user