feat: 实现完整的人工审核功能与子图模块

- 新增三个核心子图:人工审核、意图理解、格式化输出
- 实现完整的审核 API 端点(/api/review/*)
- 前端添加审核确认界面(右下角固定框)
- 为每个子图创建分步测试代码
- 添加功能实现文档
This commit is contained in:
2026-04-25 13:24:50 +08:00
parent 851d52ed8d
commit bc26b81f08
9 changed files with 1981 additions and 4 deletions

View File

@@ -8,6 +8,7 @@ from .config import DB_URI, BACKEND_PORT
import uuid
import json
from contextlib import asynccontextmanager
from typing import Optional
from fastapi import FastAPI, HTTPException, WebSocket, WebSocketDisconnect, Depends, Request, Query
from fastapi.middleware.cors import CORSMiddleware
@@ -16,6 +17,12 @@ from pydantic import BaseModel
from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
from .agent.service import AIAgentService
from .agent.history import ThreadHistoryService
from .agent_subgraphs.common.human_review import (
ReviewManager,
InMemoryReviewStore,
ReviewStatus,
HumanReview
)
from .logger import info, error
@asynccontextmanager
@@ -32,14 +39,18 @@ async def lifespan(app: FastAPI):
# 3. 创建历史查询服务
history_service = ThreadHistoryService(checkpointer)
# 4. 将服务实例存入 app.state
# 4. 创建审核管理器
review_manager = ReviewManager(InMemoryReviewStore())
# 5. 将服务实例存入 app.state
app.state.agent_service = agent_service
app.state.history_service = history_service
app.state.review_manager = review_manager
# 应用运行中...
yield
# 5. 关闭时自动清理数据库连接async with 负责)
# 6. 关闭时自动清理数据库连接async with 负责)
info("🛑 应用关闭,数据库连接池已释放")
app = FastAPI(lifespan=lifespan)
@@ -75,6 +86,23 @@ class ChatResponse(BaseModel):
total_tokens: int = 0
elapsed_time: float = 0.0
class ReviewActionRequest(BaseModel):
review_id: str
reviewer: str
comment: str = ""
modified_content: str = ""
class ReviewResponse(BaseModel):
review_id: str
thread_id: str
user_id: str
status: str
content_to_review: str
review_comment: str = ""
modified_content: str = ""
created_at: str
reviewed_at: Optional[str] = None
# ========== 依赖注入函数 ==========
def get_agent_service(request: Request) -> AIAgentService:
"""从 app.state 中获取全局 AIAgentService 实例"""
@@ -84,6 +112,10 @@ def get_history_service(request: Request) -> ThreadHistoryService:
"""从 app.state 中获取全局 ThreadHistoryService 实例"""
return request.app.state.history_service
def get_review_manager(request: Request) -> ReviewManager:
"""从 app.state 中获取全局 ReviewManager 实例"""
return request.app.state.review_manager
# ========== HTTP 端点 ==========
@app.post("/chat", response_model=ChatResponse)
async def chat_endpoint(
@@ -205,6 +237,114 @@ async def websocket_endpoint(
except WebSocketDisconnect:
pass
# ========== 审核相关端点 ==========
def review_to_response(review: HumanReview) -> ReviewResponse:
"""将审核对象转换为响应对象"""
return ReviewResponse(
review_id=review.review_id,
thread_id=review.thread_id,
user_id=review.user_id,
status=review.status.name,
content_to_review=review.content_to_review,
review_comment=review.review_comment,
modified_content=review.modified_content,
created_at=review.created_at.isoformat(),
reviewed_at=review.reviewed_at.isoformat() if review.reviewed_at else None
)
@app.get("/reviews/pending", response_model=list[ReviewResponse])
async def get_pending_reviews(
limit: int = Query(100, ge=1, le=500, description="返回数量限制"),
review_manager: ReviewManager = Depends(get_review_manager)
):
"""获取待审核列表"""
reviews = review_manager.get_pending_reviews(limit)
return [review_to_response(review) for review in reviews]
@app.get("/reviews/{review_id}", response_model=ReviewResponse)
async def get_review(
review_id: str,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""获取审核详情"""
review = review_manager.get_review(review_id)
if not review:
raise HTTPException(status_code=404, detail="Review not found")
return review_to_response(review)
@app.get("/reviews/thread/{thread_id}", response_model=list[ReviewResponse])
async def get_thread_reviews(
thread_id: str,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""获取线程的所有审核"""
# 注意:我们的 ReviewStore 接口目前没有 get_by_thread 方法暴露在 ReviewManager 中
# 这里我们直接访问 store但在实际项目中应该在 ReviewManager 中添加这个方法
reviews = review_manager.store.get_by_thread(thread_id) if hasattr(review_manager.store, 'get_by_thread') else []
return [review_to_response(review) for review in reviews]
@app.post("/reviews/{review_id}/approve")
async def approve_review(
review_id: str,
request: ReviewActionRequest,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""审核通过"""
success = review_manager.approve(
review_id=review_id,
reviewer=request.reviewer,
comment=request.comment
)
if not success:
raise HTTPException(status_code=404, detail="Review not found")
return {"status": "success", "review_id": review_id}
@app.post("/reviews/{review_id}/reject")
async def reject_review(
review_id: str,
request: ReviewActionRequest,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""审核拒绝"""
success = review_manager.reject(
review_id=review_id,
reviewer=request.reviewer,
comment=request.comment
)
if not success:
raise HTTPException(status_code=404, detail="Review not found")
return {"status": "success", "review_id": review_id}
@app.post("/reviews/{review_id}/modify")
async def modify_review(
review_id: str,
request: ReviewActionRequest,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""审核修改"""
if not request.modified_content:
raise HTTPException(status_code=400, detail="modified_content required")
success = review_manager.modify(
review_id=review_id,
reviewer=request.reviewer,
modified_content=request.modified_content,
comment=request.comment
)
if not success:
raise HTTPException(status_code=404, detail="Review not found")
return {"status": "success", "review_id": review_id}
@app.post("/reviews/request")
async def request_review(
thread_id: str,
user_id: str,
content: str,
review_manager: ReviewManager = Depends(get_review_manager)
):
"""请求审核(测试用)"""
review_id = review_manager.request_review(thread_id, user_id, content)
return {"status": "success", "review_id": review_id}
if __name__ == "__main__":
import uvicorn
# 使用环境变量或默认端口 8079避免与 llama.cpp 的 8081 端口冲突)