refactoring

2025-02-26 22:37:48 +01:00
parent 0139b79835
commit 0a18d8409e
22 changed files with 90 additions and 41 deletions
--- a/webui/actions.go
+++ b/webui/actions.go
@@ -0,0 +1,84 @@
+package webui
+
+import (
+	"context"
+	"encoding/json"
+
+	"github.com/mudler/local-agent-framework/core/action"
+	"github.com/mudler/local-agent-framework/core/state"
+	"github.com/mudler/local-agent-framework/pkg/xlog"
+
+	"github.com/mudler/local-agent-framework/core/agent"
+	"github.com/mudler/local-agent-framework/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"
+)
+
+var AvailableActions = []string{
+	ActionSearch,
+	ActionCustom,
+	ActionGithubIssueLabeler,
+	ActionGithubIssueOpener,
+	ActionGithubIssueCloser,
+	ActionGithubIssueSearcher,
+	ActionScraper,
+	ActionBrowse,
+	ActionWikipedia,
+	ActionSendMail,
+}
+
+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 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
+	}
+}