Init

internal/ai/runtime/registry/registry.go

@@ -0,0 +1,44 @@
+package registry
+
+import (
+	"strings"
+
+	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{
+		Tools:     make([]einotool.BaseTool, 0, len(r.tools)),
+		ToolCodes: make(map[string]string),
+	}
+	for _, toolDef := range r.tools {
+		if toolDef == nil || !toolDef.Enabled(ctx) {
+			continue
+		}
+		tool, err := toolDef.Build(ctx)
+		if err != nil {
+			return nil, err
+		}
+		if tool == nil {
+			continue
+		}
+		toolName := strings.TrimSpace(toolDef.Name())
+		toolCode := strings.TrimSpace(toolDef.Code())
+		if toolName == "" || toolCode == "" {
+			continue
+		}
+		ret.Tools = append(ret.Tools, tool)
+		ret.ToolCodes[toolName] = toolCode
+	}
+	return ret, nil
+}

internal/ai/runtime/registry/types.go

@@ -0,0 +1,26 @@
+package registry
+
+import (
+	"cs-agent/internal/models"
+
+	einotool "github.com/cloudwego/eino/components/tool"
+)
+
+type Context struct {
+	Conversation *models.Conversation
+	AIAgent      *models.AIAgent
+	AIConfig     *models.AIConfig
+	UserMessage  *models.Message
+}
+
+type ToolSet struct {
+	Tools     []einotool.BaseTool
+	ToolCodes map[string]string
+}
+
+type Tool interface {
+	Name() string
+	Code() string
+	Enabled(ctx Context) bool
+	Build(ctx Context) (einotool.BaseTool, error)
+}