Search might be useful for other agents

This commit is contained in:
Ettore Di Giacinto
2024-04-05 23:59:48 +02:00
parent f2c74e29e8
commit 899e3c0771
2 changed files with 10 additions and 7 deletions

View File

@@ -1,55 +0,0 @@
package action
import (
"fmt"
"github.com/mudler/local-agent-framework/action"
"github.com/sap-nocops/duckduckgogo/client"
"github.com/sashabaranov/go-openai/jsonschema"
)
func NewSearch() *SearchAction {
return &SearchAction{}
}
type SearchAction struct{}
func (a *SearchAction) Run(params action.ActionParams) (string, error) {
result := struct {
Query string `json:"query"`
}{}
err := params.Unmarshal(&result)
if err != nil {
fmt.Printf("error: %v", err)
return "", err
}
ddg := client.NewDuckDuckGoSearchClient()
res, err := ddg.SearchLimited(result.Query, 10)
if err != nil {
msg := fmt.Sprintf("error: %v", err)
fmt.Printf(msg)
return msg, err
}
results := ""
for i, r := range res {
results += fmt.Sprintf("*********** RESULT %d\nurl: %s\ntitle: %s\nsnippet: %s\n", i, r.FormattedUrl, r.Title, r.Snippet)
}
return results, nil
}
func (a *SearchAction) Definition() action.ActionDefinition {
return action.ActionDefinition{
Name: "search_internet",
Description: "Search the internet for something.",
Properties: map[string]jsonschema.Definition{
"query": {
Type: jsonschema.String,
Description: "The query to search for.",
},
},
Required: []string{"query"},
}
}

View File

@@ -12,7 +12,7 @@ import (
"github.com/donseba/go-htmx"
"github.com/donseba/go-htmx/sse"
actions "github.com/mudler/local-agent-framework/example/webui/actions"
external "github.com/mudler/local-agent-framework/external"
. "github.com/mudler/local-agent-framework/agent"
)
@@ -65,7 +65,7 @@ func main() {
)
return true
}),
WithActions(actions.NewSearch()),
WithActions(external.NewSearch(3)),
WithAgentResultCallback(func(state ActionState) {
text := fmt.Sprintf(`Reasoning: %s
Action taken: %+v