Preview Flyte 2 for productionHosted on Union.ai

Discover 71+ Plugins
and 223+ Modules

Extend your Flyte workflows with community-built integrations for data processing, ML training, experiment tracking, and more.

71

Plugins

223

Modules

152,151

Monthly Downloads

9

Categories

Popular Plugins

Recently Added

NEWFor AI-assisted Flyte development

flyte-mcp: Flyte V2 for AI agents

Give Claude, Cursor, and Claude Code accurate, versioned answers about Flyte V2 without scraping READMEs or hallucinating import paths. 17 tools covering the SDK API, canonical example patterns, plugin registry, V1 to V2 migration, and optional cluster runtime.

Claude Code

claude mcp add flyte -- uvx flyte-mcp

Try it: one-shot

uvx flyte-mcp

Cursor / Claude Desktop (~/.cursor/mcp.json or ~/.claude.json)

{
  "mcpServers": {
    "flyte": {
      "command": "uvx",
      "args": ["flyte-mcp"]
    }
  }
}

What the agent can do

  • get_flyte_symbolFull API detail for any public Flyte V2 symbol
  • search_flyte_apiKeyword search over the V2 Python API
  • get_flyte_patternCanonical example code by theme (caching, GPU, GenAI, ...)
  • find_flyte_example_forNatural-language match to example themes
  • suggest_flyte_plugin_forPlugin recommendation, prefers V2 entries
  • migrate_v1_to_v2Rewrite flytekit V1 code to flyte-sdk V2 syntax
  • run_flyte_taskExecute a task on the configured Flyte cluster

Get Involved

Learn, connect, and contribute to the Flyte ecosystem.