Trace Oh My Pi (omp) terminal coding sessions, model calls, tool usage, and token costs with Arize AX for full observability.Oh My Pi (omp) is a terminal coding agent that loads its extensions in-process inside its Bun runtime. The Arize Coding Harness Tracing instruments omp lifecycle events and exports OpenInference spans to Arize AX. Each agent run is captured as a trace with model calls, tool invocations, and inline token usage.
Launch Arize AX
To get started, sign up for a free Arize AX account and get your Space ID and API Key:- Log in at app.arize.com
- Click Settings and copy the Space ID
- Open the API Keys tab and create or copy an API key
Install
Curl installer
macOS / Linux:Local clone
~/.arize/harness/config.json, copies the hook shim into ~/.omp/extensions/arize-tracing.ts, and registers the shim’s absolute path in the extensions array of ~/.omp/agent/settings.json. omp does not auto-discover an extensions directory, so this explicit registration is required — the installer handles it for you.
Open a new omp session after install so the extension loads.
Configuration
Credentials live in~/.arize/harness/config.json. Environment variables override values in config.json and can be set in your shell profile before launching omp.
See the main README’s Environment variables section for the full list of runtime overrides.
Redaction controls
EachARIZE_LOG_* flag accepts "true" or "false" and defaults to "true". Set to "false" to opt out per category:
Observe
Once tracing is enabled, omp activity is streamed to Arize AX. There is one trace per agent run — a user prompt through the agent’s internal turn/tool-use loop to its final answer. Each trace is a tree:- Turn traces — the root span for each agent run, with the user prompt as input and the final assistant message as output
- LLM spans — one per model call in the loop, with model name, provider, prompt/completion/reasoning token counts, cache read/write tokens, and cost
- Tool spans — one per tool call, pairing the tool invocation with its result and recording name, input args, and output
- Session grouping — all runs from the same session grouped by
session.id
Spans Captured
omp exposes rich, once-fired lifecycle events that already carry final, structured data. The Arize AX hook forwards a small whitelist of them and emits the following spans:
Lifecycle events forwarded:
before_agent_start, turn_end, agent_end, and session_shutdown.
Verifying tracing
Run any omp session as you normally would. omp loads the registered extension on startup and forwards lifecycle events to the Arize AX hook.- Errors and handler stderr land in
~/.arize/harness/logs/omp.log. Setexport ARIZE_VERBOSE=truebefore launching omp to also see routine handler activity. - Set
export ARIZE_TRACE_DEBUG=trueto dump the raw event payloads under~/.arize/harness/state/debug/for inspection. - Confirm spans appear in your configured project in Arize AX.
Reference
For the full list of environment variables, default file paths, and troubleshooting steps, see the omp tracing README.Uninstall
extensions array in ~/.omp/agent/settings.json, deletes the hook file at ~/.omp/extensions/arize-tracing.ts (only if it carries the Arize header marker, so your own extensions are left alone), and removes the harnesses.omp block from ~/.arize/harness/config.json.