mlogclub
5d7c10aeab
- Updated runtime configuration to use __CS_AI_AGENT_WIDGET_CONFIG__ instead of __CS_AGENT_WIDGET_CONFIG__. - Changed message types in support host bridge from "cs-agent" to "cs-ai-agent". - Minified SDK script updated to reflect new AI agent naming conventions. - Adjusted scrollbar styles in main.scss to use .cs-ai-agent-scrollbar instead of .cs-agent-scrollbar.
83 lines
2.6 KiB
Go
83 lines
2.6 KiB
Go
package dashboard
|
|
|
|
import (
|
|
"cs-ai-agent/internal/builders"
|
|
"cs-ai-agent/internal/pkg/constants"
|
|
"cs-ai-agent/internal/pkg/dto/response"
|
|
"cs-ai-agent/internal/pkg/httpx"
|
|
"cs-ai-agent/internal/services"
|
|
|
|
"cs-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: "knowledgeBaseId"},
|
|
params.QueryFilter{ParamName: "question", Op: params.Like},
|
|
params.QueryFilter{ParamName: "channel"},
|
|
params.QueryFilter{ParamName: "scene"},
|
|
params.QueryFilter{ParamName: "chunkProvider"},
|
|
).Desc("id")
|
|
|
|
if answerStatus, ok := params.GetInt64(ctx, "answerStatus"); ok && answerStatus > 0 {
|
|
cnd.Where("answer_status = ?", answerStatus)
|
|
}
|
|
if rerankEnabled, ok := params.GetInt64(ctx, "rerankEnabled"); 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, web.JsonErrorMsg("检索日志不存在"))
|
|
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,
|
|
})
|
|
}
|