mlogclub
0b4a1b4594
- Updated OSSStorage validation errors to use internationalized messages. - Changed error messages in provider.go for unsupported file storage types. - Refactored tag_service.go to replace hardcoded error messages with internationalized versions. - Updated ticket_service.go to use internationalized error messages for various validation checks. - Refactored ticket_tag_service.go to use internationalized error messages for tag validation. - Changed ticket_view_service.go to use internationalized error messages for view validation. - Updated tool_catalog_service.go to use internationalized error messages for tool code validation. - Refactored user_service.go to replace error messages with internationalized versions. - Updated ws_service.go to use internationalized error messages for WebSocket handling. - Refactored wxwork_kf_inbound_service.go to use internationalized error messages for message handling. - Updated wxwork_kf_outbound_service.go to use internationalized error messages for outbound message handling. - Refactored wxwork_login_service.go to use internationalized error messages for login handling. - Updated login.go in wxwork package to use internationalized error messages for login state and ticket validation.
97 lines
2.9 KiB
Go
97 lines
2.9 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),
|
|
}
|
|
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
|
|
}
|