Union SDK v2

Proprietary extensions and features for Flyte, providing Union-specific functionality.

Installation

uv pip install union flyte

For development:

uv pip install -e .

Features

API Key Management

Create and manage API keys for headless authentication with Union.

Create an API Key

flyte create api-key admin --name my-api-key

This will create OAuth application credentials and output an API key that can be used for headless authentication.

Example output:

Client ID: serverless-1-myorg-my-api-key
The following API key will only be shown once. Be sure to keep it safe!
Configure your headless CLI by setting the following environment variable:

export UNION_API_KEY="<base64-encoded-credentials>"

Using the API Key

Set the generated API key as an environment variable:

export UNION_API_KEY="<your-api-key>"

Then use the Flyte CLI as normal - it will authenticate using the API key instead of prompting for interactive login.

Architecture

The Union SDK is structured as follows:

flyteplugins/union/internal/ - Proprietary protobuf definitions and generated code
flyteplugins/union/cli/ - CLI command plugins that extend the Flyte CLI
flyteplugins/union/utils/ - Shared utility functions
flyteplugins/union/remote/ - Union-specific remote objects

Plugin System

The SDK registers CLI plugins via entry points in pyproject.toml:

[project.entry-points."flyte.plugins.cli.commands"]
"create.api-key" = "flyteplugins.union.cli.api_key:create_api_key"
"delete.api-key" = "flyteplugins.union.cli.api_key:delete_api_key"
"get.api-key" = "flyteplugins.union.cli.api_key:get_api_key"

This makes Union commands available seamlessly through the Flyte CLI.

Dependencies

flyte>=2.0.0b26 - The Flyte SDK v2
rich-click>=1.8.0 - For CLI formatting

Development

To add a new CLI plugin:

Create a new file in src/flyteplugins/union/cli/
Define your click commands
Register the entry point in pyproject.toml
Shared resources (protobufs, utilities) are available in flyteplugins.union.internal.* and flyteplugins.union.utils.*

Iterating

Commands to rebuild and reinstall this plugin (into the flyte-sdk venv presumably). From this folder:

python -m build --wheel

From your flyte-sdk venv:

uv pip install --prerelease=allow --force-reinstall --no-deps --no-index --find-links /Users/ytong/go/src/github.com/unionai/flyteplugins-union/dist flyteplugins-union

Metadata

Release files for flyteplugins-union 0.1.0a0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Built distribution (wheel)

Table of built distributions (wheels) for flyteplugins-union 0.1.0a0
File	Interpreter	ABI	Platform	Reset
flyteplugins_union-0.1.0a0-py3-none-any.whl 3.2 kB Dec 6, 2025	Python 3	none	any	Details

Release files / flyteplugins_union-0.1.0a0-py3-none-any.whl

Download URL	flyteplugins_union-0.1.0a0-py3-none-any.whl
Size	3.2 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`7232344fb1f781a20e470dd43234befc3969d75159a3c3802f469a67cbf5eec2`
BLAKE2b-256 checksum How to use checksums	`9176f2fba1c1b4b05660da658359763f8944259c6904cfd72f511248169f9b8e`
Upload date	Dec 6, 2025
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.12.11`