mlogclub
34051a4631
- Updated labels in the AI Agents dashboard for clarity, changing "流程状态" to "Playbook 状态" and "未发布流程" to "未发布 Playbook". - Introduced AI Agent rollout percentage management in channel editing, allowing users to set and rollback rollout percentages. - Added new API endpoints for rolling back AI Agent rollout and fetching agent run metrics. - Implemented new UI components for displaying agent run details, including status, duration, and input/output tokens. - Enhanced type definitions for AdminChannel and AIAgent to include rollout percentages and runtime modes. - Updated navigation to include a section for agent runs. - Added new translations for agent run features in both English and Chinese.
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
package dashboard
|
|
|
|
import (
|
|
"agent-desk/internal/builders"
|
|
"agent-desk/internal/pkg/constants"
|
|
"agent-desk/internal/pkg/dto/request"
|
|
"agent-desk/internal/pkg/httpx"
|
|
"agent-desk/internal/pkg/httpx/params"
|
|
"agent-desk/internal/services"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/mlogclub/simple/web"
|
|
)
|
|
|
|
func AgentRunAnyList(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
queryParams := params.NewQueryParams(ctx)
|
|
queryParams.Cnd = *params.NewPagedSqlCnd(ctx,
|
|
params.QueryFilter{ParamName: "conversationId"},
|
|
params.QueryFilter{ParamName: "aiAgentId"},
|
|
params.QueryFilter{ParamName: "agentRevisionId"},
|
|
params.QueryFilter{ParamName: "sourceMessageId"},
|
|
params.QueryFilter{ParamName: "workflowRunId"},
|
|
params.QueryFilter{ParamName: "engineCode"},
|
|
params.QueryFilter{ParamName: "status"},
|
|
).Desc("id")
|
|
list, paging := services.AgentRunService.FindPageByParams(queryParams)
|
|
httpx.WriteJSON(ctx, &web.PageResult{Results: builders.BuildAgentRunList(list), Page: paging})
|
|
}
|
|
|
|
func AgentRunGetBy(ctx *gin.Context) {
|
|
id, ok := httpx.GetPathInt64(ctx, "id")
|
|
if !ok {
|
|
return
|
|
}
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
run, steps, toolCalls := services.AgentRunService.GetDetail(id)
|
|
if run == nil {
|
|
httpx.WriteJSON(ctx, httpx.JsonErrorMsg(ctx, "error.e0002"))
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, builders.BuildAgentRunDetail(run, steps, toolCalls, services.AgentRunService.GetQualityFeedback(run.ID)))
|
|
}
|
|
|
|
func AgentRunPostSave_quality_feedback(ctx *gin.Context) {
|
|
operator, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentUpdate)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
req := request.SaveAgentRunQualityFeedbackRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
if err := services.AgentRunService.SaveQualityFeedback(req, operator); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, nil)
|
|
}
|
|
|
|
func AgentRunAnyMetrics(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
aiAgentID, _ := params.GetInt64(ctx, "aiAgentId")
|
|
httpx.WriteJSON(ctx, services.AgentRunService.GetMetrics(aiAgentID))
|
|
}
|
|
|
|
func AgentRunAnyComparison(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
aiAgentID, _ := params.GetInt64(ctx, "aiAgentId")
|
|
httpx.WriteJSON(ctx, services.AgentRunService.GetEngineComparisons(aiAgentID))
|
|
}
|
|
|
|
func AgentRunPostEvaluate(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionAIAgentView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
req := request.RunAgentEvaluationRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
result, err := services.AgentEvaluationService.Run(ctx, req)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, result)
|
|
}
|