Skip to content

Chat Agent

Interactive AI assistant for querying Cromwell workflows and reading files via natural language.

  • Natural Language

    Ask questions about workflows in plain English

  • Built-in Tools

    Query workflows, read GCS files, check status

  • Session Persistence

    Resume conversations with context retention

Quick Start

pumbaa config init

The wizard will guide you through setting up your preferred LLM provider.

pumbaa chat

LLM Providers

Choose your preferred AI backend:

pumbaa chat --provider gemini --gemini-api-key <API_KEY>

Get an API Key

Free tier available at Google AI Studio

pumbaa chat --provider vertex --vertex-project <PROJECT_ID>

GCP Project Required

Requires an active Google Cloud project with Vertex AI API enabled.

pumbaa chat --provider ollama

Default Option

Runs locally, no API key needed. Install from ollama.ai

Capabilities

  • Query Workflows

    Search by status, name, or labels

  • Get Status

    Check execution state in real-time

  • View Metadata

    Inspect workflow details and call info

  • Get Outputs

    List workflow output files

  • View Logs

    Access workflow log paths for debugging

  • Read GCS Files

    Fetch files from Google Cloud Storage

  • Write Files

    Save scripts or notes to the working directory

  • WDL Knowledge Base

    List, search, and inspect your indexed WDL workflows (PUMBAA_WDL_DIR)

Streaming

Responses stream in real time as the model generates them. Press Esc during generation to cancel it.

Controls

Key Action
Enter Send message
Ctrl+J Insert line break
Up / Down Scroll messages (input mode)
Tab Toggle navigation mode
Y Copy selected message (navigation mode)
G / Shift+G First / last message (navigation mode)
Ctrl+S Browse and switch sessions
Ctrl+R Resume previous session for the same task
Esc Cancel generation · leave input · exit chat

Session Management

Conversations are persisted in SQLite (~/.pumbaa/sessions.db) for context retention.

# List existing sessions
pumbaa chat --list

# Resume a session
pumbaa chat --session <SESSION_ID>

Inside the TUI:

  • Ctrl+S opens a session browser to switch conversations
  • Ctrl+R resumes the most recent session for the current task
  • Opening the chat from the debug view for the same task automatically resumes that task's conversation

Configuration

Environment Variables

export PUMBAA_LLM_PROVIDER=gemini
export GEMINI_API_KEY=<your-api-key>
export GEMINI_MODEL=gemini-2.5-flash  # optional

export PUMBAA_LLM_PROVIDER=ollama
export OLLAMA_HOST=http://localhost:11434
export OLLAMA_MODEL=llama3.2:3b

export PUMBAA_LLM_PROVIDER=vertex
export VERTEX_PROJECT=<project-id>
export VERTEX_LOCATION=us-central1
export VERTEX_MODEL=gemini-2.5-flash

Session Storage

export PUMBAA_SESSION_DB=~/.pumbaa/sessions.db

Example Prompts

Try these prompts

  • "List my running workflows"
  • "What is the status of workflow abc-123?"
  • "Show me the outputs of workflow xyz-456"
  • "Read gs://bucket/path/to/file.txt"
  • "Query last 5 failed workflows"