Flyte & Union.ai Tutorials

Tutorials and examples for building AI agents, ML pipelines, and data workflows with Flyte 2.

---

👉 Tonight's workshop: RAG and Agentic Memory

tutorials/rag-agent-memory/

Build a document index, search it with no model at all, answer from it with citations, see the embedding space in 2D, then point the same vector store the other way so an agent writes its own memories into it.

Runs entirely in Colab. No cluster, no vector database to host, and no API key needed until step 2 (step 3's chart doesn't need one either). Or run it locally with flyte run --local, or on a Flyte cluster where every step becomes a container.

Along the way: what tokenization actually does to your text, why chunking is a hyperparameter rather than plumbing, where dense retrieval quietly fails, and how to swap Chroma for Qdrant without touching a single step.

git clone https://github.com/unionai/workshops
cd workshops/tutorials/rag-agent-memory
uv venv .venv --python 3.11 && source .venv/bin/activate
uv pip install -r requirements.txt
flyte run --local step0_index.py index

---

Featured

---

Get Started

---

Agents

MCP

LLM Fine-Tuning

ML / AI

Biotech / Life Sciences

Data

---

Setup

# Clone the repository
git clone https://github.com/unionai/workshops
cd workshops

# Navigate to any tutorial
cd tutorials/langgraph_agent_research

# Create virtual environment and install dependencies
uv venv .venv --python 3.11
source .venv/bin/activate
uv pip install -r requirements.txt

Flyte Cluster (for remote runs)

flyte create config \
    --endpoint <your-endpoint> \
    --auth-type headless \
    --builder remote \
    --domain development \
    --project flytesnacks

Don't have a cluster? Request access at flyte.org.

Run examples

Remote:

uv run flyte run workflow.py research_pipeline --query "Compare quantum computing approaches"

Local:

uv run flyte run --local workflow.py research_pipeline --query "Compare quantum computing approaches"

Local with TUI:

uv run flyte run --local --tui workflow.py research_pipeline --query "Compare quantum computing approaches"

Start TUI dashboard:

uv run flyte start tui