ChatGPT agent

Installation

To install the ChatGPT agent, run the following command:

pip install flytekitplugins-openai

Example usage

For an example job, see ChatGPT agent example usage.

Local testing

To test the ChatGPT agent locally, create a class for the agent task that inherits from SyncAgentExecutorMixin. This mixin can handle synchronous tasks and allows flytekit to mimic FlytePropeller’s behavior in calling the agent. For more information, see “Testing agents locally”.

Note

In some cases, you will need to store credentials in your local environment when testing locally.

Flyte deployment configuration

Note

If you are using a managed deployment of Flyte, you will need to contact your deployment administrator to configure agents in your deployment.

To enable the ChatGPT agent in your Flyte deployment, see the ChatGPT agent deployment guide.