95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
|
package handler
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"tcm-agent/internal/agent"
|
|||
|
|
|
|||
|
|
"github.com/gin-gonic/gin"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// KnowledgeHandler 知识库接口处理器
|
|||
|
|
// 提供：检索知识库、上传文档等能力
|
|||
|
|
type KnowledgeHandler struct {
|
|||
|
|
agentRunner *agent.Runner
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewKnowledgeHandler 创建知识库处理器
|
|||
|
|
func NewKnowledgeHandler(runner *agent.Runner) *KnowledgeHandler {
|
|||
|
|
return &KnowledgeHandler{agentRunner: runner}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// SearchRequest 知识检索请求
|
|||
|
|
type SearchRequest struct {
|
|||
|
|
Query string `json:"query" binding:"required"` // 检索关键词
|
|||
|
|
TopK int `json:"top_k"` // 返回条数（默认5）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Search 检索知识库
|
|||
|
|
// POST /api/v1/knowledge/search
|
|||
|
|
//
|
|||
|
|
// 直接调用MaxKB的RAG检索能力
|
|||
|
|
// 用于：查询方剂组成、药典条目、诊疗规范等
|
|||
|
|
func (h *KnowledgeHandler) Search(c *gin.Context) {
|
|||
|
|
var req SearchRequest
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
c.JSON(400, gin.H{"error": "查询关键词不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if req.TopK <= 0 {
|
|||
|
|
req.TopK = 5
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 调用MaxKB检索
|
|||
|
|
answer, err := h.agentRunner.MaxKB().Chat(c.Request.Context(), req.Query)
|
|||
|
|
if err != nil {
|
|||
|
|
c.JSON(500, gin.H{
|
|||
|
|
"error": "知识库检索失败",
|
|||
|
|
"detail": err.Error(),
|
|||
|
|
})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
c.JSON(200, gin.H{
|
|||
|
|
"code": 200,
|
|||
|
|
"data": gin.H{
|
|||
|
|
"query": req.Query,
|
|||
|
|
"answer": answer,
|
|||
|
|
"source": "MaxKB RAG",
|
|||
|
|
"top_k": req.TopK,
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IngestRequest 文档上传请求
|
|||
|
|
type IngestRequest struct {
|
|||
|
|
Title string `json:"title" binding:"required"` // 文档标题
|
|||
|
|
Content string `json:"content" binding:"required"` // 文档内容
|
|||
|
|
Category string `json:"category"` // 分类（方剂/药典/指南/病历模板）
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Ingest 上传文档到知识库
|
|||
|
|
// POST /api/v1/knowledge/ingest
|
|||
|
|
//
|
|||
|
|
// 将文档写入MaxKB知识库，触发自动向量化
|
|||
|
|
// 支持后续RAG检索
|
|||
|
|
func (h *KnowledgeHandler) Ingest(c *gin.Context) {
|
|||
|
|
var req IngestRequest
|
|||
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|||
|
|
c.JSON(400, gin.H{"error": "标题和内容不能为空"})
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 实际项目中调用MaxKB的文档上传API
|
|||
|
|
// 这里返回模拟成功
|
|||
|
|
c.JSON(200, gin.H{
|
|||
|
|
"code": 200,
|
|||
|
|
"message": "文档已加入知识库队列",
|
|||
|
|
"data": gin.H{
|
|||
|
|
"title": req.Title,
|
|||
|
|
"category": req.Category,
|
|||
|
|
"status": "pending_vectorization",
|
|||
|
|
"note": "实际项目请调用MaxKB文档上传API",
|
|||
|
|
},
|
|||
|
|
})
|
|||
|
|
}
|