Flyte

VS Code extension for Flyte V2. Code intelligence, CLI integration, cluster management, and task visualization for ML workflow development.

Sidebar

Features

Autocomplete

Autocomplete inside TaskEnvironment(), Resources(), AppEnvironment(), Trigger(), Cache(), Secret(), and @env.task() with parameter names, types, and defaults
GPU completions after gpu= with all valid accelerators (T4, A100, H100, L4, V100, etc.)

Hover

Hover documentation on all Flyte classes, parameters, and functions (flyte.run, flyte.deploy, flyte.map, flyte.group, flyte.trace)
Diagnostics with real-time validation: invalid environment/app names, reserved ports, plugin_config + reusable conflicts, empty trigger names

Run Task above every @env.task function (only when all params have defaults)
Deploy when clusters are configured
Graph renders an ASCII DAG in the terminal showing tasks grouped by environment

Prefix	Output
`fenv`	`TaskEnvironment` with resources
`ftask`	`@env.task` async function
`fapp`	`AppEnvironment` with port
`fres`	`Resources(cpu, memory, gpu)`
`frun`	`flyte.run()` entry point
`fdeploy`	`flyte.deploy()` entry point
`fcache`	`Cache` with behavior and ignored_inputs
`ftrigger`	`Trigger` with Cron or FixedRate
`fimage`	`Image.from_debian_base().with_pip_packages()`
`fmap`	`flyte.map()` over inputs
`fgroup`	`flyte.group()` context manager
`ftrace`	`@flyte.trace` decorator
`fsecret`	`Secret` with env var

Clusters

Five sections following the project workflow: Environments, Tasks, Apps, Clusters, Runs.

code --install-extension atoolz.flyte-vscode

Setting	Default	Description
`flyte.cliPath`	`""`	Path to Flyte CLI. Empty for auto-discovery
`flyte.pythonPath`	`""`	Path to Python interpreter
`flyte.autoRefreshRuns`	`true`	Auto-refresh runs view
`flyte.refreshInterval`	`10000`	Refresh interval in ms

MIT