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.
97 lines
3.0 KiB
Go
97 lines
3.0 KiB
Go
package toolx
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
|
|
"cs-ai-agent/internal/pkg/dto/request"
|
|
"cs-ai-agent/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.InvalidParam("Direct Tool 的 toolCode 与 serverCode 不一致")
|
|
}
|
|
if toolName != "" && !strings.EqualFold(toolName, parsedToolName) {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParam("Direct Tool 的 toolCode 与 toolName 不一致")
|
|
}
|
|
serverCode = parsedServerCode
|
|
toolName = parsedToolName
|
|
} else {
|
|
serverCode = ""
|
|
toolName = ""
|
|
}
|
|
} else {
|
|
toolCode = BuildMCPToolCode(serverCode, toolName)
|
|
}
|
|
if toolCode == "" {
|
|
return request.AIAgentMCPToolRequest{}, errorsx.InvalidParam("Direct Tool 的 toolCode、serverCode 和 toolName 不能为空")
|
|
}
|
|
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.InvalidParam("Direct Tool 的 toolCode 格式不合法")
|
|
}
|
|
ret := request.AIAgentMCPToolRequest{
|
|
ToolCode: toolCode,
|
|
ServerCode: serverCode,
|
|
ToolName: toolName,
|
|
Title: strings.TrimSpace(item.Title),
|
|
Description: strings.TrimSpace(item.Description),
|
|
}
|
|
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
|
|
}
|