instrumenting-with-mlflow-tracing

Instrument Python and TypeScript code with MLflow tracing for LLM and agent observability.

Updated Mar 25, 2026
One-click install
npx skills add https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher --skill instrumenting-with-mlflow-tracing-jingyiwng
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: instrumenting-with-mlflow-tracing
Source: https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher/tree/main/.claude/skills/instrumenting-with-mlflow-tracing
Command: npx skills add https://github.com/JingyiWng/databricks_ai_dev_kit_price_watcher --skill instrumenting-with-mlflow-tracing-jingyiwng

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

Many ML applications and agents run without visibility, making debugging, latency tracking, and quality evaluation hard. This Skill provides clear guidance to add MLflow Tracing to Python and TypeScript code, delivering end‑to‑end observability.

Core Features & Use Cases

  • Language‑specific guides for Python and TypeScript projects, directing users to the appropriate reference files.
  • Operation tracing for root workflows, LLM calls, retrieval, tool/function calls, agent decisions, and external services.
  • Verification steps to ensure traces are recorded and can be inspected with MLflow APIs.
  • Feedback collection mechanisms for user‑rated trace evaluation.
  • Production deployment guidance, including environment variables, async logging, sampling, and containerized setups.
  • Advanced patterns such as async function tracing, multi‑threading context propagation, and PII redaction.
  • Distributed tracing support for propagating trace context across services.

Quick Start

Add MLflow tracing to my agent for comprehensive observability.

Frequently Asked Questions about instrumenting-with-mlflow-tracing

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I add MLflow tracing to my LangChain agent for observability?

MLflow tracing instruments Python and TypeScript code to capture LLM calls, retrieval, tool usage, and agent decisions. It solves the lack of visibility in ML applications, making debugging, latency tracking, and quality evaluation significantly easier.

Does MLflow tracing work with TypeScript and Python agent frameworks?

Yes, MLflow tracing supports both TypeScript and Python projects. It provides language-specific guides to trace operations across frameworks including LangChain, LangGraph, OpenAI, DSPy, CrewAI, and AutoGen.

What's the best way to trace async functions and multi-threading context in MLflow?

MLflow tracing provides advanced patterns for async function tracing, multi-threading context propagation, and distributed tracing across services. It also supports PII redaction to protect sensitive data within captured traces.

Do I need a tracking server to inspect MLflow trace data?

Yes, a configured MLflow tracking server and the MLflow SDK are required to capture and query trace data. Verification steps ensure traces are properly recorded and can be inspected using MLflow APIs.

How do I deploy MLflow tracing in production with async logging and sampling?

Deploy MLflow tracing in production by configuring environment variables, async logging, sampling, and containerized setups. The Skill provides guidance to ensure traces are captured efficiently without blocking application performance.

Can I collect user feedback to evaluate MLflow trace quality?

Yes, MLflow tracing includes feedback collection mechanisms for user-rated trace evaluation. This allows you to assess the quality of agent decisions and LLM calls directly within the recorded trace data.