Arize AX offers 2 Model Context Protocol (MCP) servers that enhance your development workflow by integrating Arize support, documentation, and instrumentation guidance directly into your IDE or LLM.
Installation →Best for: Instrumentation guidance and implementation supportThe Arize Tracing Assistant provides hands-on help with adding tracing to your applications:
Instrumentation Guides: Step-by-step guides for adding Arize AX tracing to your codebase
Curated Examples: Framework-specific tracing examples and best practices
Direct Support Access: Natural language queries to Arize support directly from your development environment
Code Analysis: Analyzes your codebase to identify instrumentation needs and propose implementation plans
For comprehensive instrumentation assistance, use this prompt in your editor’s agent mode with the MCP server enabled:
Copy
Ask AI
Goal: Add Arize AX tracing to my application using the following workflow:PHASE 1 — ANALYSIS ONLY (READ-ONLY)- Scan the codebase to identify:- The primary application language (Python, Java, TypeScript, or JavaScript)- LLM calls, agent/orchestration frameworks, retrieval/embedding logic- Existing tracing, OpenTelemetry, or logging instrumentation- Infer the stack (LLM provider, framework, runtime)- Identify applicable OpenInference packages based on framework, platform, and/or LLM provider- When a framework is used, you should also include the OpenInference package for the LLM provider. The framework's OpenInference package does not include tracing for the LLM.- Identify where manual instrumentation is needed (if any)- Flag conflicts or overlapping instrumentation- Propose a minimal, safe instrumentation planRules for Phase 1:- Do NOT modify code- Do NOT create or write any files- Do NOT generate markdown summary files- Return a single concise summary + implementation planPHASE 2 — IMPLEMENTATION (ONLY AFTER PHASE 1)- Use auto-instrumentation first; manual spans only if required- Do NOT change business logic or behavior- Instrumentation must be optional and fail gracefully if env vars are missing- Create a single centralized instrumentation module (language-appropriate)- Avoid creating unnecessary files or markdown docsDeliverables (Phase 2, single response):- Dependency install commands (pip / npm / pnpm / maven / gradle as applicable)- Full instrumentation setup code (copy-paste ready)- Where/how to initialize at application startup- Required environment variables + .env example- Minimal before/after code examples- How to verify traces in Arize AX- Troubleshooting tipsBehavior:- Prefer inspection over mutation- Keep output concise and production-focused- Do not generate excessive documentation or summary files
The Arize AX Tracing Assistant is distributed via uv, a fast Python package manager. The AX Docs MCP Server provides HTTP-based access to the Arize AX documentation knowledge base.