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 Plugin Registry에서 직접 구축된 버전 관리 지식 팩을 제공하며, 클러스터가 구성되었을 때 태스크를 실행하기 위한 가벼운 런타임 브리지를 포함합니다. 이제 어시스턴트는 추측을 멈추고 정확한 답변을 제공합니다.

어시스턴트가 할 수 있는 일

모든 도구는 순수 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 서버가 아닌 시스템 프롬프트). 자유롭게 조합 가능합니다.

이 프로젝트는 **작성(authoring)**에 중점을 둡니다: 개발자가 Flyte에 대해 질문하는 바로 그 순간을 지원합니다.

로드맵

• 모든 flyte-sdk 릴리스 시 지식 팩을 자동 재구축하는 GitHub Action

• 공식 MCP Registry에 등록

• 소형 sentence-transformer 모델을 통한 로컬 의미론적 검색

• AST 기반 마이그레이션 codemod (현재의 정규식 방식 대체)

• 테일(tail) 지원이 포함된 로그 스트리밍 도구 (get_flyte_execution_logs)

라이선스

Apache-2.0 — Flyte 자체와 동일한 라이선스입니다.

면책 조항

독립적인 커뮤니티 프로젝트입니다. Flyte 또는 Union.ai와 공식적으로 제휴하거나 보증하지 않습니다. Flyte 이름과 로고는 각 소유자의 상표입니다.