mlogclub
847f688398
- Removed runtime mode handling from AIAgentConfigWorkbench and related components. - Updated tests to reflect changes in AI Agent policy copy and configuration. - Changed terminology from "workflow" to "revision" in various components and API responses. - Simplified agent binding logic in channel editing. - Cleaned up unused variables and types related to runtime modes. - Updated localization files for consistency with new terminology.
99 lines
3.0 KiB
Go
99 lines
3.0 KiB
Go
package toolx
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
|
|
"agent-desk/internal/pkg/dto/request"
|
|
"agent-desk/internal/pkg/errorsx"
|
|
)
|
|
|
|
func BuildMCPToolCode(serverCode, toolName string) string {
|
|
serverCode = strings.TrimSpace(serverCode)
|
|
toolName = strings.TrimSpace(toolName)
|
|
if serverCode == "" || toolName == "" {
|
|
return ""
|
|
}
|
|
return serverCode + "/" + toolName
|
|
}
|
|
|
|
func SplitMCPToolCode(toolCode string) (string, string) {
|
|
toolCode = strings.TrimSpace(toolCode)
|
|
if toolCode == "" {
|
|
return "", ""
|
|
}
|
|
idx := strings.Index(toolCode, "/")
|
|
if idx <= 0 || idx >= len(toolCode)-1 {
|
|
return "", ""
|
|
}
|
|
return strings.TrimSpace(toolCode[:idx]), strings.TrimSpace(toolCode[idx+1:])
|
|
}
|
|
|
|
func NormalizeMCPToolRequest(item request.AIAgentMCPToolRequest) (request.AIAgentMCPToolRequest, error) {
|
|
toolCode := strings.TrimSpace(item.ToolCode)
|
|
toolCode = NormalizeToolCodeAlias(toolCode)
|
|
serverCode := strings.TrimSpace(item.ServerCode)
|
|
toolName := strings.TrimSpace(item.ToolName)
|
|
if toolCode != "" {
|
|
parsedServerCode, parsedToolName := SplitMCPToolCode(toolCode)
|
|
if parsedServerCode != "" && parsedToolName != "" {
|
|
if serverCode != "" && !strings.EqualFold(serverCode, parsedServerCode) {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParamI18n("error.e0016")
|
|
}
|
|
if toolName != "" && !strings.EqualFold(toolName, parsedToolName) {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParamI18n("error.e0017")
|
|
}
|
|
serverCode = parsedServerCode
|
|
toolName = parsedToolName
|
|
} else {
|
|
serverCode = ""
|
|
toolName = ""
|
|
}
|
|
} else {
|
|
toolCode = BuildMCPToolCode(serverCode, toolName)
|
|
}
|
|
if toolCode == "" {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParamI18n("error.e0019")
|
|
}
|
|
if parsedServerCode, parsedToolName := SplitMCPToolCode(toolCode); parsedServerCode != "" && parsedToolName != "" {
|
|
serverCode = parsedServerCode
|
|
toolName = parsedToolName
|
|
}
|
|
if serverCode == "" && toolName == "" && strings.Contains(toolCode, "/") && !strings.HasPrefix(toolCode, "builtin/") {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParamI18n("error.e0018")
|
|
}
|
|
ret := request.AIAgentMCPToolRequest{
|
|
ToolCode: toolCode,
|
|
ServerCode: serverCode,
|
|
ToolName: toolName,
|
|
Title: strings.TrimSpace(item.Title),
|
|
Description: strings.TrimSpace(item.Description),
|
|
RiskLevel: strings.ToLower(strings.TrimSpace(item.RiskLevel)),
|
|
RequireConfirmation: item.RequireConfirmation,
|
|
}
|
|
if len(item.Arguments) > 0 {
|
|
ret.Arguments = make(map[string]string, len(item.Arguments))
|
|
for key, value := range item.Arguments {
|
|
key = strings.TrimSpace(key)
|
|
value = strings.TrimSpace(value)
|
|
if key == "" || value == "" {
|
|
continue
|
|
}
|
|
ret.Arguments[key] = value
|
|
}
|
|
}
|
|
return ret, nil
|
|
}
|
|
|
|
func ParseAgentMCPToolsJSON(raw string) ([]request.AIAgentMCPToolRequest, error) {
|
|
raw = strings.TrimSpace(raw)
|
|
if raw == "" {
|
|
return nil, nil
|
|
}
|
|
var ret []request.AIAgentMCPToolRequest
|
|
if err := json.Unmarshal([]byte(raw), &ret); err != nil {
|
|
return nil, err
|
|
}
|
|
return ret, nil
|
|
}
|