LLM-powered code generation and evaluation
flyteplugins-codegen
Generate code from natural language prompts and validate it by running tests in an isolated sandbox. Works with any model that supports structured outputs (GPT-4, Claude, Gemini, etc. via LiteLLM) or directly with the Agent SDK (Claude-only).
pip install flyteplugins-codegenQuick Start(example, may need adjustment)
pip install flyteplugins-codegen
from flytekit import task, workflow
from flyteplugins.codegen import AutoCoderAgent
agent = AutoCoderAgent(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (1)
Agent for single-file Python code generation with automatic testing and iteration.
extends dataclass — configuration or data structure for plugin setup
from flyteplugins.codegen import AutoCoderAgent
Dependencies
Related Plugins
Anthropic Claude
This plugin provides integration between Flyte and Anthropic's Claude API, enabling you to use Flyte tasks as tools for Claude agents.
Deck
This plugin provides additional renderers to improve task visibility within Flytekit.
Envd
envd is a command-line tool that helps you create the container-based development environment for AI/ML.
Flyte Interactive
FlyteInteractive plugin provides users' favorite interface to develop and debug a flyte task interactively. We support vscode, jupyter (WIP), and neovim (WIP).
