Refactorings

services/actions.go

@@ -0,0 +1,88 @@
+package services
+
+import (
+	"context"
+	"encoding/json"
+
+	"github.com/mudler/LocalAgent/core/action"
+	"github.com/mudler/LocalAgent/core/state"
+	"github.com/mudler/LocalAgent/pkg/xlog"
+
+	"github.com/mudler/LocalAgent/core/agent"
+	"github.com/mudler/LocalAgent/services/actions"
+)
+
+const (
+	// Actions
+	ActionSearch              = "search"
+	ActionCustom              = "custom"
+	ActionGithubIssueLabeler  = "github-issue-labeler"
+	ActionGithubIssueOpener   = "github-issue-opener"
+	ActionGithubIssueCloser   = "github-issue-closer"
+	ActionGithubIssueSearcher = "github-issue-searcher"
+	ActionScraper             = "scraper"
+	ActionWikipedia           = "wikipedia"
+	ActionBrowse              = "browse"
+	ActionSendMail            = "send_mail"
+	ActionGenerateImage       = "generate_image"
+)
+
+var AvailableActions = []string{
+	ActionSearch,
+	ActionCustom,
+	ActionGithubIssueLabeler,
+	ActionGithubIssueOpener,
+	ActionGithubIssueCloser,
+	ActionGithubIssueSearcher,
+	ActionScraper,
+	ActionBrowse,
+	ActionWikipedia,
+	ActionSendMail,
+	ActionGenerateImage,
+}
+
+func Actions(a *state.AgentConfig) func(ctx context.Context) []agent.Action {
+	return func(ctx context.Context) []agent.Action {
+		allActions := []agent.Action{}
+
+		for _, a := range a.Actions {
+			var config map[string]string
+			if err := json.Unmarshal([]byte(a.Config), &config); err != nil {
+				xlog.Error("Error unmarshalling action config", "error", err)
+				continue
+			}
+
+			switch a.Name {
+			case ActionCustom:
+				customAction, err := action.NewCustom(config, "")
+				if err != nil {
+					xlog.Error("Error creating custom action", "error", err)
+					continue
+				}
+				allActions = append(allActions, customAction)
+			case ActionGenerateImage:
+				allActions = append(allActions, actions.NewGenImage(config))
+			case ActionSearch:
+				allActions = append(allActions, actions.NewSearch(config))
+			case ActionGithubIssueLabeler:
+				allActions = append(allActions, actions.NewGithubIssueLabeler(ctx, config))
+			case ActionGithubIssueOpener:
+				allActions = append(allActions, actions.NewGithubIssueOpener(ctx, config))
+			case ActionGithubIssueCloser:
+				allActions = append(allActions, actions.NewGithubIssueCloser(ctx, config))
+			case ActionGithubIssueSearcher:
+				allActions = append(allActions, actions.NewGithubIssueSearch(ctx, config))
+			case ActionScraper:
+				allActions = append(allActions, actions.NewScraper(config))
+			case ActionWikipedia:
+				allActions = append(allActions, actions.NewWikipedia(config))
+			case ActionBrowse:
+				allActions = append(allActions, actions.NewBrowse(config))
+			case ActionSendMail:
+				allActions = append(allActions, actions.NewSendMail(config))
+			}
+		}
+
+		return allActions
+	}
+}

services/connectors.go

@@ -0,0 +1,58 @@
+package services
+
+import (
+	"encoding/json"
+
+	"github.com/mudler/LocalAgent/pkg/xlog"
+	"github.com/mudler/LocalAgent/services/connectors"
+
+	"github.com/mudler/LocalAgent/core/state"
+)
+
+const (
+	// Connectors
+	ConnectorTelegram     = "telegram"
+	ConnectorSlack        = "slack"
+	ConnectorDiscord      = "discord"
+	ConnectorGithubIssues = "github-issues"
+	ConnectorGithubPRs    = "github-prs"
+)
+
+var AvailableConnectors = []string{
+	ConnectorTelegram,
+	ConnectorSlack,
+	ConnectorDiscord,
+	ConnectorGithubIssues,
+	ConnectorGithubPRs,
+}
+
+func Connectors(a *state.AgentConfig) []state.Connector {
+	conns := []state.Connector{}
+
+	for _, c := range a.Connector {
+		var config map[string]string
+		if err := json.Unmarshal([]byte(c.Config), &config); err != nil {
+			xlog.Info("Error unmarshalling connector config", err)
+			continue
+		}
+		switch c.Type {
+		case ConnectorTelegram:
+			cc, err := connectors.NewTelegramConnector(config)
+			if err != nil {
+				xlog.Info("Error creating telegram connector", err)
+				continue
+			}
+
+			conns = append(conns, cc)
+		case ConnectorSlack:
+			conns = append(conns, connectors.NewSlack(config))
+		case ConnectorDiscord:
+			conns = append(conns, connectors.NewDiscord(config))
+		case ConnectorGithubIssues:
+			conns = append(conns, connectors.NewGithubIssueWatcher(config))
+		case ConnectorGithubPRs:
+			conns = append(conns, connectors.NewGithubPRWatcher(config))
+		}
+	}
+	return conns
+}

services/prompts.go

@@ -0,0 +1,42 @@
+package services
+
+import (
+	"encoding/json"
+
+	"github.com/mudler/LocalAgent/pkg/xlog"
+	"github.com/mudler/LocalAgent/services/prompts"
+
+	"github.com/mudler/LocalAgent/core/agent"
+	"github.com/mudler/LocalAgent/core/state"
+)
+
+const (
+	// Connectors
+	DynamicPromptCustom = "custom"
+)
+
+var AvailableBlockPrompts = []string{
+	DynamicPromptCustom,
+}
+
+func PromptBlocks(a *state.AgentConfig) []agent.PromptBlock {
+	promptblocks := []agent.PromptBlock{}
+
+	for _, c := range a.PromptBlocks {
+		var config map[string]string
+		if err := json.Unmarshal([]byte(c.Config), &config); err != nil {
+			xlog.Info("Error unmarshalling connector config", err)
+			continue
+		}
+		switch c.Type {
+		case DynamicPromptCustom:
+			prompt, err := prompts.NewDynamicPrompt(config, "")
+			if err != nil {
+				xlog.Error("Error creating custom prompt", "error", err)
+				continue
+			}
+			promptblocks = append(promptblocks, prompt)
+		}
+	}
+	return promptblocks
+}

services/prompts/custom.go

@@ -0,0 +1,97 @@
+package prompts
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/mudler/LocalAgent/core/agent"
+	"github.com/mudler/LocalAgent/pkg/xlog"
+	"github.com/traefik/yaegi/interp"
+	"github.com/traefik/yaegi/stdlib"
+)
+
+type DynamicPrompt struct {
+	config    map[string]string
+	goPkgPath string
+	i         *interp.Interpreter
+}
+
+func NewDynamicPrompt(config map[string]string, goPkgPath string) (*DynamicPrompt, error) {
+	a := &DynamicPrompt{
+		config:    config,
+		goPkgPath: goPkgPath,
+	}
+
+	if err := a.initializeInterpreter(); err != nil {
+		return nil, err
+	}
+
+	if err := a.callInit(); err != nil {
+		xlog.Error("Error calling custom action init", "error", err)
+	}
+
+	return a, nil
+}
+
+func (a *DynamicPrompt) callInit() error {
+	if a.i == nil {
+		return nil
+	}
+
+	v, err := a.i.Eval(fmt.Sprintf("%s.Init", a.config["name"]))
+	if err != nil {
+		return err
+	}
+
+	run := v.Interface().(func() error)
+
+	return run()
+}
+
+func (a *DynamicPrompt) initializeInterpreter() error {
+	if _, exists := a.config["code"]; exists && a.i == nil {
+		unsafe := strings.ToLower(a.config["unsafe"]) == "true"
+		i := interp.New(interp.Options{
+			GoPath:       a.goPkgPath,
+			Unrestricted: unsafe,
+		})
+		if err := i.Use(stdlib.Symbols); err != nil {
+			return err
+		}
+
+		if _, exists := a.config["name"]; !exists {
+			a.config["name"] = "custom"
+		}
+
+		_, err := i.Eval(fmt.Sprintf("package %s\n%s", a.config["name"], a.config["code"]))
+		if err != nil {
+			return err
+		}
+
+		a.i = i
+	}
+
+	return nil
+}
+
+func (a *DynamicPrompt) Render(c *agent.Agent) (string, error) {
+	v, err := a.i.Eval(fmt.Sprintf("%s.Render", a.config["name"]))
+	if err != nil {
+		return "", err
+	}
+
+	run := v.Interface().(func() (string, error))
+
+	return run()
+}
+
+func (a *DynamicPrompt) Role() string {
+	v, err := a.i.Eval(fmt.Sprintf("%s.Role", a.config["name"]))
+	if err != nil {
+		return "system"
+	}
+
+	run := v.Interface().(func() string)
+
+	return run()
+}