flyteorg

flyteorg/flyte-agent-plugins

Plugin Claude Code

Agent skills and MCP servers for working with Flyte.

flyteorg/flyte-agent-plugins

Plugin Claude Code

Work with Flyte 2. Deploy Flyte on AWS or kind, author and run workflows, apps, agents, data, and ML workloads with the SDK, and operate your cluster through bundled MCP servers.

flyteorg/flyte-agent-plugins

MCP server Claude CodeCodexCursor +2

MCP server "flyte-docs", hosted remotely at flyte-mcp.apps.demo.hosted.unionai.cloud, as configured in flyteorg/flyte-agent-plugins.

flyteorg/flyte-agent-plugins

MCP server Claude CodeCodexCursor +2

MCP server "flyte-cluster" as configured in flyteorg/flyte-agent-plugins. Runs locally from the flyte[mcp] Python package.

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Deploy Flyte on a kind cluster onto any host — your local machine, or a fresh cloud VM (DigitalOcean, AWS EC2, or GCP Compute Engine). Provisions the VM (firewall scoped to your IP), installs Docker/kind/kubectl/helm, then runs the kind Flyte deploy on that host and tunnels access back to your machine. Use when the…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Deploy a complete Flyte stack (flyte-binary + a hosted PostgreSQL + an object store) onto a kind cluster, running on the user's own machine or a DigitalOcean VM (droplet). PostgreSQL is hosted (Supabase or external); the object store is AWS S3 or Cloudflare R2. Use when the user wants to run Flyte on kind — either…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Use when deploying a Flyte v2 (flyte-binary / flyte2) cluster on AWS from scratch — provisions EKS + S3 + RDS PostgreSQL + AWS Load Balancer Controller, then helm-installs the flyte-binary chart behind an ALB, with optional TLS and Okta SSO. Trigger words: "deploy flyte", "flyte v2 on AWS", "flyte EKS".

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Flyte 1 task configuration, container images, resources, caching, secrets, scheduling, and CLI/config-file usage to Flyte 2 equivalents. Use when migrating Flyte 1 task configuration, images, resources, secrets, scheduling, or CLI/config-file usage to Flyte 2. Trigger words include resources, ImageSpec…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Flyte 1 branching, dynamic workflows, failure handling, and fan-out to native Flyte 2 Python. Use when migrating Flyte 1 branching, dynamic workflows, failure handling, or maptask/fan-out to Flyte 2. Trigger words: conditional, @dynamic, maptask, onfailure, branching, parallelism, fan-out, flyte.map…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Flyte 1 data types and offloaded I/O to Flyte 2. Use when migrating Flyte 1 data types and I/O (files, directories, dataframes, dataclasses) to Flyte 2, converting FlyteFile, FlyteDirectory, or StructuredDataset to flyte.io.File, flyte.io.Dir, and flyte.io.DataFrame. Trigger words are FlyteFile…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Flyte 1 machine learning code to Flyte 2 and unlocks net-new v2 patterns. Use when migrating Flyte 1 ML workloads (training, HPO, GPU/deep learning, batch inference) to Flyte 2, specifying GPU resources, or building the end-to-end pipeline pattern. Trigger words - migrate training, HPO, GPU, deep learning…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Slurm (sbatch/srun) HPC workloads to Flyte 2 (the flyte Python SDK) — job scripts become typed tasks, #SBATCH pragmas become TaskEnvironment config, job arrays become flyte.map or asyncio.gather, and multi-node training becomes a clustered task environment. Use when porting an HPC or supercomputer cluster…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Migrates Flyte 1 tasks and workflows to Flyte 2, where the @task, @workflow, and @dynamic decorators collapse into a single @env.task on a flyte.TaskEnvironment and a workflow becomes a task that calls other tasks. Use when the user is migrating Flyte 1 tasks/workflows to Flyte 2. Trigger words: migrate task, migrate…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Entry-point orchestrator for porting Flyte 1 (flytekit) code to Flyte 2 (flyte). Explains the v1 to v2 shift, the terminology mapping, a recommended migration strategy, hybrid v1/v2 pipelines, and routes to sibling migration skills. Use when the user wants to migrate, port, or upgrade Flyte 1 (flytekit) code to Flyte…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Builds durable agents with Flyte 2 — ReAct patterns, Plan-and-Execute, LangGraph integration, PydanticAI integration, OpenAI Agents SDK integration, agent memory, MCP tool integration, skills, and agent chat UI. Use when the user wants to build AI agents, implement ReAct loops, integrate agent frameworks, add agent…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Builds and serves Flyte 2 apps — FastAPI, Streamlit, vLLM, SGLang, WebSocket, and browser apps. Use when the user wants to serve a model, create a REST API, build a dashboard, deploy an LLM backend, or create a web app with Flyte. Trigger words: "app", "serve", "deploy app", "FastAPI", "Streamlit", "vLLM", "SGLang"…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Creates Flyte 2 project scaffolds (tasks, workflows, launch plans, apps), selects patterns (map tasks, traces, dynamic workflows, conditions), and generates code from templates aligned to the user's constraints. Use when the user wants to create or scaffold Flyte workflows, tasks, or apps from scratch — "write a Flyte…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Handles data engineering patterns: ETL pipelines, data processing, data quality checks, fanout/map tasks, conditions, dynamic workflows, and batch data transformations. Use when the user wants to build ETL pipelines, process large datasets, run data quality checks, fan out data processing tasks, or handle batch data…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Builds minimal evaluation harnesses (unit tests + small-run workflows) and suggests ways to validate correctness and performance early. Use when the user wants to test Flyte tasks, validate pipeline outputs, set up evaluation pipelines, or write unit tests for ML/data workflows. Trigger words: "test", "evaluate"…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Handles ML workload patterns: model training, hyperparameter optimization, experiment tracking, model evaluation and selection, batch inference, real-time serving, and model monitoring. Use when the user wants to train models, run hyperparameter search, track experiments, evaluate models, do batch or real-time…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Suggests performance improvements (task granularity, caching, resource requests, data format changes) using observed run metadata when available. Use when the user wants to optimize workflow performance, debug slow tasks, configure caching, tune resources, or improve throughput. Trigger words: "optimize"…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Runs Flyte 2 workflows, interacts with runs and actions, retrieves logs and data, and manages run lifecycle. Use when the user wants to run a workflow, check run status, view logs, get run outputs, re-run a workflow, or manage runs programmatically. Trigger words: "run", "execute", "logs", "status", "output", "input"…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Generates flyte.Image specs, Dockerfiles, dependency management, image tagging strategy, and reproducible build instructions for Flyte tasks. Use when the user needs to build container images for Flyte tasks, configure custom images, manage dependencies, or set up reproducible builds. Trigger words: "image", "Docker"…

flyteorg/flyte-agent-plugins

Skill Claude CodeCodex

Guides correct types, I/O, and serialization for common data (Pandas, Arrow, Parquet, images, audio, HF datasets), including data locality and storage best practices. Use when the user needs help with type annotations, data serialization, file I/O between tasks, custom type transformers, DataFrame handling, or…