feat: add action to call other agents (#60)

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>

services/actions.go

@@ -31,6 +31,7 @@ const (
 	ActionSendMail                       = "send_mail"
 	ActionGenerateImage                  = "generate_image"
 	ActionCounter                        = "counter"
+	ActionCallAgents                     = "call_agents"
 )
 
 var AvailableActions = []string{
@@ -51,10 +52,11 @@ var AvailableActions = []string{
 	ActionGenerateImage,
 	ActionTwitterPost,
 	ActionCounter,
+	ActionCallAgents,
 }
 
-func Actions(a *state.AgentConfig) func(ctx context.Context) []agent.Action {
-	return func(ctx context.Context) []agent.Action {
+func Actions(a *state.AgentConfig) func(ctx context.Context, pool *state.AgentPool) []agent.Action {
+	return func(ctx context.Context, pool *state.AgentPool) []agent.Action {
 		allActions := []agent.Action{}
 
 		for _, a := range a.Actions {
@@ -104,6 +106,8 @@ func Actions(a *state.AgentConfig) func(ctx context.Context) []agent.Action {
 				allActions = append(allActions, actions.NewPostTweet(config))
 			case ActionCounter:
 				allActions = append(allActions, actions.NewCounter(config))
+			case ActionCallAgents:
+				allActions = append(allActions, actions.NewCallAgent(config, pool))
 			}
 		}
 

services/actions/callagents.go

@@ -0,0 +1,87 @@
+package actions
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/mudler/LocalAgent/core/action"
+	"github.com/mudler/LocalAgent/core/agent"
+	"github.com/mudler/LocalAgent/core/state"
+	"github.com/sashabaranov/go-openai"
+	"github.com/sashabaranov/go-openai/jsonschema"
+)
+
+func NewCallAgent(config map[string]string, pool *state.AgentPool) *CallAgentAction {
+	return &CallAgentAction{
+		pool: pool,
+	}
+}
+
+type CallAgentAction struct {
+	pool *state.AgentPool
+}
+
+func (a *CallAgentAction) Run(ctx context.Context, params action.ActionParams) (action.ActionResult, error) {
+	result := struct {
+		AgentName string `json:"agent_name"`
+		Message   string `json:"message"`
+	}{}
+	err := params.Unmarshal(&result)
+	if err != nil {
+		fmt.Printf("error: %v", err)
+
+		return action.ActionResult{}, err
+	}
+
+	ag := a.pool.GetAgent(result.AgentName)
+	if ag == nil {
+		return action.ActionResult{}, fmt.Errorf("agent '%s' not found", result.AgentName)
+	}
+
+	resp := ag.Ask(
+		agent.WithConversationHistory(
+			[]openai.ChatCompletionMessage{
+				{
+					Role:    "user",
+					Content: result.Message,
+				},
+			},
+		),
+	)
+	if resp.Error != nil {
+		return action.ActionResult{}, err
+	}
+
+	return action.ActionResult{Result: resp.Response}, nil
+}
+
+func (a *CallAgentAction) Definition() action.ActionDefinition {
+	allAgents := a.pool.AllAgents()
+
+	description := "Use this tool to call another agent. Available agents and their roles are:"
+
+	for _, agent := range allAgents {
+		agentConfig := a.pool.GetConfig(agent)
+		if agentConfig == nil {
+			continue
+		}
+		description += fmt.Sprintf("\n- %s: %s", agent, agentConfig.Description)
+	}
+
+	return action.ActionDefinition{
+		Name:        "call_agent",
+		Description: description,
+		Properties: map[string]jsonschema.Definition{
+			"agent_name": {
+				Type:        jsonschema.String,
+				Description: "The name of the agent to call.",
+				Enum:        allAgents,
+			},
+			"message": {
+				Type:        jsonschema.String,
+				Description: "The message to send to the agent.",
+			},
+		},
+		Required: []string{"agent_name", "message"},
+	}
+}