首页知识库对话-rg

zhaoqingang

2025-03-18 282a631b9ceee9a634ee1d93751a5254ed37ccef

 app/api/v2/mindmap.py

@@ -14,7 +14,7 @@
from app.models.v2.chat import RetrievalRequest, ComplexChatDao
from app.models.v2.mindmap import MindmapRequest
from app.models.v2.session_model import ChatData
from app.service.v2.mindmap import service_chat_mindmap
from app.service.v2.mindmap import service_chat_mindmap, service_message_mindmap_parse

mind_map_router = APIRouter()

@@ -28,4 +28,13 @@
            return Response(code=500, msg="create failure", data={})
    else:
        return Response(code=500, msg="网络异常！failure", data={})
    return Response(code=200, msg="create success", data=data)


@mind_map_router.get("/{messageId}/parse", response_model=Response)
async def api_chat_mindmap(messageId: str, current_user: UserModel = Depends(get_current_user), db: Session = Depends(get_db)): #  current_user: UserModel = Depends(get_current_user)

    data = await service_message_mindmap_parse(db, messageId, current_user.id)
    if not data:
        return Response(code=500, msg="create failure", data={})
    return Response(code=200, msg="create success", data=data)