alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Standardize action results

Browse Source
This commit is contained in:
mudler
2025-03-01 17:27:07 +01:00
parent 8492c95cb6
commit 5b4f618ca3
22 changed files with 108 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
core/action/action_suite_test.go Normal file
View File

@@ -0,0 +1,13 @@
package action_test
import (
"testing"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
)
func TestAction(t *testing.T) {
RegisterFailHandler(Fail)
RunSpecs(t, "Agent Action test suite")
}

9
core/action/custom.go
View File

@@ -75,15 +75,16 @@ func (a *CustomAction) initializeInterpreter() error {
return nil return nil
} }
func (a *CustomAction) Run(ctx context.Context, params ActionParams) (string, error) { func (a *CustomAction) Run(ctx context.Context, params ActionParams) (ActionResult, error) {
v, err := a.i.Eval(fmt.Sprintf("%s.Run", a.config["name"])) v, err := a.i.Eval(fmt.Sprintf("%s.Run", a.config["name"]))
if err != nil { if err != nil {
return "", err return ActionResult{}, err
} }
run := v.Interface().(func(map[string]interface{}) (string, error)) run := v.Interface().(func(map[string]interface{}) (string, map[string]interface{}, error))
return run(params) res, meta, err := run(params)
return ActionResult{Result: res, Metadata: meta}, err
} }
func (a *CustomAction) Definition() ActionDefinition { func (a *CustomAction) Definition() ActionDefinition {

10
core/action/custom_test.go
View File

@@ -23,18 +23,18 @@ type Params struct {
Foo string Foo string
} }
func Run(config map[string]interface{}) (string, error) { func Run(config map[string]interface{}) (string, map[string]interface{}, error) {
p := Params{} p := Params{}
b, err := json.Marshal(config) b, err := json.Marshal(config)
if err != nil { if err != nil {
return "", err return "",map[string]interface{}{}, err
} }
if err := json.Unmarshal(b, &p); err != nil { if err := json.Unmarshal(b, &p); err != nil {
return "", err return "",map[string]interface{}{}, err
} }
return p.Foo, nil return p.Foo,map[string]interface{}{}, nil
} }
func Definition() map[string][]string { func Definition() map[string][]string {
@@ -79,7 +79,7 @@ return []string{"foo"}
"Foo": "bar", "Foo": "bar",
}) })
Expect(err).ToNot(HaveOccurred()) Expect(err).ToNot(HaveOccurred())
Expect(runResult).To(Equal("bar")) Expect(runResult.Result).To(Equal("bar"))
}) })
}) })

5
core/action/definition.go
View File

@@ -28,6 +28,11 @@ func NewContext(ctx context.Context, cancel context.CancelFunc) *ActionContext {
type ActionParams map[string]interface{} type ActionParams map[string]interface{}
type ActionResult struct {
Result string
Metadata map[string]interface{}
}
func (ap ActionParams) Read(s string) error { func (ap ActionParams) Read(s string) error {
err := json.Unmarshal([]byte(s), &ap) err := json.Unmarshal([]byte(s), &ap)
return err return err

4
core/action/intention.go
View File

@@ -21,8 +21,8 @@ type IntentResponse struct {
Reasoning string `json:"reasoning"` Reasoning string `json:"reasoning"`
} }
func (a *IntentAction) Run(context.Context, ActionParams) (string, error) { func (a *IntentAction) Run(context.Context, ActionParams) (ActionResult, error) {
return "no-op", nil return ActionResult{}, nil
} }
func (a *IntentAction) Definition() ActionDefinition { func (a *IntentAction) Definition() ActionDefinition {

4
core/action/newconversation.go
View File

@@ -18,8 +18,8 @@ type ConversationActionResponse struct {
Message string `json:"message"` Message string `json:"message"`
} }
func (a *ConversationAction) Run(context.Context, ActionParams) (string, error) { func (a *ConversationAction) Run(context.Context, ActionParams) (ActionResult, error) {
return "no-op", nil return ActionResult{}, nil
} }
func (a *ConversationAction) Definition() ActionDefinition { func (a *ConversationAction) Definition() ActionDefinition {

4
core/action/noreply.go
View File

@@ -12,8 +12,8 @@ func NewStop() *StopAction {
type StopAction struct{} type StopAction struct{}
func (a *StopAction) Run(context.Context, ActionParams) (string, error) { func (a *StopAction) Run(context.Context, ActionParams) (ActionResult, error) {
return "no-op", nil return ActionResult{}, nil
} }
func (a *StopAction) Definition() ActionDefinition { func (a *StopAction) Definition() ActionDefinition {

4
core/action/reasoning.go
View File

@@ -19,8 +19,8 @@ type ReasoningResponse struct {
Reasoning string `json:"reasoning"` Reasoning string `json:"reasoning"`
} }
func (a *ReasoningAction) Run(context.Context, ActionParams) (string, error) { func (a *ReasoningAction) Run(context.Context, ActionParams) (ActionResult, error) {
return "no-op", nil return ActionResult{}, nil
} }
func (a *ReasoningAction) Definition() ActionDefinition { func (a *ReasoningAction) Definition() ActionDefinition {

4
core/action/state.go
View File

@@ -33,8 +33,8 @@ type StateResult struct {
Goal string `json:"goal"` Goal string `json:"goal"`
} }
func (a *StateAction) Run(context.Context, ActionParams) (string, error) { func (a *StateAction) Run(context.Context, ActionParams) (ActionResult, error) {
return "internal state has been updated", nil return ActionResult{Result: "internal state has been updated"}, nil
} }
func (a *StateAction) Definition() ActionDefinition { func (a *StateAction) Definition() ActionDefinition {

2
core/agent/actions.go
View File

@@ -23,7 +23,7 @@ type ActionCurrentState struct {
// Actions is something the agent can do // Actions is something the agent can do
type Action interface { type Action interface {
Run(ctx context.Context, action action.ActionParams) (string, error) Run(ctx context.Context, action action.ActionParams) (action.ActionResult, error)
Definition() action.ActionDefinition Definition() action.ActionDefinition
} }

5
core/agent/agent.go
View File

@@ -242,9 +242,12 @@ func (a *Agent) Memory() RAGDB {
func (a *Agent) runAction(chosenAction Action, params action.ActionParams) (result string, err error) { func (a *Agent) runAction(chosenAction Action, params action.ActionParams) (result string, err error) {
for _, action := range a.systemInternalActions() { for _, action := range a.systemInternalActions() {
if action.Definition().Name == chosenAction.Definition().Name { if action.Definition().Name == chosenAction.Definition().Name {
if result, err = action.Run(a.actionContext, params); err != nil { res, err := action.Run(a.actionContext, params)
if err != nil {
return "", fmt.Errorf("error running action: %w", err) return "", fmt.Errorf("error running action: %w", err)
} }
result = res.Result
} }
} }

4
core/agent/agent_test.go
View File

@@ -36,7 +36,7 @@ type TestAction struct {
responseN int responseN int
} }
func (a *TestAction) Run(context.Context, action.ActionParams) (string, error) { func (a *TestAction) Run(context.Context, action.ActionParams) (action.ActionResult, error) {
res := a.response[a.responseN] res := a.response[a.responseN]
a.responseN++ a.responseN++
@@ -44,7 +44,7 @@ func (a *TestAction) Run(context.Context, action.ActionParams) (string, error) {
a.responseN = 0 a.responseN = 0
} }
return res, nil return action.ActionResult{Result: res}, nil
} }
func (a *TestAction) Definition() action.ActionDefinition { func (a *TestAction) Definition() action.ActionDefinition {

14
services/actions/browse.go
View File

@@ -18,7 +18,7 @@ func NewBrowse(config map[string]string) *BrowseAction {
type BrowseAction struct{} type BrowseAction struct{}
func (a *BrowseAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *BrowseAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
URL string `json:"url"` URL string `json:"url"`
}{} }{}
@@ -26,31 +26,31 @@ func (a *BrowseAction) Run(ctx context.Context, params action.ActionParams) (str
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
// download page with http.Client // download page with http.Client
client := &http.Client{} client := &http.Client{}
req, err := http.NewRequest("GET", result.URL, nil) req, err := http.NewRequest("GET", result.URL, nil)
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
resp, err := client.Do(req) resp, err := client.Do(req)
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
defer resp.Body.Close() defer resp.Body.Close()
pagebyte, err := io.ReadAll(resp.Body) pagebyte, err := io.ReadAll(resp.Body)
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
rendered, err := html2text.FromString(string(pagebyte), html2text.Options{PrettyTables: true}) rendered, err := html2text.FromString(string(pagebyte), html2text.Options{PrettyTables: true})
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
return fmt.Sprintf("The webpage '%s' content is:\n%s", result.URL, rendered), nil return action.ActionResult{Result: fmt.Sprintf("The webpage '%s' content is:\n%s", result.URL, rendered)}, nil
} }
func (a *BrowseAction) Definition() action.ActionDefinition { func (a *BrowseAction) Definition() action.ActionDefinition {

15
services/actions/genimage.go
View File

@@ -24,18 +24,18 @@ type GenImageAction struct {
imageModel string imageModel string
} }
func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Prompt string `json:"prompt"` Prompt string `json:"prompt"`
Size string `json:"size"` Size string `json:"size"`
}{} }{}
err := params.Unmarshal(&result) err := params.Unmarshal(&result)
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
if result.Prompt == "" { if result.Prompt == "" {
return "", fmt.Errorf("prompt is required") return action.ActionResult{}, fmt.Errorf("prompt is required")
} }
req := openai.ImageRequest{ req := openai.ImageRequest{
@@ -56,14 +56,17 @@ func (a *GenImageAction) Run(ctx context.Context, params action.ActionParams) (s
resp, err := a.client.CreateImage(ctx, req) resp, err := a.client.CreateImage(ctx, req)
if err != nil { if err != nil {
return "Failed to generate image " + err.Error(), err return action.ActionResult{Result: "Failed to generate image " + err.Error()}, err
} }
if len(resp.Data) == 0 { if len(resp.Data) == 0 {
return "Failed to generate image", nil return action.ActionResult{Result: "Failed to generate image"}, nil
} }
return fmt.Sprintf("The image was generated and available at: %s", resp.Data[0].URL), nil return action.ActionResult{
Result: fmt.Sprintf("The image was generated and available at: %s", resp.Data[0].URL), Metadata: map[string]interface{}{
"url": resp.Data[0].URL,
}}, nil
} }
func (a *GenImageAction) Definition() action.ActionDefinition { func (a *GenImageAction) Definition() action.ActionDefinition {

8
services/actions/githubissuecloser.go
View File

@@ -24,7 +24,7 @@ func NewGithubIssueCloser(ctx context.Context, config map[string]string) *Github
} }
} }
func (g *GithubIssuesCloser) Run(ctx context.Context, params action.ActionParams) (string, error) { func (g *GithubIssuesCloser) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Repository string `json:"repository"` Repository string `json:"repository"`
Owner string `json:"owner"` Owner string `json:"owner"`
@@ -34,7 +34,7 @@ func (g *GithubIssuesCloser) Run(ctx context.Context, params action.ActionParams
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
// _, _, err = g.client.Issues.CreateComment( // _, _, err = g.client.Issues.CreateComment(
@@ -57,14 +57,14 @@ func (g *GithubIssuesCloser) Run(ctx context.Context, params action.ActionParams
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
resultString := fmt.Sprintf("Closed issue %d in repository %s/%s", result.IssueNumber, result.Owner, result.Repository) resultString := fmt.Sprintf("Closed issue %d in repository %s/%s", result.IssueNumber, result.Owner, result.Repository)
if err != nil { if err != nil {
resultString = fmt.Sprintf("Error closing issue %d in repository %s/%s: %v", result.IssueNumber, result.Owner, result.Repository, err) resultString = fmt.Sprintf("Error closing issue %d in repository %s/%s: %v", result.IssueNumber, result.Owner, result.Repository, err)
} }
return resultString, err return action.ActionResult{Result: resultString}, err
} }
func (g *GithubIssuesCloser) Definition() action.ActionDefinition { func (g *GithubIssuesCloser) Definition() action.ActionDefinition {

8
services/actions/githubissuelabeler.go
View File

@@ -36,7 +36,7 @@ func NewGithubIssueLabeler(ctx context.Context, config map[string]string) *Githu
} }
} }
func (g *GithubIssuesLabeler) Run(ctx context.Context, params action.ActionParams) (string, error) { func (g *GithubIssuesLabeler) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Repository string `json:"repository"` Repository string `json:"repository"`
Owner string `json:"owner"` Owner string `json:"owner"`
@@ -45,9 +45,7 @@ func (g *GithubIssuesLabeler) Run(ctx context.Context, params action.ActionParam
}{} }{}
err := params.Unmarshal(&result) err := params.Unmarshal(&result)
if err != nil { if err != nil {
fmt.Printf("error: %v", err) return action.ActionResult{}, err
return "", err
} }
labels, _, err := g.client.Issues.AddLabelsToIssue(g.context, result.Owner, result.Repository, result.IssueNumber, []string{result.Label}) labels, _, err := g.client.Issues.AddLabelsToIssue(g.context, result.Owner, result.Repository, result.IssueNumber, []string{result.Label})
@@ -61,7 +59,7 @@ func (g *GithubIssuesLabeler) Run(ctx context.Context, params action.ActionParam
if err != nil { if err != nil {
resultString = fmt.Sprintf("Error adding label '%s' to issue %d in repository %s/%s: %v", result.Label, result.IssueNumber, result.Owner, result.Repository, err) resultString = fmt.Sprintf("Error adding label '%s' to issue %d in repository %s/%s: %v", result.Label, result.IssueNumber, result.Owner, result.Repository, err)
} }
return resultString, err return action.ActionResult{Result: resultString}, err
} }
func (g *GithubIssuesLabeler) Definition() action.ActionDefinition { func (g *GithubIssuesLabeler) Definition() action.ActionDefinition {

6
services/actions/githubissueopener.go
View File

@@ -25,7 +25,7 @@ func NewGithubIssueOpener(ctx context.Context, config map[string]string) *Github
} }
} }
func (g *GithubIssuesOpener) Run(ctx context.Context, params action.ActionParams) (string, error) { func (g *GithubIssuesOpener) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Title string `json:"title"` Title string `json:"title"`
Body string `json:"text"` Body string `json:"text"`
@@ -36,7 +36,7 @@ func (g *GithubIssuesOpener) Run(ctx context.Context, params action.ActionParams
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
issue := &github.IssueRequest{ issue := &github.IssueRequest{
@@ -52,7 +52,7 @@ func (g *GithubIssuesOpener) Run(ctx context.Context, params action.ActionParams
resultString = fmt.Sprintf("Created issue %d in repository %s/%s", createdIssue.GetNumber(), result.Owner, result.Repository) resultString = fmt.Sprintf("Created issue %d in repository %s/%s", createdIssue.GetNumber(), result.Owner, result.Repository)
} }
return resultString, err return action.ActionResult{Result: resultString}, err
} }
func (g *GithubIssuesOpener) Definition() action.ActionDefinition { func (g *GithubIssuesOpener) Definition() action.ActionDefinition {

8
services/actions/githubissuesearch.go
View File

@@ -26,7 +26,7 @@ func NewGithubIssueSearch(ctx context.Context, config map[string]string) *Github
} }
} }
func (g *GithubIssueSearch) Run(ctx context.Context, params action.ActionParams) (string, error) { func (g *GithubIssueSearch) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Query string `json:"query"` Query string `json:"query"`
Repository string `json:"repository"` Repository string `json:"repository"`
@@ -36,7 +36,7 @@ func (g *GithubIssueSearch) Run(ctx context.Context, params action.ActionParams)
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
query := fmt.Sprintf("%s in:%s user:%s", result.Query, result.Repository, result.Owner) query := fmt.Sprintf("%s in:%s user:%s", result.Query, result.Repository, result.Owner)
@@ -48,7 +48,7 @@ func (g *GithubIssueSearch) Run(ctx context.Context, params action.ActionParams)
}) })
if err != nil { if err != nil {
resultString = fmt.Sprintf("Error listing issues: %v", err) resultString = fmt.Sprintf("Error listing issues: %v", err)
return resultString, err return action.ActionResult{Result: resultString}, err
} }
for _, i := range issues.Issues { for _, i := range issues.Issues {
xlog.Info("Issue found", "title", i.GetTitle()) xlog.Info("Issue found", "title", i.GetTitle())
@@ -57,7 +57,7 @@ func (g *GithubIssueSearch) Run(ctx context.Context, params action.ActionParams)
// resultString += fmt.Sprintf("Body: %s\n", i.GetBody()) // resultString += fmt.Sprintf("Body: %s\n", i.GetBody())
} }
return resultString, err return action.ActionResult{Result: resultString}, err
} }
func (g *GithubIssueSearch) Definition() action.ActionDefinition { func (g *GithubIssueSearch) Definition() action.ActionDefinition {

14
services/actions/scrape.go
View File

@@ -16,7 +16,7 @@ func NewScraper(config map[string]string) *ScraperAction {
type ScraperAction struct{} type ScraperAction struct{}
func (a *ScraperAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *ScraperAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
URL string `json:"url"` URL string `json:"url"`
}{} }{}
@@ -24,15 +24,21 @@ func (a *ScraperAction) Run(ctx context.Context, params action.ActionParams) (st
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
scraper, err := scraper.New() scraper, err := scraper.New()
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
return scraper.Call(ctx, result.URL) res, err := scraper.Call(ctx, result.URL)
if err != nil {
fmt.Printf("error: %v", err)
return action.ActionResult{}, err
}
return action.ActionResult{Result: res}, nil
} }
func (a *ScraperAction) Definition() action.ActionDefinition { func (a *ScraperAction) Definition() action.ActionDefinition {

14
services/actions/search.go
View File

@@ -28,7 +28,7 @@ func NewSearch(config map[string]string) *SearchAction {
type SearchAction struct{ results int } type SearchAction struct{ results int }
func (a *SearchAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *SearchAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Query string `json:"query"` Query string `json:"query"`
}{} }{}
@@ -36,15 +36,21 @@ func (a *SearchAction) Run(ctx context.Context, params action.ActionParams) (str
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
ddg, err := duckduckgo.New(a.results, "LocalAgent") ddg, err := duckduckgo.New(a.results, "LocalAgent")
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
return ddg.Call(ctx, result.Query) res, err := ddg.Call(ctx, result.Query)
if err != nil {
fmt.Printf("error: %v", err)
return action.ActionResult{}, err
}
return action.ActionResult{Result: res}, nil
} }
func (a *SearchAction) Definition() action.ActionDefinition { func (a *SearchAction) Definition() action.ActionDefinition {

8
services/actions/sendmail.go
View File

@@ -27,7 +27,7 @@ type SendMailAction struct {
smtpPort string smtpPort string
} }
func (a *SendMailAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *SendMailAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Message string `json:"message"` Message string `json:"message"`
To string `json:"to"` To string `json:"to"`
@@ -37,7 +37,7 @@ func (a *SendMailAction) Run(ctx context.Context, params action.ActionParams) (s
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
// Authentication. // Authentication.
@@ -50,9 +50,9 @@ func (a *SendMailAction) Run(ctx context.Context, params action.ActionParams) (s
result.To, result.To,
}, []byte(result.Message)) }, []byte(result.Message))
if err != nil { if err != nil {
return "", err return action.ActionResult{}, err
} }
return fmt.Sprintf("Email sent to %s", result.To), nil return action.ActionResult{Result: fmt.Sprintf("Email sent to %s", result.To)}, nil
} }
func (a *SendMailAction) Definition() action.ActionDefinition { func (a *SendMailAction) Definition() action.ActionDefinition {

12
services/actions/wikipedia.go
View File

@@ -15,7 +15,7 @@ func NewWikipedia(config map[string]string) *WikipediaAction {
type WikipediaAction struct{} type WikipediaAction struct{}
func (a *WikipediaAction) Run(ctx context.Context, params action.ActionParams) (string, error) { func (a *WikipediaAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
result := struct { result := struct {
Query string `json:"query"` Query string `json:"query"`
}{} }{}
@@ -23,10 +23,16 @@ func (a *WikipediaAction) Run(ctx context.Context, params action.ActionParams) (
if err != nil { if err != nil {
fmt.Printf("error: %v", err) fmt.Printf("error: %v", err)
return "", err return action.ActionResult{}, err
} }
wiki := wikipedia.New("LocalAgent") wiki := wikipedia.New("LocalAgent")
return wiki.Call(ctx, result.Query) res, err := wiki.Call(ctx, result.Query)
if err != nil {
fmt.Printf("error: %v", err)
return action.ActionResult{}, err
}
return action.ActionResult{Result: res}, nil
} }
func (a *WikipediaAction) Definition() action.ActionDefinition { func (a *WikipediaAction) Definition() action.ActionDefinition {