Skip to content

OpenAI ChatGPT agent

Installation

To install the ChatGPT agent, run the following command:

shell
$ pip install flytekitplugins-openai

Example usage

For a usage example, see ChatGPT agent example.

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.

Union cluster deployment

After you have finished testing the agent locally, contact the Union team to enable it in your cluster.