feat: 实现完整的人工审核功能与子图模块
- 新增三个核心子图:人工审核、意图理解、格式化输出 - 实现完整的审核 API 端点(/api/review/*) - 前端添加审核确认界面(右下角固定框) - 为每个子图创建分步测试代码 - 添加功能实现文档
This commit is contained in:
@@ -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 端口冲突)
|
||||
|
||||
Reference in New Issue
Block a user