106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package handler
|
||
|
||
import (
|
||
"tcm-agent/internal/agent"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// ========================================================================
|
||
// 通用 Agent 对话接口
|
||
// ========================================================================
|
||
// 用于：自由问诊、多轮对话、调试 Agent 行为
|
||
// ========================================================================
|
||
|
||
// AgentHandler 通用 Agent 对话处理器
|
||
type AgentHandler struct {
|
||
runner *agent.Runner
|
||
}
|
||
|
||
// NewAgentHandler 创建 Agent 处理器
|
||
func NewAgentHandler(runner *agent.Runner) *AgentHandler {
|
||
return &AgentHandler{runner: runner}
|
||
}
|
||
|
||
// ChatRequest 对话请求
|
||
type ChatRequest struct {
|
||
SessionID string `json:"session_id"` // 可选，不传则创建新会话
|
||
Message string `json:"message" binding:"required"`
|
||
UserID string `json:"user_id"` // 用户 ID
|
||
Scene string `json:"scene"` // 场景名（决定用哪个模型）
|
||
}
|
||
|
||
// Chat 通用 Agent 对话接口
|
||
//
|
||
// POST /api/v1/agent/chat
|
||
//
|
||
// 支持多轮对话：传 session_id 则延续上下文。
|
||
// 不传 scene 时使用默认路由。
|
||
func (h *AgentHandler) Chat(c *gin.Context) {
|
||
var req ChatRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
c.JSON(400, gin.H{"error": "消息内容不能为空"})
|
||
return
|
||
}
|
||
|
||
// 获取或创建会话
|
||
sessionID := req.SessionID
|
||
if sessionID == "" {
|
||
userID := req.UserID
|
||
if userID == "" {
|
||
userID = c.GetString("user_id")
|
||
}
|
||
// 使用请求的 scene（为空则用默认）
|
||
session := h.runner.CreateSession(userID, req.Scene)
|
||
sessionID = session.ID
|
||
}
|
||
|
||
// 执行 Agent 推理
|
||
response, err := h.runner.Run(c.Request.Context(), sessionID, req.Message)
|
||
if err != nil {
|
||
c.JSON(500, gin.H{
|
||
"error": "Agent 执行失败",
|
||
"detail": err.Error(),
|
||
})
|
||
return
|
||
}
|
||
|
||
session, _ := h.runner.GetSession(sessionID)
|
||
|
||
c.JSON(200, gin.H{
|
||
"code": 200,
|
||
"data": gin.H{
|
||
"session_id": sessionID,
|
||
"response": response,
|
||
"history_len": len(session.History),
|
||
"status": session.Status,
|
||
},
|
||
})
|
||
}
|
||
|
||
// GetSession 查看会话详情
|
||
//
|
||
// GET /api/v1/agent/session/:id
|
||
func (h *AgentHandler) GetSession(c *gin.Context) {
|
||
id := c.Param("id")
|
||
session, ok := h.runner.GetSession(id)
|
||
if !ok {
|
||
c.JSON(404, gin.H{"error": "会话不存在"})
|
||
return
|
||
}
|
||
|
||
c.JSON(200, gin.H{
|
||
"code": 200,
|
||
"data": gin.H{
|
||
"id": session.ID,
|
||
"user_id": session.UserID,
|
||
"scene": session.Scene,
|
||
"status": session.Status,
|
||
"history": session.History,
|
||
"state": session.State,
|
||
"created_at": session.CreatedAt,
|
||
"updated_at": session.UpdatedAt,
|
||
},
|
||
})
|
||
}
|