amp-instrumentation 0.1.3

Automatic instrumentation for Python agent applications, monitored via WSO2 Agent Management Platform

WSO2 AI Agent Management Platform Instrumentation

Zero-code OpenTelemetry instrumentation for Python agents using the Traceloop SDK, with trace visibility in the WSO2 AI Agent Management Platform.

Overview

amp-instrumentation enables zero-code instrumentation for Python agents, automatically capturing traces for LLM calls, MCP requests, and other operations. It seamlessly wraps your agent’s execution with OpenTelemetry tracing powered by the Traceloop SDK.

Features

• Zero Code Changes: Instrument existing applications without modifying code
• Automatic Tracing: Traces LLM calls, MCP requests, database queries, and more
• OpenTelemetry Compatible: Uses industry-standard OpenTelemetry protocol
• Flexible Configuration: Configure via environment variables
• Framework Agnostic: Works with any Python application built using a wide range of agent frameworks supported by the TraceLoop SDK

Installation

pip install amp-instrumentation

Quick Start

1. Register Your Agent

First, register your agent at the WSO2 AI Agent Management Platform to obtain your agent API key and configuration details.

2. Set Required Environment Variables

export AMP_OTEL_ENDPOINT="https://amp-otel-endpoint.com" # AMP OTEL endpoint
export AMP_AGENT_API_KEY="your-agent-api-key" # Agent-specific key generated after registration

Optionally, you can set custom trace attributes:

export AMP_TRACE_ATTRIBUTES="environment-uid=env,component-uid=comp" # Optional: Custom key-value pairs for trace metadata

3. Run Your Application

Use the amp-instrument command to wrap your application run command:

# Run a Python script
amp-instrument python my_script.py

# Run with uvicorn
amp-instrument uvicorn app:main --reload

# Run with any package manager
amp-instrument poetry run python script.py
amp-instrument uv run python script.py

That's it! Your application is now instrumented and sending traces to the WSO2 AI Agent Management Platform.