flyte-migrate-data-io

flyte-migrate-data-io is a skill for Claude Code, Codex from flyteorg/flyte-agent-plugins. It costs 118 tokens per session (2,891 once invoked), scanned A, original, Apache-2.0.

A guide for moving Flyte 1 data types and file-handling code to Flyte 2. It explains how references to large files, directories, tables, dataframes, and structured Python data are represented in the newer SDK.

In plain words
What is it for?
Use it to replace FlyteFile, FlyteDirectory, and StructuredDataset with Flyte 2 equivalents, and to update file, directory, dataframe, dataclass, and Pydantic task inputs and outputs.
Why use it?
It removes confusion caused by renamed types and asynchronous file or data access. Passing references instead of loading all large data into memory can remain the same overall approach.

Skill for Claude CodeCodex

Part of the flyte plugin — 21 skills, 2 MCP servers shipped together

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

agentmods
npx agentmods add skills/flyteorg/flyte-agent-plugins/flyte-migrate-data-io
Any agent
npx skills add flyteorg/flyte-agent-plugins --skill flyte-migrate-data-io
Clone the repo
git clone --depth 1 https://github.com/flyteorg/flyte-agent-plugins

Made for: Claude Code, Codex.

Or install flyte, the plugin that ships this one along with the rest of its 21 skills, 2 MCP servers.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for flyte-migrate-data-io

README.md
[![agentmods](https://agentmods.dev/badge/skills/flyteorg/flyte-agent-plugins/flyte-migrate-data-io.svg)](https://agentmods.dev/skills/flyteorg/flyte-agent-plugins/flyte-migrate-data-io)
Your own site
<a href="https://agentmods.dev/skills/flyteorg/flyte-agent-plugins/flyte-migrate-data-io"><img src="https://agentmods.dev/badge/skills/flyteorg/flyte-agent-plugins/flyte-migrate-data-io.svg" alt="Measured on agentmods" height="20"></a>
Per session 118 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 2,891 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5 $0.00118 $0.02891
Opus 5 $0.00059 $0.01445
Sonnet 5 $0.00024 $0.00578
Haiku 4.5 $0.00012 $0.00289

Measured 4d ago against content hash 1eb54f23dfb8, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

flyte-migrate-data-io scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 4d ago.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

plugins/flyte/skills/flyte-migrate-data-io/SKILL.md · 310 lines

How it starts

The opening of the file, as written. The whole thing — 310 lines — stays where its author put it; the contents beside it link to each section on GitHub.

Flyte 1 to 2 Migration: Data Types and I/O

Flyte 2 renames the offloaded-data types and makes their I/O async, but the mental model is the same: pass lightweight references to large data between tasks, not the materialized bytes. FlyteFile, FlyteDirectory, and StructuredDataset become flyte.io.File, flyte.io.Dir, and flyte.io.DataFrame. Plain dataclasses and Pydantic BaseModels work directly as task I/O with no JSON mixin.

Grounding References

Resource URL
Migration guide https://www.union.ai/docs/v2/flyte/user-guide/migration/flyte-2/data-io/
Official docs https://www.union.ai/docs/v2/flyte
Docs index (LLMs) https://www.union.ai/docs/v2/flyte/llms.txt
SDK API reference https://www.union.ai/docs/v2/union/api-reference/flyte-sdk/
Example code https://github.com/unionai/unionai-examples
Flyte MCP tools Available via flyte-mcp server

Type Mapping

Flyte 1 Flyte 2 Notes
flytekit.types.file.FlyteFile flyte.io.File I/O is async
flytekit.types.directory.FlyteDirectory flyte.io.Dir I/O is async
flytekit.types.structured.StructuredDataset flyte.io.DataFrame build with from_df, read with open(...).all()
@dataclass_json + @dataclass plain @dataclass no mixin needed
Pydantic BaseModel (+ config) plain Pydantic BaseModel works directly as task I/O

Offloaded Data: The Mental Model

File, Dir, and DataFrame are lightweight references (pointers) to data offloaded in blob storage — not the materialized bytes. In Flyte 2 the read/write operations are async: upload with await File.from_local(local_path), read with async with f.open("rb") as fh: await fh.read(), build a frame with flyte.io.DataFrame.from_df(df) (sync constructor), and read it with await fdf.open(pandas.DataFrame).all().

Files and Directories

FlyteFile and FlyteDirectory become flyte.io.File and flyte.io.Dir — the way you pass model artifacts and datasets between tasks. Use await File.from_local(...) to upload and async with file.open(...) to read.

Read the full file on GitHub · 310 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 4d ago First seen · 310 lines · 118 tokens per session scan A 1eb54f23dfb8

Subscribe to this mod's changes

flyte-migrate-data-io is a skill published in the GitHub repository flyteorg/flyte-agent-plugins (2 stars, last pushed 2d ago), licensed Apache-2.0. It adds 118 tokens to every session and 2,891 once invoked, about $0.0006 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other skills, from other repositories

systematic-debugging

Use when encountering any bug, test failure, or unexpected behavior, before proposing fixes.

obra/superpowers · 21 tokens

brainstorming

You MUST use this before any creative work - creating features, building components, adding functionality, or modifying behavior. Explores user intent, requirements and design before implementation.

obra/superpowers · 37 tokens

auto-perf-optimize

Run agent-driven VS Code performance or memory investigations. Use when asked to launch Code OSS, automate a VS Code scenario, run the Chat memory smoke runner, capture renderer heap snapshots, take workflow screenshots, compare run summaries, or drive a repeatable scenario before heap-snapshot analysis.

microsoft/vscode · 62 tokens

chat-perf

Run chat perf benchmarks and memory leak checks against the local dev build or any published VS Code version. Use when investigating chat rendering regressions, validating perf-sensitive changes to chat UI, or checking for memory leaks in the chat response pipeline.

microsoft/vscode · 51 tokens

chat-pet-sprite-creation

Use when creating or changing VS Code chat pet sprite art, sprite sheets, state animations, eye treatments, Stable/Insiders variants, or pet transitions under src/vs/workbench/contrib/chat/browser/widget/media/chatPet.

microsoft/vscode · 53 tokens

cpu-profile-analysis

Analyze V8/Chrome CPU profiles (.cpuprofile) and DevTools trace files (Trace-.json). Use when: profiling performance, investigating slow functions, comparing code paths, finding bottlenecks, analyzing timeToRequest, understanding call trees from sampling profiler data, analyzing layout/paint/rendering, investigating…

microsoft/vscode · 71 tokens