flyte-mcp

claude mcp add flyte -- uvx flyte-mcp

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

将此内容添加到 ~/.claude.json、~/.cursor/mcp.json 或您客户端的配置文件中。

为什么存在这个项目

询问任何 AI 助手“编写一个带有缓存和 4 个 GPU 的 Flyte V2 任务”，你得到的往往是错误的代码：V1 导入、虚构的装饰器、幻觉产生的资源关键字参数。由于助手没有可靠的渠道获取 Flyte 生态系统信息，它只能用训练数据中的猜测来填补空白。

flyte-mcp 就是那个渠道。它提供了一个直接从 flyte-sdk 源码树和 Flyte 插件注册表 构建的版本化知识包，外加一个轻量级的运行时桥接器，用于在配置了集群时执行任务。助手不再猜测，而是开始给出准确的答案。

你的助手能做什么

所有工具均为纯 Python 实现，使用 stdio 传输，除非你明确使用运行时桥接器，否则不会进行任何网络调用。

有效的示例提示词

• 如何缓存任务并在输入更改时失效？

• 给我展示一个带有 A100 的分布式 PyTorch 训练示例。

• 我应该为 Snowflake 使用哪个 Flyte 插件，导入语句是什么？

• 将此 flytekit V1 工作流迁移到 V2。

• TaskEnvironment 的签名是什么？

你的助手会选择正确的工具并组合出准确的答案。

重建知识包

贡献者和发布自动化流程可以从源码重新生成该包：

python scripts/build_knowledge.py \
  --sdk-path /path/to/flyte-sdk \
  --registry /path/to/flyte-plugin-registry/src/data/plugins.json \
  --out src/flyte_mcp/data/flyte-v2-knowledge.json

使用的来源：

• flyte-sdk/src/flyte/__init__.py — 通过 AST 获取公共符号

• flyte-sdk/examples/* — 按主题分类的规范模式

• flyte-sdk/README.md, FEATURES.md, CONTRIBUTING.md — 元文档

• flyte-plugin-registry — 精选插件目录

flytesnacks 被特意排除在外：V2 已将示例整合到代码库中。

与其他 Flyte MCP 项目的关系

• wherobots/flyte-mcp — 仅运行时。通过 API 密钥在已部署的 Flyte 实例上发现并执行任务。互补而非竞争。

• unionai/claude-agents-public — Claude Code 自定义代理（系统提示词，而非 MCP 服务器）。可自由组合。

本项目专注于 创作：即开发者输入关于 Flyte 的提示词的那一刻。

路线图

• GitHub Action：在每次 flyte-sdk 发布时自动重建知识包

• 在官方 MCP 注册表 中列出

• 通过小型 sentence-transformer 模型实现本地语义搜索

• 基于 AST 的迁移代码转换 (codemod)（替换当前的正则表达式处理）

• 日志流工具 (get_flyte_execution_logs)，支持 tail 功能

许可证

Apache-2.0 — 与 Flyte 本身使用的许可证相同。

免责声明

独立的社区项目。未与 Flyte 或 Union.ai 官方关联或背书。Flyte 名称和徽标是其各自所有者的商标。