@databricks/appkit 0.59.0 → 0.60.0

package/dist/appkit/package.js

@@ -1,6 +1,6 @@
 //#region package.json
 var name = "@databricks/appkit";
-var version = "0.59.0";
+var version = "0.60.0";
 
 //#endregion
 export { name, version };

package/dist/plugins/agents/agents.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agents.d.ts","names":[],"sources":["../../../src/plugins/agents/agents.ts"],"mappings":";;;;;;;;;;cAsIa,YAAA,SAAqB,MAAA,YAAkB,YAAA;EAAA,OAC3C,QAAA,EAAuB,cAAA;EAAA,OACvB,KAAA,EAAO,WAAA;EAAA,UAEI,MAAA,EAAQ,kBAAA;EAAA,QAElB,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,aAAA;EARgB;;;;;;EAAA,QAkBhB,gBAAA;EAAA,QACA,SAAA;EAAA,QACA,WAAA;EAAA,QACA,YAAA;cAEI,MAAA,EAAQ,kBAAA;EA8rBJ;;;;;;;;;;EAAA,QA/pBR,oBAAA;EAAA,YAKI,sBAAA,CAAA;EA3DoB;EAAA,YAqFpB,cAAA,CAAA;EApFL;EAAA,QAuGC,gBAAA;EAtGD;;;;;;EAAA,QAgHC,WAAA;EAhGA;;;;;EAAA,QAiHA,aAAA;EAYF,KAAA,CAAA,GAAK,OAAA;EAzFH;;;;;;EAuGF,MAAA,CAAA,GAAU,OAAA;EAdL;;;;;EAAA,QAwCG,kBAAA;EAAA,QAmEN,iBAAA;EAAA,QAMM,mBAAA;EAoEA;;;;;;EAAA,QA1CN,mBAAA;EAAA,QAmBM,oBAAA;EAAA,QAuBA,cAAA;EAyUY;;;;;EAAA,QA7RZ,cAAA;EAqWE;;;;;;;;;;;EAAA,QArPR,eAAA;EA6iCM;;;;;;;;;;;;EAAA,QAjhCN,eAAA;EAAA,QAkBM,gBAAA;EAAA,QA0DA,kBAAA;EAiEd,aAAA,CAAA,GAAiB,mBAAA;EAIX,gBAAA,CAAA,GAAoB,OAAA;;;;;;;UAYlB,iBAAA;EAUR,YAAA,CAAa,MAAA,EAAQ,UAAA;EAkDrB,YAAA,CAAA,GAAgB,MAAA;EAAA,QAOF,WAAA;;;;;;;;;;;;UA6EN,gCAAA;;;;;;;;;;;UAuBM,aAAA;EAAA,QAoFA,YAAA;;;;;;;;;;;;;;;;;;;;;;;;UA6LA,qBAAA;;;;;;;;;;;;UA6IA,gBAAA;;;;;;;;;;;;;;;UAyHA,WAAA;EAAA,QAiGA,aAAA;EAAA,QA2BA,cAAA;EAAA,QAuCA,kBAAA;EAAA,QASA,gBAAA;EAAA,QAUA,mBAAA;EAAA,QAaN,YAAA;EAAA,QASA,aAAA;EAgBF,QAAA,CAAA,GAAY,OAAA;EAQlB,OAAA,CAAA;6BAE2B,GAAA,EAAO,eAAA,KAAe,OAAA;;2BAG3B,eAAA;;;oCAGS,OAAA,CAAA,MAAA;EAAA;EAAA,QAIjB,iBAAA;AAAA;;;;;;;;;;;;;;;;cAsJH,MAAA,EAAM,QAAA,QAAA,YAAA,EAAA,kBAAA"}
+{"version":3,"file":"agents.d.ts","names":[],"sources":["../../../src/plugins/agents/agents.ts"],"mappings":";;;;;;;;;;cA6Ia,YAAA,SAAqB,MAAA,YAAkB,YAAA;EAAA,OAK3C,QAAA,EAAkC,cAAA;EAAA,OAClC,KAAA,EAAO,WAAA;EAAA,UAEI,MAAA,EAAQ,kBAAA;EAAA,QAElB,MAAA;EAAA,QACA,gBAAA;EAAA,QACA,aAAA;EAZgB;;;;;;EAAA,QAsBhB,gBAAA;EAAA,QACA,SAAA;EAAA,QACA,WAAA;EAAA,QACA,YAAA;cAEI,MAAA,EAAQ,kBAAA;EA+rBJ;;;;;;;;;;EAAA,QAhqBR,oBAAA;EAAA,YAKI,sBAAA,CAAA;EA/DoB;EAAA,YAyFpB,cAAA,CAAA;EApFL;EAAA,QAuGC,gBAAA;EAtGD;;;;;;EAAA,QAgHC,WAAA;EAhGA;;;;;EAAA,QAiHA,aAAA;EAYF,KAAA,CAAA,GAAK,OAAA;EAzFH;;;;;;EAwGF,MAAA,CAAA,GAAU,OAAA;EAfL;;;;;EAAA,QAyCG,kBAAA;EAAA,QAmEN,iBAAA;EAAA,QAMM,mBAAA;EAoEA;;;;;;EAAA,QA1CN,mBAAA;EAAA,QAmBM,oBAAA;EAAA,QAuBA,cAAA;EAyUY;;;;;EAAA,QA7RZ,cAAA;EAqWE;;;;;;;;;;;EAAA,QArPR,eAAA;EAynCM;;;;;;;;;;;;EAAA,QA7lCN,eAAA;EAAA,QAkBM,gBAAA;EAAA,QA0DA,kBAAA;EAiEd,aAAA,CAAA,GAAiB,mBAAA;EAIX,gBAAA,CAAA,GAAoB,OAAA;;;;;;;UAYlB,iBAAA;EAUR,YAAA,CAAa,MAAA,EAAQ,UAAA;EAkDrB,YAAA,CAAA,GAAgB,MAAA;EAAA,QAOF,WAAA;;;;;;;;;;;;UA6EN,gCAAA;;;;;;;;;;;UAuBM,aAAA;EAAA,QA2FA,YAAA;;;;;;;;;;;;;;;;;;;;;;;;UA2NA,qBAAA;;;;;;;;;;;;UA0KA,gBAAA;;;;;;;;;;;;;;;UAmIA,WAAA;EAAA,QAiGA,aAAA;EAAA,QA2BA,cAAA;EAAA,QAuCA,kBAAA;EAAA,QASA,gBAAA;EAAA,QAUA,mBAAA;EAAA,QAaN,YAAA;EAAA,QASA,aAAA;EAgBF,QAAA,CAAA,GAAY,OAAA;EAQlB,OAAA,CAAA;6BAE2B,GAAA,EAAO,eAAA,KAAe,OAAA;;2BAG3B,eAAA;;;oCAGS,OAAA,CAAA,MAAA;EAAA;EAAA,QAIjB,iBAAA;AAAA;;;;;;;;;;;;;;;;cAsJH,MAAA,EAAM,QAAA,QAAA,YAAA,EAAA,kBAAA"}

package/dist/plugins/agents/agents.js

@@ -20,6 +20,7 @@ import { agentStreamDefaults } from "./defaults.js";
 import { EventChannel } from "./event-channel.js";
 import { AgentEventTranslator } from "./event-translator.js";
 import manifest_default from "./manifest.js";
+import { currentTraceId, initAgentTracing, linkTraceToRun, traceAgent, traceTool } from "./mlflow.js";
 import { approvalRequestSchema, cancelRequestSchema, chatRequestSchema, invocationsRequestSchema } from "./schemas.js";
 import { InMemoryThreadStore } from "./thread-store.js";
 import { ToolApprovalGate } from "./tool-approval-gate.js";
@@ -147,6 +148,7 @@ var AgentsPlugin = class extends Plugin {
 		else this.userStreamCounts.set(entry.userId, next);
 	}
 	async setup() {
+		await initAgentTracing();
 		const { agents, defaultAgentName } = await this.buildAgentRegistry();
 		this.agents = agents;
 		this.defaultAgentName = defaultAgentName;
@@ -550,7 +552,7 @@ var AgentsPlugin = class extends Plugin {
 			});
 			return;
 		}
-		const { message, threadId, agent: agentName } = parsed.data;
+		const { message, threadId, agent: agentName, mlflowRunId } = parsed.data;
 		const registered = this.resolveAgent(agentName);
 		if (!registered) {
 			res.status(400).json({ error: agentName ? `Agent "${agentName}" not found` : "No agent registered" });
@@ -583,7 +585,7 @@ var AgentsPlugin = class extends Plugin {
 			res.status(500).json({ error: "Thread operation failed" });
 			return;
 		}
-		return this._streamAgent(req, res, registered, thread, userId);
+		return this._streamAgent(req, res, registered, thread, userId, mlflowRunId);
 	}
 	/**
 	* Returns the names of tools in `registered.toolIndex` whose annotations
@@ -621,7 +623,7 @@ var AgentsPlugin = class extends Plugin {
 			});
 			return;
 		}
-		const { input } = parsed.data;
+		const { input, mlflowRunId } = parsed.data;
 		const registered = this.resolveAgent();
 		if (!registered) {
 			res.status(400).json({ error: "No agent registered" });
@@ -664,9 +666,9 @@ var AgentsPlugin = class extends Plugin {
 			res.status(500).json({ error: "Thread operation failed" });
 			return;
 		}
-		return this._runAgentNonStreaming(req, res, registered, thread, userId);
+		return this._runAgentNonStreaming(req, res, registered, thread, userId, mlflowRunId);
 	}
-	async _streamAgent(req, res, registered, thread, userId) {
+	async _streamAgent(req, res, registered, thread, userId, mlflowRunId) {
 		const abortController = new AbortController();
 		const signal = abortController.signal;
 		const requestId = randomUUID();
@@ -695,37 +697,54 @@ var AgentsPlugin = class extends Plugin {
 					type: "metadata",
 					data: { threadId: thread.id }
 				})) outboundEvents.push(evt);
-				const pluginNames = this.context ? this.context.getPluginNames().filter((n) => n !== this.name && n !== "server") : [];
-				const messagesWithSystem = [{
-					id: "system",
-					role: "system",
-					content: composePromptForAgent(registered, this.config.baseSystemPrompt, {
-						agentName: registered.name,
-						pluginNames,
-						toolNames: tools.map((t) => t.name)
-					}),
-					createdAt: /* @__PURE__ */ new Date()
-				}, ...thread.messages];
-				const fullContent = await consumeAdapterStream(registered.adapter.run({
-					messages: messagesWithSystem,
-					tools,
-					threadId: thread.id,
-					signal,
-					extensions: buildAdapterExtensions(registered.toolIndex)
-				}, {
-					executeTool,
-					signal
-				}), {
-					signal,
-					onEvent: (event) => {
-						for (const translated of translator.translate(event)) outboundEvents.push(translated);
+				await traceAgent(registered.name ?? "agent", { messages: thread.messages.map((m) => ({
+					role: m.role,
+					content: m.content
+				})) }, async (span) => {
+					if (mlflowRunId) linkTraceToRun(mlflowRunId);
+					const pluginNames = this.context ? this.context.getPluginNames().filter((n) => n !== this.name && n !== "server") : [];
+					const messagesWithSystem = [{
+						id: "system",
+						role: "system",
+						content: composePromptForAgent(registered, this.config.baseSystemPrompt, {
+							agentName: registered.name,
+							pluginNames,
+							toolNames: tools.map((t) => t.name)
+						}),
+						createdAt: /* @__PURE__ */ new Date()
+					}, ...thread.messages];
+					const fullContent = await consumeAdapterStream(registered.adapter.run({
+						messages: messagesWithSystem,
+						tools,
+						threadId: thread.id,
+						signal,
+						extensions: buildAdapterExtensions(registered.toolIndex)
+					}, {
+						executeTool,
+						signal
+					}), {
+						signal,
+						onEvent: (event) => {
+							for (const translated of translator.translate(event)) outboundEvents.push(translated);
+						}
+					});
+					if (fullContent) {
+						span.setOutputs({
+							role: "assistant",
+							content: fullContent
+						});
+						await this.threadStore.addMessage(thread.id, userId, {
+							id: randomUUID(),
+							role: "assistant",
+							content: fullContent,
+							createdAt: /* @__PURE__ */ new Date()
+						});
 					}
-				});
-				if (fullContent) await this.threadStore.addMessage(thread.id, userId, {
-					id: randomUUID(),
-					role: "assistant",
-					content: fullContent,
-					createdAt: /* @__PURE__ */ new Date()
+					const mlflowTraceId = currentTraceId();
+					if (mlflowTraceId) for (const evt of translator.translate({
+						type: "metadata",
+						data: { mlflowTraceId }
+					})) outboundEvents.push(evt);
 				});
 				for (const evt of translator.finalize()) outboundEvents.push(evt);
 			} catch (error) {
@@ -784,12 +803,14 @@ var AgentsPlugin = class extends Plugin {
 	* synthesis (which would have been a coding bug given the precheck) is
 	* a dropped no-op instead of a runtime crash.
 	*/
-	async _runAgentNonStreaming(req, res, registered, thread, userId) {
+	async _runAgentNonStreaming(req, res, registered, thread, userId, mlflowRunId) {
 		const abortController = new AbortController();
 		const signal = abortController.signal;
 		const requestId = randomUUID();
 		this.trackStream(requestId, userId, abortController);
 		const tools = Array.from(registered.toolIndex.values()).map((e) => e.def);
+		const limits = this.resolvedLimits;
+		let mlflowTraceId;
 		const runState = {
 			req,
 			userId,
@@ -800,7 +821,7 @@ var AgentsPlugin = class extends Plugin {
 				requireForDestructive: false,
 				timeoutMs: 0
 			},
-			limits: this.resolvedLimits,
+			limits,
 			translator: new AgentEventTranslator(),
 			outboundEvents: new EventChannel(),
 			toolCallsUsed: { count: 0 }
@@ -808,31 +829,44 @@ var AgentsPlugin = class extends Plugin {
 		const executeTool = (name, args) => this.dispatchToolCall(runState, registered.toolIndex, name, args, 0);
 		let fullContent = "";
 		try {
-			const pluginNames = this.context ? this.context.getPluginNames().filter((n) => n !== this.name && n !== "server") : [];
-			const messagesWithSystem = [{
-				id: "system",
-				role: "system",
-				content: composePromptForAgent(registered, this.config.baseSystemPrompt, {
-					agentName: registered.name,
-					pluginNames,
-					toolNames: tools.map((t) => t.name)
-				}),
-				createdAt: /* @__PURE__ */ new Date()
-			}, ...thread.messages];
-			fullContent = await consumeAdapterStream(registered.adapter.run({
-				messages: messagesWithSystem,
-				tools,
-				threadId: thread.id,
-				signal
-			}, {
-				executeTool,
-				signal
-			}), { signal });
-			if (fullContent) await this.threadStore.addMessage(thread.id, userId, {
-				id: randomUUID(),
-				role: "assistant",
-				content: fullContent,
-				createdAt: /* @__PURE__ */ new Date()
+			await traceAgent(registered.name ?? "agent", { messages: thread.messages.map((m) => ({
+				role: m.role,
+				content: m.content
+			})) }, async (span) => {
+				if (mlflowRunId) linkTraceToRun(mlflowRunId);
+				const pluginNames = this.context ? this.context.getPluginNames().filter((n) => n !== this.name && n !== "server") : [];
+				const messagesWithSystem = [{
+					id: "system",
+					role: "system",
+					content: composePromptForAgent(registered, this.config.baseSystemPrompt, {
+						agentName: registered.name,
+						pluginNames,
+						toolNames: tools.map((t) => t.name)
+					}),
+					createdAt: /* @__PURE__ */ new Date()
+				}, ...thread.messages];
+				fullContent = await consumeAdapterStream(registered.adapter.run({
+					messages: messagesWithSystem,
+					tools,
+					threadId: thread.id,
+					signal
+				}, {
+					executeTool,
+					signal
+				}), { signal });
+				if (fullContent) {
+					span.setOutputs({
+						role: "assistant",
+						content: fullContent
+					});
+					await this.threadStore.addMessage(thread.id, userId, {
+						id: randomUUID(),
+						role: "assistant",
+						content: fullContent,
+						createdAt: /* @__PURE__ */ new Date()
+					});
+				}
+				mlflowTraceId = currentTraceId();
 			});
 		} catch (error) {
 			if (signal.aborted) {
@@ -869,6 +903,7 @@ var AgentsPlugin = class extends Plugin {
 			created_at: Math.floor(Date.now() / 1e3),
 			status: "completed",
 			thread_id: thread.id,
+			...mlflowTraceId ? { mlflow_trace_id: mlflowTraceId } : {},
 			output: [message]
 		});
 	}
@@ -908,24 +943,26 @@ var AgentsPlugin = class extends Plugin {
 				timeoutMs: runState.approvalPolicy.timeoutMs
 			}) === "deny") return `Tool execution denied by user approval gate (tool: ${name}).`;
 		}
-		let result;
-		if (entry.source === "toolkit") {
-			if (!this.context) throw new Error("Plugin tool execution requires PluginContext; this should never happen through createApp");
-			result = await this.context.executeTool(runState.req, entry.pluginName, entry.localName, args, runState.signal, runState.limits.toolCallTimeoutMs);
-		} else if (entry.source === "function") {
-			if (typeof args !== "object" || args === null || Array.isArray(args)) throw new Error(`Function tool '${name}' received non-object arguments (got ${args === null ? "null" : Array.isArray(args) ? "array" : typeof args}); expected a JSON object.`);
-			result = await entry.functionTool.execute(args);
-		} else if (entry.source === "mcp") {
-			if (!this.mcpClient) throw new Error("MCP client not connected");
-			const oboToken = runState.req.headers["x-forwarded-access-token"];
-			const mcpAuth = typeof oboToken === "string" ? { Authorization: `Bearer ${oboToken}` } : void 0;
-			result = await this.mcpClient.callTool(entry.mcpToolName, args, mcpAuth);
-		} else if (entry.source === "subagent") {
-			const childAgent = this.agents.get(entry.agentName);
-			if (!childAgent) throw new Error(`Sub-agent not found: ${entry.agentName}`);
-			result = await this.runSubAgent(runState, childAgent, args, depth + 1);
-		} else if (entry.source === "hosted-supervisor") throw new Error(`Tool '${name}' is a hosted-supervisor tool and cannot be invoked from the Node process. It is executed server-side by the Databricks AI Gateway and is only reachable when the agent's model is a Supervisor API adapter.`);
-		return normalizeToolResult(result);
+		return normalizeToolResult(await traceTool(name, args, async () => {
+			let result;
+			if (entry.source === "toolkit") {
+				if (!this.context) throw new Error("Plugin tool execution requires PluginContext; this should never happen through createApp");
+				result = await this.context.executeTool(runState.req, entry.pluginName, entry.localName, args, runState.signal, runState.limits.toolCallTimeoutMs);
+			} else if (entry.source === "function") {
+				if (typeof args !== "object" || args === null || Array.isArray(args)) throw new Error(`Function tool '${name}' received non-object arguments (got ${args === null ? "null" : Array.isArray(args) ? "array" : typeof args}); expected a JSON object.`);
+				result = await entry.functionTool.execute(args);
+			} else if (entry.source === "mcp") {
+				if (!this.mcpClient) throw new Error("MCP client not connected");
+				const oboToken = runState.req.headers["x-forwarded-access-token"];
+				const mcpAuth = typeof oboToken === "string" ? { Authorization: `Bearer ${oboToken}` } : void 0;
+				result = await this.mcpClient.callTool(entry.mcpToolName, args, mcpAuth);
+			} else if (entry.source === "subagent") {
+				const childAgent = this.agents.get(entry.agentName);
+				if (!childAgent) throw new Error(`Sub-agent not found: ${entry.agentName}`);
+				result = await this.runSubAgent(runState, childAgent, args, depth + 1);
+			} else if (entry.source === "hosted-supervisor") throw new Error(`Tool '${name}' is a hosted-supervisor tool and cannot be invoked from the Node process. It is executed server-side by the Databricks AI Gateway and is only reachable when the agent's model is a Supervisor API adapter.`);
+			return result;
+		}));
 	}
 	/**
 	* Runs a sub-agent in response to an `agent-<key>` tool call. Returns the