instrumenting-with-mlflow-tracing

Instrument Python and TypeScript code with MLflow Tracing for AI workflows.

Updated May 20, 2026
One-click install
npx skills add https://github.com/ice-droid-99/BI-Dashboard-Generation-Agent-in-Databricks- --skill instrumenting-with-mlflow-tracing-ice-droid-99
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: instrumenting-with-mlflow-tracing
Source: https://github.com/ice-droid-99/BI-Dashboard-Generation-Agent-in-Databricks-/tree/main/.gemini/skills/instrumenting-with-mlflow-tracing
Command: npx skills add https://github.com/ice-droid-99/BI-Dashboard-Generation-Agent-in-Databricks- --skill instrumenting-with-mlflow-tracing-ice-droid-99

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

It resolves blind spots in AI/agent workflows by providing end-to-end observability for LLM calls, retrieval, tool execution, and agent decisions so you can quickly diagnose latency, errors, and quality issues.

Core Features & Use Cases

  • Trace end-to-end workflow execution: Capture root operations for pipelines and agent steps to understand where time is spent.
  • Trace LLM, retrieval, and tool calls: Record prompt/response details, token usage, retrieval activity, and external dependency calls for debugging.
  • Support common AI frameworks: Instrument popular stacks such as LangChain, LangGraph, OpenAI, DSPy, CrewAI, and AutoGen, especially when tracing setup is part of evaluation.

Quick Start

Ask the AI to instrument your project for MLflow tracing and to verify traces appear by running your agent once and then checking traces with mlflow.search_traces().

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 debug agent behavior?

Trace LLM calls and tool execution in MLflow by instrumenting your code paths during workflow setup. This records prompt and response details, token usage, and external dependency calls, allowing you to verify traces appear by running your agent and checking with mlflow.search_traces().

Does MLflow tracing work with LangChain and AutoGen frameworks?

Trace end-to-end agent workflow execution in MLflow to capture root operations for pipelines and agent steps. This provides visibility into where time is spent across your AI workflow, helping you quickly diagnose latency, errors, and quality issues.

How do I verify that MLflow traces are being logged correctly?

You need to load MLflow tracing during workflow setup to ensure it captures end-to-end workflow execution. Tracing must be initialized as part of the pipeline configuration to successfully record root operations, LLM calls, and tool interactions.

What is LLM observability and when do I need tracing for AI workflows?

Trace retrieval and tool calls in MLflow to debug agent workflows by recording retrieval activity and external dependency calls. This resolves blind spots in AI pipelines by capturing prompt and response details alongside token usage for debugging.

Can I instrument TypeScript code paths with MLflow tracing for agents?

Instrument TypeScript code paths with MLflow tracing to provide observability for AI agent workflows. It captures LLM calls, tool execution, and agent routing decisions across both Python and TypeScript implementations, supporting framework instrumentation.