chore: import clean Agent Desk snapshot (2/7)

internal/ai/runtime/registry/registry.go

@@ -0,0 +1,103 @@
+package registry
+
+import (
+	"strings"
+
+	"agent-desk/internal/pkg/toolx"
+
+	einotool "github.com/cloudwego/eino/components/tool"
+)
+
+type Registry struct {
+	tools []Tool
+}
+
+func NewRegistry(tools ...Tool) *Registry {
+	return &Registry{
+		tools: tools,
+	}
+}
+
+func (r *Registry) Resolve(ctx Context) (*ToolSet, error) {
+	ret := &ToolSet{
+		StaticTools:        make([]einotool.BaseTool, 0, len(r.tools)),
+		StaticToolCodes:    make(map[string]string),
+		StaticToolMetadata: make(map[string]ToolMetadata),
+	}
+	allowedToolCodes := makeAllowedToolCodeSet(ctx.AllowedToolCodes)
+	for _, toolDef := range r.tools {
+		if toolDef == nil || !toolDef.Enabled(ctx) {
+			continue
+		}
+		spec := toolDef.Spec()
+		toolCode := strings.TrimSpace(spec.Code)
+		if toolCode == "" {
+			toolCode = strings.TrimSpace(toolDef.Code())
+		}
+		if len(allowedToolCodes) > 0 && !isAllowedToolCode(toolCode, allowedToolCodes) {
+			continue
+		}
+		tool, err := toolDef.Build(ctx)
+		if err != nil {
+			return nil, err
+		}
+		if tool == nil {
+			continue
+		}
+		toolName := strings.TrimSpace(spec.Name)
+		if toolName == "" {
+			toolName = strings.TrimSpace(toolDef.Name())
+		}
+		if toolName == "" || toolCode == "" {
+			continue
+		}
+		ret.StaticTools = append(ret.StaticTools, tool)
+		ret.StaticToolCodes[toolName] = toolCode
+		resolvedMetadata := toolx.ResolveToolMetadata(toolCode, toolName)
+		if strings.TrimSpace(resolvedMetadata.ServerCode) == "" {
+			resolvedMetadata.ServerCode = strings.TrimSpace(spec.ServerCode)
+		}
+		if strings.TrimSpace(resolvedMetadata.ToolName) == "" {
+			resolvedMetadata.ToolName = toolName
+		}
+		if resolvedMetadata.SourceType == "" {
+			resolvedMetadata.SourceType = spec.SourceType
+		}
+		ret.StaticToolMetadata[toolName] = ToolMetadata{
+			ToolCode:   resolvedMetadata.ToolCode,
+			ServerCode: resolvedMetadata.ServerCode,
+			ToolName:   resolvedMetadata.ToolName,
+			SourceType: resolvedMetadata.SourceType,
+		}
+	}
+	return ret, nil
+}
+
+func isAllowedToolCode(toolCode string, allowedToolCodes map[string]struct{}) bool {
+	if len(allowedToolCodes) == 0 {
+		return true
+	}
+	toolCode = toolx.NormalizeToolCodeAlias(strings.TrimSpace(toolCode))
+	if _, ok := allowedToolCodes[toolCode]; ok {
+		return true
+	}
+	if toolx.IsAlwaysAllowedToolCode(toolCode) {
+		return true
+	}
+	return toolx.IsImpliedAllowedToolCode(toolCode, allowedToolCodes)
+}
+
+func makeAllowedToolCodeSet(input []string) map[string]struct{} {
+	if len(input) == 0 {
+		return nil
+	}
+	ret := make(map[string]struct{}, len(input))
+	for _, item := range input {
+		item = toolx.NormalizeToolCodeAlias(strings.TrimSpace(item))
+		if item == "" {
+			continue
+		}
+		ret[item] = struct{}{}
+	}
+	return ret
+}