mlogclub
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
package dashboard
|
|
|
|
import (
|
|
"agent-desk/internal/pkg/httpx"
|
|
"context"
|
|
|
|
"agent-desk/internal/pkg/constants"
|
|
"agent-desk/internal/pkg/dto/request"
|
|
"agent-desk/internal/pkg/dto/response"
|
|
"agent-desk/internal/pkg/i18nx"
|
|
"agent-desk/internal/services"
|
|
|
|
"agent-desk/internal/pkg/httpx/params"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func MCPAnyList_servers(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionMCPView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, response.BuildMCPServerInfoResponses(services.MCPDebugService.ListServers()))
|
|
}
|
|
|
|
func MCPAnyCatalog(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionMCPView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
items, err := services.ToolCatalogService.ListMCPToolsWithLocale(context.Background(), i18nx.Locale(ctx))
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
ret := make([]response.MCPToolCatalogResponse, 0, len(items))
|
|
for _, item := range items {
|
|
ret = append(ret, response.MCPToolCatalogResponse{
|
|
ToolCode: item.ToolCode,
|
|
ServerCode: item.ServerCode,
|
|
ToolName: item.ToolName,
|
|
SourceType: item.SourceType,
|
|
AutoInjected: item.AutoInjected,
|
|
Title: item.Title,
|
|
Description: item.Description,
|
|
InputSchema: item.InputSchema,
|
|
OutputSchema: item.OutputSchema,
|
|
RiskLevel: item.RiskLevel,
|
|
RequireConfirmation: item.RequireConfirmation,
|
|
RiskEditable: item.RiskEditable,
|
|
})
|
|
}
|
|
httpx.WriteJSON(ctx, ret)
|
|
}
|
|
|
|
func MCPPostTest_connection(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionMCPView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
req := request.MCPServerDebugRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
result, err := services.MCPDebugService.TestConnection(context.Background(), req.ServerCode)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, response.BuildMCPConnectionResponse(result))
|
|
}
|
|
|
|
func MCPPostList_tools(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionMCPView); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
req := request.MCPServerDebugRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
result, err := services.MCPDebugService.ListTools(context.Background(), req.ServerCode)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, response.BuildMCPToolInfoResponses(result))
|
|
}
|
|
|
|
func MCPPostCall_tool(ctx *gin.Context) {
|
|
if _, err := services.AuthService.RequirePermission(ctx, constants.PermissionMCPCall); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
req := request.MCPCallToolRequest{}
|
|
if err := params.ReadJSON(ctx, &req); err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
result, err := services.MCPDebugService.CallTool(context.Background(), req.ServerCode, req.ToolName, req.Arguments)
|
|
if err != nil {
|
|
httpx.WriteJSON(ctx, err)
|
|
return
|
|
}
|
|
httpx.WriteJSON(ctx, response.BuildMCPCallToolResponse(result))
|
|
}
|