@dbx-tools/appkit-mastra 0.6.13 → 0.6.15

package/lib/src/trace-io.js

@@ -0,0 +1,117 @@
+/**
+ * Copy each Mastra chat turn's request/response onto the active OTel
+ * root span so Databricks-managed MLflow can display them.
+ *
+ * Why this exists: MLflow's UC `<target>_trace_unified` view picks the
+ * one span whose `parent_span_id` is empty and reads request/response
+ * from `mlflow.spanInputs` / `mlflow.spanOutputs` (or the `gen_ai.*`
+ * equivalents) on THAT span only. Mastra records the turn on its
+ * `invoke_agent` child under `mastra.agent_run.input` / `.output`, keys
+ * the view never reads, so chat traces arrive with both columns null
+ * even when the spans themselves look healthy.
+ *
+ * The active span inside the Mastra sub-app is still AppKit's HTTP
+ * server span (OTel context is async-local). With `OTEL_PROPAGATORS=none`
+ * that span is the trace root - see the package README's observability
+ * section for why Apps ingress `traceparent` otherwise hides every turn.
+ *
+ * @module
+ */
+import { json, log, object } from "@dbx-tools/shared-core";
+import { trace } from "@opentelemetry/api";
+const logger = log.logger("mastra/trace-io");
+/** Cap on each payload copied onto a span, so one turn cannot bloat the export. */
+export const TRACE_IO_LIMIT = 8_000;
+/**
+ * Mount-relative Mastra agent invoke paths that carry a chat turn body
+ * (`messages`) and stream an assistant answer. Resume / approve verbs
+ * are intentionally excluded: their bodies are tool decisions, not the
+ * user prompt, and they rarely produce a fresh answer worth surfacing.
+ */
+const AGENT_TURN_ROUTE = /^\/agents\/[^/]+\/(stream|generate)(\/|$)/i;
+/** Attribute keys the MLflow UC `*_trace_unified` view reads from the root span. */
+export const MLFLOW_SPAN_INPUTS_ATTR = "mlflow.spanInputs";
+export const MLFLOW_SPAN_OUTPUTS_ATTR = "mlflow.spanOutputs";
+/**
+ * Concatenate the assistant's answer out of an AI SDK SSE transcript.
+ *
+ * The agent streams its reply as `text-delta` frames
+ * (`data: {"type":"text-delta","payload":{"text":"..."}}`), so the full
+ * answer only ever exists as deltas on the wire and has to be
+ * reassembled here.
+ */
+export function assistantTextFromSse(body) {
+    const parts = [];
+    for (const line of body.split("\n")) {
+        if (!line.startsWith("data:"))
+            continue;
+        // SSE carries comment/keepalive lines that are not JSON; they never hold
+        // assistant text, so skipping them is the intended path rather than an error.
+        const frame = json.parse(line.slice(5));
+        if (!object.isRecord(frame) || frame.type !== "text-delta")
+            continue;
+        const text = object.isRecord(frame.payload) ? frame.payload.text : undefined;
+        if (typeof text === "string")
+            parts.push(text);
+    }
+    return parts.join("");
+}
+/**
+ * Express middleware that stamps chat turn I/O onto the active OTel span.
+ *
+ * Tee `res.write` / `res.end` instead of listening for `finish`: the HTTP
+ * instrumentation ends the root span on finish, and an attribute set after
+ * a span ends never reaches the exporter.
+ *
+ * `path` is mount-relative (what the Mastra sub-app sees), e.g.
+ * `/agents/support/stream`.
+ */
+export function chatTurnTraceIoMiddleware(req, res, next) {
+    if (req.method !== "POST" || !AGENT_TURN_ROUTE.test(req.path)) {
+        next();
+        return;
+    }
+    const span = trace.getActiveSpan();
+    if (!span) {
+        next();
+        return;
+    }
+    const messages = req.body?.messages;
+    if (messages !== undefined) {
+        span.setAttribute(MLFLOW_SPAN_INPUTS_ATTR, JSON.stringify(messages).slice(0, TRACE_IO_LIMIT));
+    }
+    const chunks = [];
+    const passThroughWrite = res.write.bind(res);
+    const passThroughEnd = res.end.bind(res);
+    const collect = (chunk) => {
+        if (typeof chunk === "string")
+            chunks.push(chunk);
+        else if (Buffer.isBuffer(chunk))
+            chunks.push(chunk.toString("utf8"));
+    };
+    res.write = ((chunk, ...rest) => {
+        collect(chunk);
+        return passThroughWrite(chunk, ...rest);
+    });
+    res.end = ((chunk, ...rest) => {
+        collect(chunk);
+        const answer = assistantTextFromSse(chunks.join(""));
+        if (answer) {
+            span.setAttribute(MLFLOW_SPAN_OUTPUTS_ATTR, answer.slice(0, TRACE_IO_LIMIT));
+        }
+        return passThroughEnd(chunk, ...rest);
+    });
+    next();
+}
+/**
+ * Install {@link chatTurnTraceIoMiddleware} on a Mastra Express sub-app.
+ *
+ * Call before `MastraServer.init()` so the layer sits ahead of the agent
+ * routes. Safe to call unconditionally: with no active span (local, no
+ * OTLP) the middleware is a no-op.
+ */
+export function attachChatTurnTraceIo(app) {
+    app.use(chatTurnTraceIoMiddleware);
+    logger.info("chat turn I/O middleware attached");
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHJhY2UtaW8uanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyIuLi8uLi9zcmMvdHJhY2UtaW8udHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUE7Ozs7Ozs7Ozs7Ozs7Ozs7OztHQWtCRztBQUVILE9BQU8sRUFBRSxJQUFJLEVBQUUsR0FBRyxFQUFFLE1BQU0sRUFBRSxNQUFNLHdCQUF3QixDQUFDO0FBQzNELE9BQU8sRUFBRSxLQUFLLEVBQUUsTUFBTSxvQkFBb0IsQ0FBQztBQUczQyxNQUFNLE1BQU0sR0FBRyxHQUFHLENBQUMsTUFBTSxDQUFDLGlCQUFpQixDQUFDLENBQUM7QUFFN0MsbUZBQW1GO0FBQ25GLE1BQU0sQ0FBQyxNQUFNLGNBQWMsR0FBRyxLQUFLLENBQUM7QUFFcEM7Ozs7O0dBS0c7QUFDSCxNQUFNLGdCQUFnQixHQUFHLDRDQUE0QyxDQUFDO0FBRXRFLG9GQUFvRjtBQUNwRixNQUFNLENBQUMsTUFBTSx1QkFBdUIsR0FBRyxtQkFBbUIsQ0FBQztBQUMzRCxNQUFNLENBQUMsTUFBTSx3QkFBd0IsR0FBRyxvQkFBb0IsQ0FBQztBQUU3RDs7Ozs7OztHQU9HO0FBQ0gsTUFBTSxVQUFVLG9CQUFvQixDQUFDLElBQVk7SUFDL0MsTUFBTSxLQUFLLEdBQWEsRUFBRSxDQUFDO0lBQzNCLEtBQUssTUFBTSxJQUFJLElBQUksSUFBSSxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDO1FBQ3BDLElBQUksQ0FBQyxJQUFJLENBQUMsVUFBVSxDQUFDLE9BQU8sQ0FBQztZQUFFLFNBQVM7UUFDeEMseUVBQXlFO1FBQ3pFLDhFQUE4RTtRQUM5RSxNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQztRQUN4QyxJQUFJLENBQUMsTUFBTSxDQUFDLFFBQVEsQ0FBQyxLQUFLLENBQUMsSUFBSSxLQUFLLENBQUMsSUFBSSxLQUFLLFlBQVk7WUFBRSxTQUFTO1FBQ3JFLE1BQU0sSUFBSSxHQUFHLE1BQU0sQ0FBQyxRQUFRLENBQUMsS0FBSyxDQUFDLE9BQU8sQ0FBQyxDQUFDLENBQUMsQ0FBQyxLQUFLLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsU0FBUyxDQUFDO1FBQzdFLElBQUksT0FBTyxJQUFJLEtBQUssUUFBUTtZQUFFLEtBQUssQ0FBQyxJQUFJLENBQUMsSUFBSSxDQUFDLENBQUM7SUFDakQsQ0FBQztJQUNELE9BQU8sS0FBSyxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUMsQ0FBQztBQUN4QixDQUFDO0FBRUQ7Ozs7Ozs7OztHQVNHO0FBQ0gsTUFBTSxVQUFVLHlCQUF5QixDQUN2QyxHQUFvQixFQUNwQixHQUFxQixFQUNyQixJQUEwQjtJQUUxQixJQUFJLEdBQUcsQ0FBQyxNQUFNLEtBQUssTUFBTSxJQUFJLENBQUMsZ0JBQWdCLENBQUMsSUFBSSxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDO1FBQzlELElBQUksRUFBRSxDQUFDO1FBQ1AsT0FBTztJQUNULENBQUM7SUFDRCxNQUFNLElBQUksR0FBRyxLQUFLLENBQUMsYUFBYSxFQUFFLENBQUM7SUFDbkMsSUFBSSxDQUFDLElBQUksRUFBRSxDQUFDO1FBQ1YsSUFBSSxFQUFFLENBQUM7UUFDUCxPQUFPO0lBQ1QsQ0FBQztJQUVELE1BQU0sUUFBUSxHQUFJLEdBQUcsQ0FBQyxJQUEyQyxFQUFFLFFBQVEsQ0FBQztJQUM1RSxJQUFJLFFBQVEsS0FBSyxTQUFTLEVBQUUsQ0FBQztRQUMzQixJQUFJLENBQUMsWUFBWSxDQUFDLHVCQUF1QixFQUFFLElBQUksQ0FBQyxTQUFTLENBQUMsUUFBUSxDQUFDLENBQUMsS0FBSyxDQUFDLENBQUMsRUFBRSxjQUFjLENBQUMsQ0FBQyxDQUFDO0lBQ2hHLENBQUM7SUFFRCxNQUFNLE1BQU0sR0FBYSxFQUFFLENBQUM7SUFDNUIsTUFBTSxnQkFBZ0IsR0FBRyxHQUFHLENBQUMsS0FBSyxDQUFDLElBQUksQ0FBQyxHQUFHLENBQW9DLENBQUM7SUFDaEYsTUFBTSxjQUFjLEdBQUcsR0FBRyxDQUFDLEdBQUcsQ0FBQyxJQUFJLENBQUMsR0FBRyxDQUFvQyxDQUFDO0lBQzVFLE1BQU0sT0FBTyxHQUFHLENBQUMsS0FBYyxFQUFRLEVBQUU7UUFDdkMsSUFBSSxPQUFPLEtBQUssS0FBSyxRQUFRO1lBQUUsTUFBTSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsQ0FBQzthQUM3QyxJQUFJLE1BQU0sQ0FBQyxRQUFRLENBQUMsS0FBSyxDQUFDO1lBQUUsTUFBTSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsUUFBUSxDQUFDLE1BQU0sQ0FBQyxDQUFDLENBQUM7SUFDdkUsQ0FBQyxDQUFDO0lBRUYsR0FBRyxDQUFDLEtBQUssR0FBRyxDQUFDLENBQUMsS0FBYyxFQUFFLEdBQUcsSUFBZSxFQUFFLEVBQUU7UUFDbEQsT0FBTyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ2YsT0FBTyxnQkFBZ0IsQ0FBQyxLQUFLLEVBQUUsR0FBRyxJQUFJLENBQUMsQ0FBQztJQUMxQyxDQUFDLENBQXFCLENBQUM7SUFFdkIsR0FBRyxDQUFDLEdBQUcsR0FBRyxDQUFDLENBQUMsS0FBZSxFQUFFLEdBQUcsSUFBZSxFQUFFLEVBQUU7UUFDakQsT0FBTyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ2YsTUFBTSxNQUFNLEdBQUcsb0JBQW9CLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUMsQ0FBQyxDQUFDO1FBQ3JELElBQUksTUFBTSxFQUFFLENBQUM7WUFDWCxJQUFJLENBQUMsWUFBWSxDQUFDLHdCQUF3QixFQUFFLE1BQU0sQ0FBQyxLQUFLLENBQUMsQ0FBQyxFQUFFLGNBQWMsQ0FBQyxDQUFDLENBQUM7UUFDL0UsQ0FBQztRQUNELE9BQU8sY0FBYyxDQUFDLEtBQUssRUFBRSxHQUFHLElBQUksQ0FBQyxDQUFDO0lBQ3hDLENBQUMsQ0FBbUIsQ0FBQztJQUVyQixJQUFJLEVBQUUsQ0FBQztBQUNULENBQUM7QUFFRDs7Ozs7O0dBTUc7QUFDSCxNQUFNLFVBQVUscUJBQXFCLENBQUMsR0FBb0I7SUFDeEQsR0FBRyxDQUFDLEdBQUcsQ0FBQyx5QkFBeUIsQ0FBQyxDQUFDO0lBQ25DLE1BQU0sQ0FBQyxJQUFJLENBQUMsbUNBQW1DLENBQUMsQ0FBQztBQUNuRCxDQUFDIiwic291cmNlc0NvbnRlbnQiOlsiLyoqXG4gKiBDb3B5IGVhY2ggTWFzdHJhIGNoYXQgdHVybidzIHJlcXVlc3QvcmVzcG9uc2Ugb250byB0aGUgYWN0aXZlIE9UZWxcbiAqIHJvb3Qgc3BhbiBzbyBEYXRhYnJpY2tzLW1hbmFnZWQgTUxmbG93IGNhbiBkaXNwbGF5IHRoZW0uXG4gKlxuICogV2h5IHRoaXMgZXhpc3RzOiBNTGZsb3cncyBVQyBgPHRhcmdldD5fdHJhY2VfdW5pZmllZGAgdmlldyBwaWNrcyB0aGVcbiAqIG9uZSBzcGFuIHdob3NlIGBwYXJlbnRfc3Bhbl9pZGAgaXMgZW1wdHkgYW5kIHJlYWRzIHJlcXVlc3QvcmVzcG9uc2VcbiAqIGZyb20gYG1sZmxvdy5zcGFuSW5wdXRzYCAvIGBtbGZsb3cuc3Bhbk91dHB1dHNgIChvciB0aGUgYGdlbl9haS4qYFxuICogZXF1aXZhbGVudHMpIG9uIFRIQVQgc3BhbiBvbmx5LiBNYXN0cmEgcmVjb3JkcyB0aGUgdHVybiBvbiBpdHNcbiAqIGBpbnZva2VfYWdlbnRgIGNoaWxkIHVuZGVyIGBtYXN0cmEuYWdlbnRfcnVuLmlucHV0YCAvIGAub3V0cHV0YCwga2V5c1xuICogdGhlIHZpZXcgbmV2ZXIgcmVhZHMsIHNvIGNoYXQgdHJhY2VzIGFycml2ZSB3aXRoIGJvdGggY29sdW1ucyBudWxsXG4gKiBldmVuIHdoZW4gdGhlIHNwYW5zIHRoZW1zZWx2ZXMgbG9vayBoZWFsdGh5LlxuICpcbiAqIFRoZSBhY3RpdmUgc3BhbiBpbnNpZGUgdGhlIE1hc3RyYSBzdWItYXBwIGlzIHN0aWxsIEFwcEtpdCdzIEhUVFBcbiAqIHNlcnZlciBzcGFuIChPVGVsIGNvbnRleHQgaXMgYXN5bmMtbG9jYWwpLiBXaXRoIGBPVEVMX1BST1BBR0FUT1JTPW5vbmVgXG4gKiB0aGF0IHNwYW4gaXMgdGhlIHRyYWNlIHJvb3QgLSBzZWUgdGhlIHBhY2thZ2UgUkVBRE1FJ3Mgb2JzZXJ2YWJpbGl0eVxuICogc2VjdGlvbiBmb3Igd2h5IEFwcHMgaW5ncmVzcyBgdHJhY2VwYXJlbnRgIG90aGVyd2lzZSBoaWRlcyBldmVyeSB0dXJuLlxuICpcbiAqIEBtb2R1bGVcbiAqL1xuXG5pbXBvcnQgeyBqc29uLCBsb2csIG9iamVjdCB9IGZyb20gXCJAZGJ4LXRvb2xzL3NoYXJlZC1jb3JlXCI7XG5pbXBvcnQgeyB0cmFjZSB9IGZyb20gXCJAb3BlbnRlbGVtZXRyeS9hcGlcIjtcbmltcG9ydCB0eXBlIGV4cHJlc3MgZnJvbSBcImV4cHJlc3NcIjtcblxuY29uc3QgbG9nZ2VyID0gbG9nLmxvZ2dlcihcIm1hc3RyYS90cmFjZS1pb1wiKTtcblxuLyoqIENhcCBvbiBlYWNoIHBheWxvYWQgY29waWVkIG9udG8gYSBzcGFuLCBzbyBvbmUgdHVybiBjYW5ub3QgYmxvYXQgdGhlIGV4cG9ydC4gKi9cbmV4cG9ydCBjb25zdCBUUkFDRV9JT19MSU1JVCA9IDhfMDAwO1xuXG4vKipcbiAqIE1vdW50LXJlbGF0aXZlIE1hc3RyYSBhZ2VudCBpbnZva2UgcGF0aHMgdGhhdCBjYXJyeSBhIGNoYXQgdHVybiBib2R5XG4gKiAoYG1lc3NhZ2VzYCkgYW5kIHN0cmVhbSBhbiBhc3Npc3RhbnQgYW5zd2VyLiBSZXN1bWUgLyBhcHByb3ZlIHZlcmJzXG4gKiBhcmUgaW50ZW50aW9uYWxseSBleGNsdWRlZDogdGhlaXIgYm9kaWVzIGFyZSB0b29sIGRlY2lzaW9ucywgbm90IHRoZVxuICogdXNlciBwcm9tcHQsIGFuZCB0aGV5IHJhcmVseSBwcm9kdWNlIGEgZnJlc2ggYW5zd2VyIHdvcnRoIHN1cmZhY2luZy5cbiAqL1xuY29uc3QgQUdFTlRfVFVSTl9ST1VURSA9IC9eXFwvYWdlbnRzXFwvW14vXStcXC8oc3RyZWFtfGdlbmVyYXRlKShcXC98JCkvaTtcblxuLyoqIEF0dHJpYnV0ZSBrZXlzIHRoZSBNTGZsb3cgVUMgYCpfdHJhY2VfdW5pZmllZGAgdmlldyByZWFkcyBmcm9tIHRoZSByb290IHNwYW4uICovXG5leHBvcnQgY29uc3QgTUxGTE9XX1NQQU5fSU5QVVRTX0FUVFIgPSBcIm1sZmxvdy5zcGFuSW5wdXRzXCI7XG5leHBvcnQgY29uc3QgTUxGTE9XX1NQQU5fT1VUUFVUU19BVFRSID0gXCJtbGZsb3cuc3Bhbk91dHB1dHNcIjtcblxuLyoqXG4gKiBDb25jYXRlbmF0ZSB0aGUgYXNzaXN0YW50J3MgYW5zd2VyIG91dCBvZiBhbiBBSSBTREsgU1NFIHRyYW5zY3JpcHQuXG4gKlxuICogVGhlIGFnZW50IHN0cmVhbXMgaXRzIHJlcGx5IGFzIGB0ZXh0LWRlbHRhYCBmcmFtZXNcbiAqIChgZGF0YToge1widHlwZVwiOlwidGV4dC1kZWx0YVwiLFwicGF5bG9hZFwiOntcInRleHRcIjpcIi4uLlwifX1gKSwgc28gdGhlIGZ1bGxcbiAqIGFuc3dlciBvbmx5IGV2ZXIgZXhpc3RzIGFzIGRlbHRhcyBvbiB0aGUgd2lyZSBhbmQgaGFzIHRvIGJlXG4gKiByZWFzc2VtYmxlZCBoZXJlLlxuICovXG5leHBvcnQgZnVuY3Rpb24gYXNzaXN0YW50VGV4dEZyb21Tc2UoYm9keTogc3RyaW5nKTogc3RyaW5nIHtcbiAgY29uc3QgcGFydHM6IHN0cmluZ1tdID0gW107XG4gIGZvciAoY29uc3QgbGluZSBvZiBib2R5LnNwbGl0KFwiXFxuXCIpKSB7XG4gICAgaWYgKCFsaW5lLnN0YXJ0c1dpdGgoXCJkYXRhOlwiKSkgY29udGludWU7XG4gICAgLy8gU1NFIGNhcnJpZXMgY29tbWVudC9rZWVwYWxpdmUgbGluZXMgdGhhdCBhcmUgbm90IEpTT047IHRoZXkgbmV2ZXIgaG9sZFxuICAgIC8vIGFzc2lzdGFudCB0ZXh0LCBzbyBza2lwcGluZyB0aGVtIGlzIHRoZSBpbnRlbmRlZCBwYXRoIHJhdGhlciB0aGFuIGFuIGVycm9yLlxuICAgIGNvbnN0IGZyYW1lID0ganNvbi5wYXJzZShsaW5lLnNsaWNlKDUpKTtcbiAgICBpZiAoIW9iamVjdC5pc1JlY29yZChmcmFtZSkgfHwgZnJhbWUudHlwZSAhPT0gXCJ0ZXh0LWRlbHRhXCIpIGNvbnRpbnVlO1xuICAgIGNvbnN0IHRleHQgPSBvYmplY3QuaXNSZWNvcmQoZnJhbWUucGF5bG9hZCkgPyBmcmFtZS5wYXlsb2FkLnRleHQgOiB1bmRlZmluZWQ7XG4gICAgaWYgKHR5cGVvZiB0ZXh0ID09PSBcInN0cmluZ1wiKSBwYXJ0cy5wdXNoKHRleHQpO1xuICB9XG4gIHJldHVybiBwYXJ0cy5qb2luKFwiXCIpO1xufVxuXG4vKipcbiAqIEV4cHJlc3MgbWlkZGxld2FyZSB0aGF0IHN0YW1wcyBjaGF0IHR1cm4gSS9PIG9udG8gdGhlIGFjdGl2ZSBPVGVsIHNwYW4uXG4gKlxuICogVGVlIGByZXMud3JpdGVgIC8gYHJlcy5lbmRgIGluc3RlYWQgb2YgbGlzdGVuaW5nIGZvciBgZmluaXNoYDogdGhlIEhUVFBcbiAqIGluc3RydW1lbnRhdGlvbiBlbmRzIHRoZSByb290IHNwYW4gb24gZmluaXNoLCBhbmQgYW4gYXR0cmlidXRlIHNldCBhZnRlclxuICogYSBzcGFuIGVuZHMgbmV2ZXIgcmVhY2hlcyB0aGUgZXhwb3J0ZXIuXG4gKlxuICogYHBhdGhgIGlzIG1vdW50LXJlbGF0aXZlICh3aGF0IHRoZSBNYXN0cmEgc3ViLWFwcCBzZWVzKSwgZS5nLlxuICogYC9hZ2VudHMvc3VwcG9ydC9zdHJlYW1gLlxuICovXG5leHBvcnQgZnVuY3Rpb24gY2hhdFR1cm5UcmFjZUlvTWlkZGxld2FyZShcbiAgcmVxOiBleHByZXNzLlJlcXVlc3QsXG4gIHJlczogZXhwcmVzcy5SZXNwb25zZSxcbiAgbmV4dDogZXhwcmVzcy5OZXh0RnVuY3Rpb24sXG4pOiB2b2lkIHtcbiAgaWYgKHJlcS5tZXRob2QgIT09IFwiUE9TVFwiIHx8ICFBR0VOVF9UVVJOX1JPVVRFLnRlc3QocmVxLnBhdGgpKSB7XG4gICAgbmV4dCgpO1xuICAgIHJldHVybjtcbiAgfVxuICBjb25zdCBzcGFuID0gdHJhY2UuZ2V0QWN0aXZlU3BhbigpO1xuICBpZiAoIXNwYW4pIHtcbiAgICBuZXh0KCk7XG4gICAgcmV0dXJuO1xuICB9XG5cbiAgY29uc3QgbWVzc2FnZXMgPSAocmVxLmJvZHkgYXMgeyBtZXNzYWdlcz86IHVua25vd24gfSB8IHVuZGVmaW5lZCk/Lm1lc3NhZ2VzO1xuICBpZiAobWVzc2FnZXMgIT09IHVuZGVmaW5lZCkge1xuICAgIHNwYW4uc2V0QXR0cmlidXRlKE1MRkxPV19TUEFOX0lOUFVUU19BVFRSLCBKU09OLnN0cmluZ2lmeShtZXNzYWdlcykuc2xpY2UoMCwgVFJBQ0VfSU9fTElNSVQpKTtcbiAgfVxuXG4gIGNvbnN0IGNodW5rczogc3RyaW5nW10gPSBbXTtcbiAgY29uc3QgcGFzc1Rocm91Z2hXcml0ZSA9IHJlcy53cml0ZS5iaW5kKHJlcykgYXMgKC4uLmFyZ3M6IHVua25vd25bXSkgPT4gYm9vbGVhbjtcbiAgY29uc3QgcGFzc1Rocm91Z2hFbmQgPSByZXMuZW5kLmJpbmQocmVzKSBhcyAoLi4uYXJnczogdW5rbm93bltdKSA9PiB1bmtub3duO1xuICBjb25zdCBjb2xsZWN0ID0gKGNodW5rOiB1bmtub3duKTogdm9pZCA9PiB7XG4gICAgaWYgKHR5cGVvZiBjaHVuayA9PT0gXCJzdHJpbmdcIikgY2h1bmtzLnB1c2goY2h1bmspO1xuICAgIGVsc2UgaWYgKEJ1ZmZlci5pc0J1ZmZlcihjaHVuaykpIGNodW5rcy5wdXNoKGNodW5rLnRvU3RyaW5nKFwidXRmOFwiKSk7XG4gIH07XG5cbiAgcmVzLndyaXRlID0gKChjaHVuazogdW5rbm93biwgLi4ucmVzdDogdW5rbm93bltdKSA9PiB7XG4gICAgY29sbGVjdChjaHVuayk7XG4gICAgcmV0dXJuIHBhc3NUaHJvdWdoV3JpdGUoY2h1bmssIC4uLnJlc3QpO1xuICB9KSBhcyB0eXBlb2YgcmVzLndyaXRlO1xuXG4gIHJlcy5lbmQgPSAoKGNodW5rPzogdW5rbm93biwgLi4ucmVzdDogdW5rbm93bltdKSA9PiB7XG4gICAgY29sbGVjdChjaHVuayk7XG4gICAgY29uc3QgYW5zd2VyID0gYXNzaXN0YW50VGV4dEZyb21Tc2UoY2h1bmtzLmpvaW4oXCJcIikpO1xuICAgIGlmIChhbnN3ZXIpIHtcbiAgICAgIHNwYW4uc2V0QXR0cmlidXRlKE1MRkxPV19TUEFOX09VVFBVVFNfQVRUUiwgYW5zd2VyLnNsaWNlKDAsIFRSQUNFX0lPX0xJTUlUKSk7XG4gICAgfVxuICAgIHJldHVybiBwYXNzVGhyb3VnaEVuZChjaHVuaywgLi4ucmVzdCk7XG4gIH0pIGFzIHR5cGVvZiByZXMuZW5kO1xuXG4gIG5leHQoKTtcbn1cblxuLyoqXG4gKiBJbnN0YWxsIHtAbGluayBjaGF0VHVyblRyYWNlSW9NaWRkbGV3YXJlfSBvbiBhIE1hc3RyYSBFeHByZXNzIHN1Yi1hcHAuXG4gKlxuICogQ2FsbCBiZWZvcmUgYE1hc3RyYVNlcnZlci5pbml0KClgIHNvIHRoZSBsYXllciBzaXRzIGFoZWFkIG9mIHRoZSBhZ2VudFxuICogcm91dGVzLiBTYWZlIHRvIGNhbGwgdW5jb25kaXRpb25hbGx5OiB3aXRoIG5vIGFjdGl2ZSBzcGFuIChsb2NhbCwgbm9cbiAqIE9UTFApIHRoZSBtaWRkbGV3YXJlIGlzIGEgbm8tb3AuXG4gKi9cbmV4cG9ydCBmdW5jdGlvbiBhdHRhY2hDaGF0VHVyblRyYWNlSW8oYXBwOiBleHByZXNzLkV4cHJlc3MpOiB2b2lkIHtcbiAgYXBwLnVzZShjaGF0VHVyblRyYWNlSW9NaWRkbGV3YXJlKTtcbiAgbG9nZ2VyLmluZm8oXCJjaGF0IHR1cm4gSS9PIG1pZGRsZXdhcmUgYXR0YWNoZWRcIik7XG59XG4iXX0=

package/lib/tsconfig.tsbuildinfo

@@ -1 +1 @@
-{"root":["../src/agents.ts","../src/chart.ts","../src/config.ts","../src/defaults.ts","../src/filesystems.ts","../src/genie.ts","../src/history.ts","../src/identity.ts","../src/mcp.ts","../src/memory.ts","../src/mlflow.ts","../src/model.ts","../src/observability.ts","../src/pagination.ts","../src/plugin.ts","../src/processors.ts","../src/remote-skills.ts","../src/rest.ts","../src/server.ts","../src/serving-sanitize.ts","../src/serving.ts","../src/statement.ts","../src/storage-schema.ts","../src/summarize.ts","../src/threads.ts","../src/validation.ts","../src/workspaces.ts","../src/writer.ts","../index.ts"],"version":"5.9.3"}
+{"root":["../src/agents.ts","../src/chart.ts","../src/config.ts","../src/defaults.ts","../src/filesystems.ts","../src/genie.ts","../src/history.ts","../src/identity.ts","../src/mcp.ts","../src/memory.ts","../src/mlflow.ts","../src/model.ts","../src/observability.ts","../src/pagination.ts","../src/plugin.ts","../src/processors.ts","../src/remote-skills.ts","../src/rest.ts","../src/server.ts","../src/serving-sanitize.ts","../src/serving.ts","../src/statement.ts","../src/storage-schema.ts","../src/summarize.ts","../src/threads.ts","../src/trace-io.ts","../src/validation.ts","../src/workspaces.ts","../src/writer.ts","../index.ts"],"version":"5.9.3"}

package/package.json

@@ -32,25 +32,25 @@
     "express": "^5.1.0",
     "pg": "^8.22.0",
     "zod": "4.3.6",
-    "@dbx-tools/appkit": "0.6.13",
-    "@dbx-tools/core": "0.6.13",
-    "@dbx-tools/databricks": "0.6.13",
-    "@dbx-tools/genie": "0.6.13",
-    "@dbx-tools/fs": "0.6.13",
-    "@dbx-tools/model": "0.6.13",
-    "@dbx-tools/path": "0.6.13",
-    "@dbx-tools/shared-genie": "0.6.13",
-    "@dbx-tools/shared-model": "0.6.13",
-    "@dbx-tools/shared-fs": "0.6.13",
-    "@dbx-tools/shared-core": "0.6.13",
-    "@dbx-tools/shared-mastra": "0.6.13"
+    "@dbx-tools/core": "0.6.15",
+    "@dbx-tools/appkit": "0.6.15",
+    "@dbx-tools/databricks": "0.6.15",
+    "@dbx-tools/fs": "0.6.15",
+    "@dbx-tools/model": "0.6.15",
+    "@dbx-tools/path": "0.6.15",
+    "@dbx-tools/shared-genie": "0.6.15",
+    "@dbx-tools/shared-core": "0.6.15",
+    "@dbx-tools/shared-fs": "0.6.15",
+    "@dbx-tools/shared-mastra": "0.6.15",
+    "@dbx-tools/genie": "0.6.15",
+    "@dbx-tools/shared-model": "0.6.15"
   },
   "main": "./lib/index.js",
   "license": "UNLICENSED",
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.6.13",
+  "version": "0.6.15",
   "types": "./lib/index.d.ts",
   "type": "module",
   "exports": {

package/src/mlflow.ts

@@ -5,7 +5,9 @@
  *
  * Feedback attaches to a trace, and the plugin's spans reach MLflow
  * through the same OTel pipeline as every other AppKit span (see
- * `observability.ts`). MLflow derives its trace id from the OpenTelemetry
+ * `observability.ts`). On Databricks Apps that pipeline is the UC
+ * sidecar injected by `telemetry_export_destinations`, not a direct
+ * workspace OTLP URL. MLflow derives its trace id from the OpenTelemetry
  * trace id (`tr-<hex(otelTraceId)>`), so the server stamps the active
  * trace id on each turn's response and the client sends it back here.
  *

package/src/observability.ts

@@ -30,6 +30,15 @@
  * registered at all (unless `observability: true` forces it on), so
  * Mastra does not emit `[OtelBridge] No OTEL span found` warnings.
  *
+ * On Databricks Apps the supported sink is Unity Catalog via
+ * `telemetry_export_destinations` (the platform injects the local OTLP
+ * sidecar). Managed MLflow has no OTLP ingest endpoint - do not point
+ * `OTEL_EXPORTER_OTLP_*` at the workspace host. Set
+ * `OTEL_PROPAGATORS=none` so Apps ingress `traceparent` does not hide
+ * every request-scoped root span from the UC `*_trace_unified` view.
+ * Chat request/response on that root span is stamped by
+ * `traceIo.attachChatTurnTraceIo` (see `trace-io.ts`).
+ *
  * @module
  */
 
@@ -39,6 +48,7 @@ import { Observability } from "@mastra/observability";
 import { OtelBridge } from "@mastra/otel-bridge";
 
 import { TRACE_REQUEST_CONTEXT_KEYS } from "./config.ts";
+import { mlflowEnabled } from "./mlflow.ts";
 
 const logger = log.logger("mastra/observability");
 
@@ -127,11 +137,17 @@ export async function buildObservability(
     : otelBase
       ? `${otelBase.replace(/\/+$/, "")}/v1/traces`
       : undefined;
+  // `mlflowEnabled` is the same gate that flips chat feedback on, so the
+  // boot line is the one-glance check that both halves of the Apps UC
+  // pipeline (OTLP sidecar + experiment id) are present.
+  const feedback = mlflowEnabled();
   logger.info("Mastra observability wired through OTel bridge", {
     serviceName,
     requestContextKeys,
     otelBase: otelBase ?? "<unset>",
     resolvedTracesUrl: resolvedTracesUrl ?? "<unset>",
+    feedback,
+    observability: feedback ? "mlflow" : "otel",
   });
 
   return new Observability({

package/src/plugin.ts

@@ -103,6 +103,7 @@ import {
   isMastraRequestAllowed,
   MastraServer,
 } from "./server.ts";
+import { attachChatTurnTraceIo } from "./trace-io.ts";
 import { resolveServingConfig } from "./serving.ts";
 import { fetchStatementData, STATEMENT_ROW_CAP } from "./statement.ts";
 import { threadsRoute } from "./threads.ts";
@@ -1061,6 +1062,12 @@ export class MastraPlugin extends Plugin<MastraPluginConfig> {
     });
     this.mastraApp = express();
     attachRoutePatchMiddleware(this.mastraApp);
+    // Stamp chat request/response onto the HTTP root span so MLflow's
+    // UC `*_trace_unified` view can surface them (Mastra's own
+    // `mastra.agent_run.*` attributes sit on a child span the view
+    // never reads). Must run before `MastraServer.init` so the layer
+    // sits ahead of the agent routes.
+    attachChatTurnTraceIo(this.mastraApp);
     this.mastraServer = new MastraServer(this.config, {
       app: this.mastraApp,
       mastra: this.mastra,

package/src/trace-io.ts

@@ -0,0 +1,129 @@
+/**
+ * Copy each Mastra chat turn's request/response onto the active OTel
+ * root span so Databricks-managed MLflow can display them.
+ *
+ * Why this exists: MLflow's UC `<target>_trace_unified` view picks the
+ * one span whose `parent_span_id` is empty and reads request/response
+ * from `mlflow.spanInputs` / `mlflow.spanOutputs` (or the `gen_ai.*`
+ * equivalents) on THAT span only. Mastra records the turn on its
+ * `invoke_agent` child under `mastra.agent_run.input` / `.output`, keys
+ * the view never reads, so chat traces arrive with both columns null
+ * even when the spans themselves look healthy.
+ *
+ * The active span inside the Mastra sub-app is still AppKit's HTTP
+ * server span (OTel context is async-local). With `OTEL_PROPAGATORS=none`
+ * that span is the trace root - see the package README's observability
+ * section for why Apps ingress `traceparent` otherwise hides every turn.
+ *
+ * @module
+ */
+
+import { json, log, object } from "@dbx-tools/shared-core";
+import { trace } from "@opentelemetry/api";
+import type express from "express";
+
+const logger = log.logger("mastra/trace-io");
+
+/** Cap on each payload copied onto a span, so one turn cannot bloat the export. */
+export const TRACE_IO_LIMIT = 8_000;
+
+/**
+ * Mount-relative Mastra agent invoke paths that carry a chat turn body
+ * (`messages`) and stream an assistant answer. Resume / approve verbs
+ * are intentionally excluded: their bodies are tool decisions, not the
+ * user prompt, and they rarely produce a fresh answer worth surfacing.
+ */
+const AGENT_TURN_ROUTE = /^\/agents\/[^/]+\/(stream|generate)(\/|$)/i;
+
+/** Attribute keys the MLflow UC `*_trace_unified` view reads from the root span. */
+export const MLFLOW_SPAN_INPUTS_ATTR = "mlflow.spanInputs";
+export const MLFLOW_SPAN_OUTPUTS_ATTR = "mlflow.spanOutputs";
+
+/**
+ * Concatenate the assistant's answer out of an AI SDK SSE transcript.
+ *
+ * The agent streams its reply as `text-delta` frames
+ * (`data: {"type":"text-delta","payload":{"text":"..."}}`), so the full
+ * answer only ever exists as deltas on the wire and has to be
+ * reassembled here.
+ */
+export function assistantTextFromSse(body: string): string {
+  const parts: string[] = [];
+  for (const line of body.split("\n")) {
+    if (!line.startsWith("data:")) continue;
+    // SSE carries comment/keepalive lines that are not JSON; they never hold
+    // assistant text, so skipping them is the intended path rather than an error.
+    const frame = json.parse(line.slice(5));
+    if (!object.isRecord(frame) || frame.type !== "text-delta") continue;
+    const text = object.isRecord(frame.payload) ? frame.payload.text : undefined;
+    if (typeof text === "string") parts.push(text);
+  }
+  return parts.join("");
+}
+
+/**
+ * Express middleware that stamps chat turn I/O onto the active OTel span.
+ *
+ * Tee `res.write` / `res.end` instead of listening for `finish`: the HTTP
+ * instrumentation ends the root span on finish, and an attribute set after
+ * a span ends never reaches the exporter.
+ *
+ * `path` is mount-relative (what the Mastra sub-app sees), e.g.
+ * `/agents/support/stream`.
+ */
+export function chatTurnTraceIoMiddleware(
+  req: express.Request,
+  res: express.Response,
+  next: express.NextFunction,
+): void {
+  if (req.method !== "POST" || !AGENT_TURN_ROUTE.test(req.path)) {
+    next();
+    return;
+  }
+  const span = trace.getActiveSpan();
+  if (!span) {
+    next();
+    return;
+  }
+
+  const messages = (req.body as { messages?: unknown } | undefined)?.messages;
+  if (messages !== undefined) {
+    span.setAttribute(MLFLOW_SPAN_INPUTS_ATTR, JSON.stringify(messages).slice(0, TRACE_IO_LIMIT));
+  }
+
+  const chunks: string[] = [];
+  const passThroughWrite = res.write.bind(res) as (...args: unknown[]) => boolean;
+  const passThroughEnd = res.end.bind(res) as (...args: unknown[]) => unknown;
+  const collect = (chunk: unknown): void => {
+    if (typeof chunk === "string") chunks.push(chunk);
+    else if (Buffer.isBuffer(chunk)) chunks.push(chunk.toString("utf8"));
+  };
+
+  res.write = ((chunk: unknown, ...rest: unknown[]) => {
+    collect(chunk);
+    return passThroughWrite(chunk, ...rest);
+  }) as typeof res.write;
+
+  res.end = ((chunk?: unknown, ...rest: unknown[]) => {
+    collect(chunk);
+    const answer = assistantTextFromSse(chunks.join(""));
+    if (answer) {
+      span.setAttribute(MLFLOW_SPAN_OUTPUTS_ATTR, answer.slice(0, TRACE_IO_LIMIT));
+    }
+    return passThroughEnd(chunk, ...rest);
+  }) as typeof res.end;
+
+  next();
+}
+
+/**
+ * Install {@link chatTurnTraceIoMiddleware} on a Mastra Express sub-app.
+ *
+ * Call before `MastraServer.init()` so the layer sits ahead of the agent
+ * routes. Safe to call unconditionally: with no active span (local, no
+ * OTLP) the middleware is a no-op.
+ */
+export function attachChatTurnTraceIo(app: express.Express): void {
+  app.use(chatTurnTraceIoMiddleware);
+  logger.info("chat turn I/O middleware attached");
+}