This is the Flyte SDK (v2) version of this plugin. The Flytekit version is available as flytekitplugins-openai.
OpenAI
flyteplugins-openai
Flyte SDK (v2)Model Servingopenaillmagentsai
This plugin provides a drop-in replacement for OpenAI packages. It provides
Install
pip install flyteplugins-openaiQuick Start(example, may need adjustment)
See full examplespip install flyteplugins-openai
from flytekit import task, workflow
from flyteplugins.openai.agents import function_tool
@task
def my_task() -> None:
function_tool(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (1)
taskfunction_tool
Task for OpenAI.
from flyteplugins.openai.agents import function_tool
Dependencies
openai-agents
Related Plugins
OpenAI
The plugin currently features ChatGPT and Batch API connectors.
SGLang
Serve large language models using SGLang with Flyte Apps.
vLLM
Serve large language models using vLLM with Flyte Apps.
Dgxc-lepton
A professional Flytekit plugin that enables seamless deployment and management of AI inference endpoints using Lepton AI infrastructure within Flyte workflows.
Package Info
Min Flyte SDK
Modules1
Downloads
Last day56
Last week274
Last month1,020
