t
18c9354095
- 注入数据库、运行时配置、统一响应、文件存储和平台 AI 能力，补充业务读写工具与客户快捷操作契约。 - 移除模块内重复的组织、客户、工单、标签、技能、旧工作流、MCP 和迁移实现，将身份权限与业务主体交由宿主管理。 - 使用 libSQL 重构向量存储，并完善图片消息、访客身份、排队调度、企业微信和支持聊天页面。 - 统一 HTTP、DTO 与 WebSocket 的 snake_case 协议，补齐模块初始化、业务动作和公共载荷等回归测试。
83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package dashboard
|
|
|
|
import (
|
|
"code.tczkiot.com/wlw/ai-agent/internal/builders"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/constants"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/dto/response"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/httpx"
|
|
"code.tczkiot.com/wlw/ai-agent/internal/services"
|
|
|
|
"code.tczkiot.com/wlw/ai-agent/internal/pkg/httpx/params"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/mlogclub/simple/web"
|
|
)
|
|
|
|
func KnowledgeRetrieveLogAnyList(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeDocumentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
|
|
cnd := params.NewPagedSqlCnd(ctx,
|
|
params.QueryFilter{ParamName: "knowledge_base_id"},
|
|
params.QueryFilter{ParamName: "question", Op: params.Like},
|
|
params.QueryFilter{ParamName: "channel"},
|
|
params.QueryFilter{ParamName: "scene"},
|
|
params.QueryFilter{ParamName: "chunk_provider"},
|
|
).Desc("id")
|
|
|
|
if answerStatus, ok := params.GetInt64(ctx, "answer_status"); ok && answerStatus > 0 {
|
|
cnd.Where("answer_status = ?", answerStatus)
|
|
}
|
|
if rerankEnabled, ok := params.GetInt64(ctx, "rerank_enabled"); ok {
|
|
cnd.Where("rerank_enabled = ?", rerankEnabled > 0)
|
|
}
|
|
|
|
queryParams := params.NewQueryParams(ctx)
|
|
queryParams.Cnd = *cnd
|
|
list, paging := services.KnowledgeRetrieveLogService.FindPageByParams(queryParams)
|
|
results := make([]response.KnowledgeRetrieveLogResponse, 0, len(list))
|
|
for _, item := range list {
|
|
resp := builders.BuildKnowledgeRetrieveLog(&item)
|
|
if knowledgeBase := services.KnowledgeBaseService.Get(item.KnowledgeBaseID); knowledgeBase != nil {
|
|
resp.KnowledgeBaseName = knowledgeBase.Name
|
|
}
|
|
results = append(results, resp)
|
|
}
|
|
httpx.WriteJSON(ctx, &web.PageResult{Results: results, Page: paging})
|
|
}
|
|
|
|
func KnowledgeRetrieveLogGetBy(ctx *gin.Context) {
|
|
id, ok := httpx.GetPathInt64(ctx, "id")
|
|
if !ok {
|
|
return
|
|
}
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionKnowledgeDocumentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
|
|
logItem := services.KnowledgeRetrieveLogService.Get(id)
|
|
if logItem == nil {
|
|
httpx.WriteJSON(ctx, httpx.JsonErrorMsg(ctx, "error.e0241"))
|
|
return
|
|
}
|
|
|
|
logResp := builders.BuildKnowledgeRetrieveLog(logItem)
|
|
if knowledgeBase := services.KnowledgeBaseService.Get(logItem.KnowledgeBaseID); knowledgeBase != nil {
|
|
logResp.KnowledgeBaseName = knowledgeBase.Name
|
|
}
|
|
|
|
hits := services.KnowledgeRetrieveLogService.FindHitsByRetrieveLogID(id)
|
|
hitResults := make([]response.KnowledgeRetrieveHitResponse, 0, len(hits))
|
|
for _, item := range hits {
|
|
hitResults = append(hitResults, builders.BuildKnowledgeRetrieveHitResponse(&item))
|
|
}
|
|
|
|
httpx.WriteJSON(ctx, response.KnowledgeRetrieveLogDetailResponse{
|
|
Log: logResp,
|
|
Hits: hitResults,
|
|
})
|
|
}
|