Adding Actions to Agents
Empower agents to take actions during conversations.
Note: Support for actions is currently limited to ChatGPTAgent
.
What are actions?
Actions refer to specific tools an agent can execute during the conversation. These actions can encompass various activities like writing an email,
scheduling an appointment, and so forth. They are implemented as classes derived from the BaseAction
class.
The BaseAction
class has the following key structure:
Every action class derived from BaseAction must implement the run method, which is called to execute the action. Importantly, each action class should include a docstring for this method that explains the expected parameters, a config that can be used to create the action, and a response class.
- This docstring is critical as it provides the information that the AI will consume in its prompt when instructing the execution of the action.
- The config contains all the parameters needed to customize your action.
- The response object is returned to the agent upon action completion and encodes its output, as well as if it was a success or failure.
How Agents use Actions
The agent is responsible for managing and executing actions within a conversation. The agent consumes a configuration object at instantiation, which specifies the actions that the agent can perform.
The agent configuration lists the actions available to the agent:
ActionsWorker
: how actions are executed and consumed
The ActionsWorker
class plays a crucial role in the async processing of actions within the system. It’s a specialized form of the InterruptibleWorker
class,
designed to handle the execution of actions and passing results back to the agent.
The ActionsWorker
is initialized with an input queue and an output queue. It uses an ActionFactory
instance to create and execute actions based on the inputs it receives.
The flow of actions is as follows:
- Agent sends action requests to the
ActionsWorker
through the worker’s input queue. ActionsWorker
reads the action request from the input queue. It then creates an instance of the appropriate action using theActionFactory
, and executes it using the provided parameters.- The executed action returns an
ActionOutput
object which encapsulates the result of the action. ActionsWorker
creates anActionResultAgentInput
from theActionOutput
, and puts it in its output queue.- The agent then consumes the
ActionResultAgentInput
from the queue in its process method. This result is added to the transcript of the conversation, and can influence the behavior of the agent in subsequent interactions.
Implementing your own action: Nylas email example
In this section, we provide an example of an action, NylasSendEmail
, which extends the BaseAction
class. It implements the run method to send an email using the Nylas API.
We will also show how to add this action to an agent and use it in a conversation.
Creating the Action
We define a structured set of parameters that the LLM will fill, a response structure that the agent can ingest and use as context for the rest of the conversation, and an action
config that crucially contains the action_type
(but doesn’t have any other specific parameters). We also add an _end_of_run_hook()
, which we customized to log that the action successfully completed.
Making a custom ActionFactory
To use our new action with an agent in a conversation, we will need to create an ActionFactory
that can produce instances of the action.
We will store the code above in nylas_send_email.py
and make a factory that can create this action for an agent:
Using your action and action factory in an agent
Now you can use your action in any conversation by adding the action config to an agent config. For example in a ChatGPTAgentConfig,
When you pass in your new action factory created above, any agent factory can use this config to generate an agent for conversations. See Agent Factory for more information on how to use your action factory in an agent factory, as well as how to plug it into a telephony server.