@databricks/ai-sdk-provider 0.2.2 → 0.3.0

package/dist/{index.js → index.mjs}

@@ -46,26 +46,6 @@ const chatAgentResponseSchema = z.object({
 	messages: z.array(chatAgentMessageSchema)
 });
 
-//#endregion
-//#region src/tools.ts
-const DATABRICKS_TOOL_CALL_ID = "databricks-tool-call";
-/**
-* The AI-SDK requires that tools used by the model are defined ahead of time.
-*
-* Since tool calls can be orchestrated by Databricks' agents we don't know the name, input, or output schemas
-* of the tools until the model is called.
-*
-* In the DatabricksProvider we transform all tool calls to fit this definition, and keep the
-* original name as part of the metadata. This allows us to parse any tool orchestrated by Databricks' agents,
-* while still being able to render the tool call and result in the UI, and pass it back to the model with the correct name.
-*/
-const DATABRICKS_TOOL_DEFINITION = {
-	name: DATABRICKS_TOOL_CALL_ID,
-	description: "Databricks tool call",
-	inputSchema: z.any(),
-	outputSchema: z.any()
-};
-
 //#endregion
 //#region src/chat-agent-language-model/chat-agent-convert-to-message-parts.ts
 const convertChatAgentChunkToMessagePart = (chunk) => {
@@ -81,14 +61,16 @@ const convertChatAgentChunkToMessagePart = (chunk) => {
 				type: "tool-call",
 				toolCallId: toolCall.id,
 				input: toolCall.function.arguments,
-				toolName: toolCall.function.name
+				toolName: toolCall.function.name,
+				dynamic: true,
+				providerExecuted: true
 			});
 		});
 	} else if (chunk.delta.role === "tool") parts.push({
 		type: "tool-result",
 		toolCallId: chunk.delta.tool_call_id,
 		result: chunk.delta.content,
-		toolName: DATABRICKS_TOOL_CALL_ID
+		toolName: chunk.delta.name ?? "unknown"
 	});
 	return parts;
 };
@@ -103,20 +85,22 @@ const convertChatAgentResponseToMessagePart = (response) => {
 			type: "tool-call",
 			toolCallId: part.id,
 			input: part.function.arguments,
-			toolName: part.function.name
+			toolName: part.function.name,
+			dynamic: true,
+			providerExecuted: true
 		});
 	} else if (message.role === "tool") parts.push({
 		type: "tool-result",
 		toolCallId: message.tool_call_id,
 		result: message.content,
-		toolName: DATABRICKS_TOOL_CALL_ID
+		toolName: message.name ?? "unknown"
 	});
 	return parts;
 };
 
 //#endregion
 //#region src/chat-agent-language-model/chat-agent-convert-to-input.ts
-const convertLanguageModelV2PromptToChatAgentResponse = (prompt) => {
+const convertLanguageModelV3PromptToChatAgentResponse = (prompt) => {
 	const messages = [];
 	let messageIndex = 0;
 	for (const msg of prompt) switch (msg.role) {
@@ -143,10 +127,9 @@ const convertLanguageModelV2PromptToChatAgentResponse = (prompt) => {
 	return messages;
 };
 const convertUserMessage$1 = (msg, messageIndex) => {
-	const text = (msg.content ?? []).filter((part) => part.type === "text").map((part) => part.text).join("\n");
 	return {
 		role: "user",
-		content: text,
+		content: (msg.content ?? []).filter((part) => part.type === "text").map((part) => part.text).join("\n"),
 		id: `user-${messageIndex}`
 	};
 };
@@ -219,10 +202,7 @@ const convertToolResultOutput = (output) => {
 function composeDatabricksStreamPartTransformers(...transformers) {
 	return (initialParts, last = null) => {
 		let currentParts = initialParts;
-		for (const fn of transformers) {
-			const result = fn(currentParts, last);
-			currentParts = result.out;
-		}
+		for (const fn of transformers) currentParts = fn(currentParts, last).out;
 		return { out: currentParts };
 	};
 }
@@ -239,8 +219,7 @@ const applyDeltaBoundaryTransform = (parts, last) => {
 		const incomingDeltaType = maybeGetDeltaType(incoming);
 		const incomingId = getPartId$1(incoming);
 		const lastId = getPartId$1(last);
-		const incomingMatchesLast = Boolean(isDeltaPart(last) && isDeltaPart(incoming)) && Boolean(lastDeltaType && incomingDeltaType) && Boolean(lastDeltaType === incomingDeltaType) && Boolean(incomingId && lastId && incomingId === lastId);
-		if (incomingMatchesLast) {
+		if (Boolean(isDeltaPart(last) && isDeltaPart(incoming)) && Boolean(lastDeltaType && incomingDeltaType) && Boolean(lastDeltaType === incomingDeltaType) && Boolean(incomingId && lastId && incomingId === lastId)) {
 			out.push(incoming);
 			continue;
 		}
@@ -256,7 +235,6 @@ const applyDeltaBoundaryTransform = (parts, last) => {
 			continue;
 		}
 		out.push(incoming);
-		continue;
 	}
 	return { out };
 };
@@ -275,7 +253,6 @@ const getDeltaType = (part) => {
 const isDeltaPart = (part) => part?.type === "text-delta" || part?.type === "reasoning-delta";
 const getPartId$1 = (part) => {
 	if (part && "id" in part) return part.id;
-	return void 0;
 };
 
 //#endregion
@@ -288,7 +265,7 @@ const getPartId$1 = (part) => {
 */
 const getDatabricksLanguageModelTransformStream = () => {
 	let lastChunk = null;
-	const deltaEndByTypeAndId = new Set();
+	const deltaEndByTypeAndId = /* @__PURE__ */ new Set();
 	const transformerStreamParts = composeDatabricksStreamPartTransformers(applyDeltaBoundaryTransform);
 	return new TransformStream({
 		transform(chunk, controller) {
@@ -330,14 +307,13 @@ const getDeltaGroup = (type) => {
 };
 const getPartId = (part) => {
 	if ("id" in part) return part.id;
-	return void 0;
 };
 const makeEndKey = (id, group) => id && group ? `${group}:${id}` : null;
 
 //#endregion
 //#region src/chat-agent-language-model/chat-agent-language-model.ts
 var DatabricksChatAgentLanguageModel = class {
-	specificationVersion = "v2";
+	specificationVersion = "v3";
 	modelId;
 	config;
 	constructor(modelId, config) {
@@ -349,14 +325,13 @@ var DatabricksChatAgentLanguageModel = class {
 	}
 	supportedUrls = {};
 	async doGenerate(options) {
-		const networkArgs = this.getArgs({
-			config: this.config,
-			options,
-			stream: false,
-			modelId: this.modelId
-		});
 		const { value: response } = await postJsonToApi({
-			...networkArgs,
+			...this.getArgs({
+				config: this.config,
+				options,
+				stream: false,
+				modelId: this.modelId
+			}),
 			successfulResponseHandler: createJsonResponseHandler(chatAgentResponseSchema),
 			failedResponseHandler: createJsonErrorResponseHandler({
 				errorSchema: z.any(),
@@ -366,11 +341,22 @@ var DatabricksChatAgentLanguageModel = class {
 		});
 		return {
 			content: convertChatAgentResponseToMessagePart(response),
-			finishReason: "stop",
+			finishReason: {
+				raw: void 0,
+				unified: "stop"
+			},
 			usage: {
-				inputTokens: 0,
-				outputTokens: 0,
-				totalTokens: 0
+				inputTokens: {
+					total: 0,
+					noCache: 0,
+					cacheRead: 0,
+					cacheWrite: 0
+				},
+				outputTokens: {
+					total: 0,
+					text: 0,
+					reasoning: 0
+				}
 			},
 			warnings: []
 		};
@@ -391,7 +377,10 @@ var DatabricksChatAgentLanguageModel = class {
 			}),
 			successfulResponseHandler: createEventSourceResponseHandler(chatAgentChunkSchema)
 		});
-		let finishReason = "unknown";
+		let finishReason = {
+			raw: void 0,
+			unified: "other"
+		};
 		return {
 			stream: response.pipeThrough(new TransformStream({
 				start(controller) {
@@ -406,7 +395,10 @@ var DatabricksChatAgentLanguageModel = class {
 						rawValue: chunk.rawValue
 					});
 					if (!chunk.success) {
-						finishReason = "error";
+						finishReason = {
+							raw: void 0,
+							unified: "error"
+						};
 						controller.enqueue({
 							type: "error",
 							error: chunk.error
@@ -421,9 +413,17 @@ var DatabricksChatAgentLanguageModel = class {
 						type: "finish",
 						finishReason,
 						usage: {
-							inputTokens: 0,
-							outputTokens: 0,
-							totalTokens: 0
+							inputTokens: {
+								total: 0,
+								noCache: 0,
+								cacheRead: 0,
+								cacheWrite: 0
+							},
+							outputTokens: {
+								total: 0,
+								text: 0,
+								reasoning: 0
+							}
 						}
 					});
 				}
@@ -437,7 +437,7 @@ var DatabricksChatAgentLanguageModel = class {
 			body: {
 				model: modelId,
 				stream,
-				messages: convertLanguageModelV2PromptToChatAgentResponse(options.prompt)
+				messages: convertLanguageModelV3PromptToChatAgentResponse(options.prompt)
 			},
 			url: config.url({ path: "/completions" }),
 			headers: combineHeaders(config.headers(), options.headers),
@@ -668,7 +668,6 @@ function isMcpApprovalResponse(metadata) {
 */
 function extractDatabricksMetadata(part) {
 	if ("callProviderMetadata" in part && part.callProviderMetadata?.databricks) return part.callProviderMetadata.databricks;
-	return void 0;
 }
 /**
 * Extract the approval status boolean from an output object.
@@ -683,7 +682,6 @@ function extractDatabricksMetadata(part) {
 */
 function extractApprovalStatus(output) {
 	if (isApprovalStatusOutput(output)) return output[MCP_APPROVAL_STATUS_KEY];
-	return void 0;
 }
 /**
 * Extract approval status from a tool result's output value.
@@ -697,7 +695,6 @@ function extractApprovalStatusFromToolResult(output) {
 		const value = output.value[MCP_APPROVAL_STATUS_KEY];
 		if (typeof value === "boolean") return value;
 	}
-	return void 0;
 }
 /**
 * Create an approval status output object.
@@ -733,7 +730,7 @@ function getMcpApprovalState(output) {
 
 //#endregion
 //#region src/responses-agent-language-model/responses-convert-to-message-parts.ts
-const convertResponsesAgentChunkToMessagePart = (chunk) => {
+const convertResponsesAgentChunkToMessagePart = (chunk, options = { useRemoteToolCalling: false }) => {
 	const parts = [];
 	if ("error" in chunk) {
 		parts.push({
@@ -763,12 +760,12 @@ const convertResponsesAgentChunkToMessagePart = (chunk) => {
 			parts.push({
 				type: "tool-result",
 				toolCallId: chunk.call_id,
-				result: chunk.output,
-				toolName: DATABRICKS_TOOL_CALL_ID
+				result: chunk.output != null ? chunk.output : {},
+				toolName: options.toolNamesByCallId?.get(chunk.call_id) ?? "unknown"
 			});
 			break;
 		case "response.output_item.done":
-			parts.push(...convertOutputItemDone(chunk.item));
+			parts.push(...convertOutputItemDone(chunk.item, options));
 			break;
 		case "response.output_text.annotation.added":
 			parts.push({
@@ -789,7 +786,7 @@ const convertResponsesAgentChunkToMessagePart = (chunk) => {
 	}
 	return parts;
 };
-const convertOutputItemDone = (item) => {
+const convertOutputItemDone = (item, options) => {
 	switch (item.type) {
 		case "message": {
 			const firstContent = item.content[0];
@@ -807,18 +804,19 @@ const convertOutputItemDone = (item) => {
 		case "function_call": return [{
 			type: "tool-call",
 			toolCallId: item.call_id,
-			toolName: DATABRICKS_TOOL_CALL_ID,
+			toolName: item.name,
 			input: item.arguments,
-			providerMetadata: { databricks: {
-				toolName: item.name,
-				itemId: item.id
-			} }
+			...options.useRemoteToolCalling && {
+				dynamic: true,
+				providerExecuted: true
+			},
+			providerMetadata: { databricks: { itemId: item.id } }
 		}];
 		case "function_call_output": return [{
 			type: "tool-result",
 			toolCallId: item.call_id,
-			result: item.output,
-			toolName: DATABRICKS_TOOL_CALL_ID
+			result: item.output != null ? item.output : {},
+			toolName: options.toolNamesByCallId?.get(item.call_id) ?? "unknown"
 		}];
 		case "reasoning": {
 			const firstSummary = item.summary[0];
@@ -843,11 +841,14 @@ const convertOutputItemDone = (item) => {
 		case "mcp_approval_request": return [{
 			type: "tool-call",
 			toolCallId: item.id,
-			toolName: DATABRICKS_TOOL_CALL_ID,
+			toolName: item.name,
 			input: item.arguments,
+			...options.useRemoteToolCalling && {
+				dynamic: true,
+				providerExecuted: true
+			},
 			providerMetadata: { databricks: {
 				type: MCP_APPROVAL_REQUEST_TYPE,
-				toolName: item.name,
 				itemId: item.id,
 				serverLabel: item.server_label
 			} }
@@ -855,7 +856,7 @@ const convertOutputItemDone = (item) => {
 		case "mcp_approval_response": return [{
 			type: "tool-result",
 			toolCallId: item.approval_request_id,
-			toolName: DATABRICKS_TOOL_CALL_ID,
+			toolName: options.toolNamesByCallId?.get(item.approval_request_id) ?? "mcp_approval",
 			result: createApprovalStatusOutput(item.approve),
 			providerMetadata: { databricks: {
 				type: MCP_APPROVAL_RESPONSE_TYPE,
@@ -865,27 +866,30 @@ const convertOutputItemDone = (item) => {
 		default: return [];
 	}
 };
-const convertResponsesAgentResponseToMessagePart = (response) => {
+const convertResponsesAgentResponseToMessagePart = (response, options = { useRemoteToolCalling: false }) => {
 	const parts = [];
+	const toolNamesByCallId = /* @__PURE__ */ new Map();
+	for (const output of response.output) if (output.type === "function_call") toolNamesByCallId.set(output.call_id, output.name);
+	else if (output.type === "mcp_approval_request") toolNamesByCallId.set(output.id, output.name);
 	for (const output of response.output) switch (output.type) {
-		case "message": {
+		case "message":
 			for (const content of output.content) if (content.type === "output_text") parts.push({
 				type: "text",
 				text: content.text,
 				providerMetadata: { databricks: { itemId: output.id } }
 			});
 			break;
-		}
 		case "function_call":
 			parts.push({
 				type: "tool-call",
 				toolCallId: output.call_id,
-				toolName: DATABRICKS_TOOL_CALL_ID,
+				toolName: output.name,
 				input: output.arguments,
-				providerMetadata: { databricks: {
-					toolName: output.name,
-					itemId: output.id
-				} }
+				...options.useRemoteToolCalling && {
+					dynamic: true,
+					providerExecuted: true
+				},
+				providerMetadata: { databricks: { itemId: output.id } }
 			});
 			break;
 		case "reasoning":
@@ -900,18 +904,17 @@ const convertResponsesAgentResponseToMessagePart = (response) => {
 				type: "tool-result",
 				result: output.output,
 				toolCallId: output.call_id,
-				toolName: DATABRICKS_TOOL_CALL_ID
+				toolName: toolNamesByCallId.get(output.call_id) ?? "unknown"
 			});
 			break;
 		case "mcp_approval_request":
 			parts.push({
 				type: "tool-call",
 				toolCallId: output.id,
-				toolName: DATABRICKS_TOOL_CALL_ID,
+				toolName: output.name,
 				input: output.arguments,
 				providerMetadata: { databricks: {
 					type: MCP_APPROVAL_REQUEST_TYPE,
-					toolName: output.name,
 					itemId: output.id,
 					serverLabel: output.server_label
 				} }
@@ -921,7 +924,7 @@ const convertResponsesAgentResponseToMessagePart = (response) => {
 			parts.push({
 				type: "tool-result",
 				toolCallId: output.approval_request_id,
-				toolName: DATABRICKS_TOOL_CALL_ID,
+				toolName: toolNamesByCallId.get(output.approval_request_id) ?? "mcp_approval",
 				result: createApprovalStatusOutput(output.approve),
 				providerMetadata: { databricks: {
 					type: MCP_APPROVAL_RESPONSE_TYPE,
@@ -944,7 +947,7 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 		return reduction;
 	}, {});
 	for (const { role, content } of prompt) switch (role) {
-		case "system": {
+		case "system":
 			switch (systemMessageMode) {
 				case "system":
 					input.push({
@@ -970,7 +973,6 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 				}
 			}
 			break;
-		}
 		case "user":
 			input.push({
 				role: "user",
@@ -994,7 +996,7 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 				});
 				const itemId = providerOptions?.itemId ?? void 0;
 				switch (part.type) {
-					case "text": {
+					case "text":
 						input.push({
 							role: "assistant",
 							content: [{
@@ -1004,7 +1006,6 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 							id: itemId
 						});
 						break;
-					}
 					case "tool-call": {
 						const toolName = providerOptions?.toolName ?? part.toolName;
 						if (providerOptions?.type === MCP_APPROVAL_REQUEST_TYPE) {
@@ -1055,7 +1056,7 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 						});
 						break;
 					}
-					case "tool-result": {
+					case "tool-result":
 						if (providerOptions?.type === MCP_APPROVAL_RESPONSE_TYPE) {
 							const approvalRequestId = providerOptions?.approvalRequestId ?? part.toolCallId;
 							const approve = providerOptions?.approve ?? false;
@@ -1075,8 +1076,7 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 							output: convertToolResultOutputToString(part.output)
 						});
 						break;
-					}
-					case "reasoning": {
+					case "reasoning":
 						if (!itemId) break;
 						input.push({
 							type: "reasoning",
@@ -1087,7 +1087,6 @@ async function convertToResponsesInput({ prompt, systemMessageMode }) {
 							id: itemId
 						});
 						break;
-					}
 				}
 			}
 			break;
@@ -1115,7 +1114,10 @@ const convertToolResultOutputToString = (output) => {
 	switch (output.type) {
 		case "text":
 		case "error-text": return output.value;
-		default: return JSON.stringify(output.value);
+		case "execution-denied": return output.reason ?? "Execution denied";
+		case "json":
+		case "error-json":
+		case "content": return JSON.stringify(output.value);
 	}
 };
 
@@ -1133,7 +1135,7 @@ function prepareResponsesTools({ tools, toolChoice }) {
 	};
 	const responsesTools = [];
 	for (const tool of tools) {
-		if (tool.type === "provider-defined" || tool.name === DATABRICKS_TOOL_CALL_ID) continue;
+		if (tool.type === "provider") continue;
 		responsesTools.push({
 			type: "function",
 			name: tool.name,
@@ -1145,14 +1147,13 @@ function prepareResponsesTools({ tools, toolChoice }) {
 		tools: void 0,
 		toolChoice: void 0
 	};
-	const convertedToolChoice = convertResponsesToolChoice(toolChoice);
 	return {
 		tools: responsesTools,
-		toolChoice: convertedToolChoice
+		toolChoice: convertResponsesToolChoice(toolChoice)
 	};
 }
 function convertResponsesToolChoice(toolChoice) {
-	if (!toolChoice) return void 0;
+	if (!toolChoice) return;
 	switch (toolChoice.type) {
 		case "auto": return "auto";
 		case "none": return "none";
@@ -1161,23 +1162,103 @@ function convertResponsesToolChoice(toolChoice) {
 			type: "function",
 			name: toolChoice.toolName
 		};
-		default: return void 0;
+		default: return;
 	}
 }
 
+//#endregion
+//#region src/responses-agent-language-model/call-options-to-responses-args.ts
+/**
+* Converts AI SDK LanguageModelV3CallOptions to Databricks Responses API body parameters.
+*
+* Inspired by the getArgs method in:
+* https://github.com/vercel/ai/blob/main/packages/openai/src/responses/openai-responses-language-model.ts#L118
+*
+* Complies with the API described in:
+* https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/api-reference#responses-api-request
+*/
+function callOptionsToResponsesArgs(options) {
+	const warnings = [];
+	const databricksOptions = options.providerOptions?.databricks;
+	if (options.topK != null) warnings.push({
+		type: "unsupported",
+		feature: "topK",
+		details: "topK is not supported by the Databricks Responses API"
+	});
+	if (options.presencePenalty != null) warnings.push({
+		type: "unsupported",
+		feature: "presencePenalty",
+		details: "presencePenalty is not supported by the Databricks Responses API"
+	});
+	if (options.frequencyPenalty != null) warnings.push({
+		type: "unsupported",
+		feature: "frequencyPenalty",
+		details: "frequencyPenalty is not supported by the Databricks Responses API"
+	});
+	if (options.seed != null) warnings.push({
+		type: "unsupported",
+		feature: "seed",
+		details: "seed is not supported by the Databricks Responses API"
+	});
+	if (options.stopSequences != null && options.stopSequences.length > 0) warnings.push({
+		type: "unsupported",
+		feature: "stopSequences",
+		details: "stopSequences is not supported by the Databricks Responses API"
+	});
+	const args = {};
+	if (options.maxOutputTokens != null) args.max_output_tokens = options.maxOutputTokens;
+	if (options.temperature != null) args.temperature = options.temperature;
+	if (options.topP != null) args.top_p = options.topP;
+	if (options.responseFormat != null) switch (options.responseFormat.type) {
+		case "text":
+			args.text = { format: { type: "text" } };
+			break;
+		case "json":
+			if (options.responseFormat.schema != null) args.text = { format: {
+				type: "json_schema",
+				json_schema: {
+					name: options.responseFormat.name ?? "response",
+					description: options.responseFormat.description,
+					schema: options.responseFormat.schema,
+					strict: true
+				}
+			} };
+			else args.text = { format: { type: "json_object" } };
+			break;
+	}
+	if (databricksOptions?.parallelToolCalls != null) args.parallel_tool_calls = databricksOptions.parallelToolCalls;
+	if (databricksOptions?.metadata != null) args.metadata = databricksOptions.metadata;
+	if (databricksOptions?.reasoning != null) args.reasoning = databricksOptions.reasoning;
+	return {
+		args,
+		warnings
+	};
+}
+
 //#endregion
 //#region src/responses-agent-language-model/responses-agent-language-model.ts
 function mapResponsesFinishReason({ finishReason, hasToolCalls }) {
+	let unified;
 	switch (finishReason) {
 		case void 0:
-		case null: return hasToolCalls ? "tool-calls" : "stop";
-		case "max_output_tokens": return "length";
-		case "content_filter": return "content-filter";
-		default: return hasToolCalls ? "tool-calls" : "other";
+		case null:
+			unified = hasToolCalls ? "tool-calls" : "stop";
+			break;
+		case "max_output_tokens":
+			unified = "length";
+			break;
+		case "content_filter":
+			unified = "content-filter";
+			break;
+		default: unified = hasToolCalls ? "tool-calls" : "other";
 	}
+	return {
+		raw: finishReason ?? void 0,
+		unified
+	};
 }
 var DatabricksResponsesAgentLanguageModel = class {
-	specificationVersion = "v2";
+	specificationVersion = "v3";
 	modelId;
 	config;
 	constructor(modelId, config) {
@@ -1189,7 +1270,7 @@ var DatabricksResponsesAgentLanguageModel = class {
 	}
 	supportedUrls = {};
 	async doGenerate(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings, ...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: false,
@@ -1213,15 +1294,23 @@ var DatabricksResponsesAgentLanguageModel = class {
 				hasToolCalls
 			}),
 			usage: {
-				inputTokens: response.usage?.input_tokens ?? 0,
-				outputTokens: response.usage?.output_tokens ?? 0,
-				totalTokens: response.usage?.total_tokens ?? 0
+				inputTokens: {
+					total: response.usage?.input_tokens ?? 0,
+					noCache: 0,
+					cacheRead: 0,
+					cacheWrite: 0
+				},
+				outputTokens: {
+					total: response.usage?.output_tokens ?? 0,
+					text: 0,
+					reasoning: 0
+				}
 			},
-			warnings: []
+			warnings
 		};
 	}
 	async doStream(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings, ...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: true,
@@ -1237,19 +1326,32 @@ var DatabricksResponsesAgentLanguageModel = class {
 			successfulResponseHandler: createEventSourceResponseHandler(looseResponseAgentChunkSchema),
 			abortSignal: options.abortSignal
 		});
-		let finishReason = "unknown";
+		let finishReason = {
+			raw: void 0,
+			unified: "stop"
+		};
 		const usage = {
-			inputTokens: 0,
-			outputTokens: 0,
-			totalTokens: 0
+			inputTokens: {
+				total: 0,
+				noCache: 0,
+				cacheRead: 0,
+				cacheWrite: 0
+			},
+			outputTokens: {
+				total: 0,
+				text: 0,
+				reasoning: 0
+			}
 		};
 		const allParts = [];
+		const useRemoteToolCalling = this.config.useRemoteToolCalling ?? false;
+		const toolNamesByCallId = /* @__PURE__ */ new Map();
 		return {
 			stream: response.pipeThrough(new TransformStream({
 				start(controller) {
 					controller.enqueue({
 						type: "stream-start",
-						warnings: []
+						warnings
 					});
 				},
 				transform(chunk, controller) {
@@ -1258,7 +1360,10 @@ var DatabricksResponsesAgentLanguageModel = class {
 						rawValue: chunk.rawValue
 					});
 					if (!chunk.success) {
-						finishReason = "error";
+						finishReason = {
+							raw: void 0,
+							unified: "error"
+						};
 						controller.enqueue({
 							type: "error",
 							error: chunk.error
@@ -1271,12 +1376,15 @@ var DatabricksResponsesAgentLanguageModel = class {
 							finishReason: chunk.value.response.incomplete_details?.reason,
 							hasToolCalls
 						});
-						usage.inputTokens = chunk.value.response.usage.input_tokens;
-						usage.outputTokens = chunk.value.response.usage.output_tokens;
-						usage.totalTokens = chunk.value.response.usage.total_tokens;
+						usage.inputTokens.total = chunk.value.response.usage.input_tokens;
+						usage.outputTokens.total = chunk.value.response.usage.output_tokens;
 						return;
 					}
-					const parts = convertResponsesAgentChunkToMessagePart(chunk.value);
+					if (chunk.value.type === "response.output_item.done" && chunk.value.item.type === "function_call") toolNamesByCallId.set(chunk.value.item.call_id, chunk.value.item.name);
+					const parts = convertResponsesAgentChunkToMessagePart(chunk.value, {
+						useRemoteToolCalling,
+						toolNamesByCallId
+					});
 					allParts.push(...parts);
 					/**
 					* Check if the last chunk was a tool result without a tool call
@@ -1291,19 +1399,33 @@ var DatabricksResponsesAgentLanguageModel = class {
 						if (!matchingToolCallInParts && !matchingToolCallInStream) {
 							const toolCallFromPreviousMessages = options.prompt.flatMap((message) => {
 								if (typeof message.content === "string") return [];
-								return message.content;
-							}).find((p) => p.type === "tool-call" && p.toolCallId === part.toolCallId);
+								return message.content.filter((p) => p.type === "tool-call");
+							}).find((p) => p.toolCallId === part.toolCallId);
 							if (!toolCallFromPreviousMessages) throw new Error("No matching tool call found in previous message");
-							if (toolCallFromPreviousMessages.type === "tool-call") controller.enqueue({
-								...toolCallFromPreviousMessages,
+							controller.enqueue({
+								type: "tool-call",
+								toolCallId: toolCallFromPreviousMessages.toolCallId,
+								toolName: toolCallFromPreviousMessages.toolName,
 								input: JSON.stringify(toolCallFromPreviousMessages.input)
 							});
 						}
 					}
 					if (shouldDedupeOutputItemDone(parts, allParts.slice(0, -parts.length))) return;
-					for (const part$1 of parts) controller.enqueue(part$1);
+					for (const part of parts) controller.enqueue(part);
 				},
 				flush(controller) {
+					if (!useRemoteToolCalling) {
+						const toolCalls = allParts.filter((p) => p.type === "tool-call");
+						const toolResults = allParts.filter((p) => p.type === "tool-result");
+						for (const toolCall of toolCalls) {
+							if (toolCall.providerMetadata?.databricks?.type === MCP_APPROVAL_REQUEST_TYPE) continue;
+							if (!toolResults.some((r) => r.toolCallId === toolCall.toolCallId)) controller.enqueue({
+								...toolCall,
+								providerExecuted: true,
+								dynamic: true
+							});
+						}
+					}
 					controller.enqueue({
 						type: "finish",
 						finishReason,
@@ -1324,6 +1446,7 @@ var DatabricksResponsesAgentLanguageModel = class {
 			tools: options.tools,
 			toolChoice: options.toolChoice
 		});
+		const { args: callArgs, warnings } = callOptionsToResponsesArgs(options);
 		return {
 			url: config.url({ path: "/responses" }),
 			headers: combineHeaders(config.headers(), options.headers),
@@ -1332,8 +1455,10 @@ var DatabricksResponsesAgentLanguageModel = class {
 				input,
 				stream,
 				...tools ? { tools } : {},
-				...toolChoice && tools ? { tool_choice: toolChoice } : {}
+				...toolChoice && tools ? { tool_choice: toolChoice } : {},
+				...callArgs
 			},
+			warnings,
 			fetch: config.fetch
 		};
 	}
@@ -1353,8 +1478,7 @@ function shouldDedupeOutputItemDone(incomingParts, previousParts) {
 	* corresponds to a specific message and we should only compare against text streamed for that message.
 	*/
 	const lastDoneIndex = previousParts.findLastIndex((part) => part.type === "text-delta" && part.providerMetadata?.databricks?.itemType === "response.output_item.done");
-	const partsAfterLastDone = previousParts.slice(lastDoneIndex + 1);
-	const { texts: reconstructuredTexts, current } = partsAfterLastDone.reduce((acc, part) => {
+	const { texts: reconstructuredTexts, current } = previousParts.slice(lastDoneIndex + 1).reduce((acc, part) => {
 		if (part.type === "text-delta") return {
 			...acc,
 			current: acc.current + part.delta
@@ -1368,9 +1492,9 @@ function shouldDedupeOutputItemDone(incomingParts, previousParts) {
 		texts: [],
 		current: ""
 	});
-	reconstructuredTexts.push(current);
+	if (current.length > 0) reconstructuredTexts.push(current);
 	if (reconstructuredTexts.length === 0) return false;
-	const allTextsFoundInOrder = reconstructuredTexts.reduce((acc, text) => {
+	return reconstructuredTexts.reduce((acc, text) => {
 		if (!acc.found) return acc;
 		const index = doneTextDelta.delta.indexOf(text, acc.lastIndex);
 		if (index === -1) return {
@@ -1384,8 +1508,7 @@ function shouldDedupeOutputItemDone(incomingParts, previousParts) {
 	}, {
 		found: true,
 		lastIndex: 0
-	});
-	return allTextsFoundInOrder.found;
+	}).found;
 }
 
 //#endregion
@@ -1519,12 +1642,12 @@ const convertFmapiChunkToMessagePart = (chunk, toolCallIdsByIndex) => {
 	} else if (Array.isArray(choice.delta.content)) parts.push(...mapContentItemsToStreamParts(choice.delta.content, chunk.id));
 	return parts;
 };
-const convertFmapiResponseToMessagePart = (response) => {
+const convertFmapiResponseToMessagePart = (response, options = { useRemoteToolCalling: false }) => {
 	const parts = [];
 	if (response.choices.length === 0) return parts;
 	const choice = response.choices[0];
 	if (choice.message.tool_calls && choice.message.tool_calls.length > 0) {
-		for (const toolCall of choice.message.tool_calls) parts.push(convertToolCallToContent(toolCall));
+		for (const toolCall of choice.message.tool_calls) parts.push(convertToolCallToContent(toolCall, options));
 		if (typeof choice.message.content === "string" && choice.message.content) parts.push({
 			type: "text",
 			text: choice.message.content
@@ -1538,13 +1661,16 @@ const convertFmapiResponseToMessagePart = (response) => {
 	else parts.push(...mapContentItemsToProviderContent(choice.message.content ?? []));
 	return parts;
 };
-const convertToolCallToContent = (toolCall) => {
+const convertToolCallToContent = (toolCall, options) => {
 	return {
 		type: "tool-call",
 		toolCallId: toolCall.id,
-		toolName: DATABRICKS_TOOL_CALL_ID,
+		toolName: toolCall.function.name,
 		input: toolCall.function.arguments,
-		providerMetadata: { databricks: { toolName: toolCall.function.name } }
+		...options.useRemoteToolCalling && {
+			dynamic: true,
+			providerExecuted: true
+		}
 	};
 };
 const mapContentItemsToStreamParts = (items, id) => {
@@ -1558,14 +1684,13 @@ const mapContentItemsToStreamParts = (items, id) => {
 			});
 			break;
 		case "image": break;
-		case "reasoning": {
+		case "reasoning":
 			for (const summary of item.summary.filter((s) => s.type === "summary_text")) parts.push({
 				type: "reasoning-delta",
 				id,
 				delta: summary.text
 			});
 			break;
-		}
 	}
 	return parts;
 };
@@ -1579,13 +1704,12 @@ const mapContentItemsToProviderContent = (items) => {
 			});
 			break;
 		case "image": break;
-		case "reasoning": {
+		case "reasoning":
 			for (const summary of item.summary.filter((s) => s.type === "summary_text")) parts.push({
 				type: "reasoning",
 				text: summary.text
 			});
 			break;
-		}
 	}
 	return parts;
 };
@@ -1721,31 +1845,108 @@ const convertToolResultOutputToContentValue = (output) => {
 };
 const ProviderOptionsSchema = z.object({ toolName: z.string().nullish() });
 const getToolNameFromPart = async (part) => {
-	const providerOptions = await parseProviderOptions({
+	return (await parseProviderOptions({
 		provider: "databricks",
 		providerOptions: part.providerOptions,
 		schema: ProviderOptionsSchema
-	});
-	return providerOptions?.toolName ?? part.toolName;
+	}))?.toolName ?? part.toolName;
 };
 
 //#endregion
 //#region src/fmapi-language-model/fmapi-finish-reason.ts
 function mapFmapiFinishReason(finishReason) {
 	switch (finishReason) {
-		case "stop": return "stop";
-		case "length": return "length";
-		case "content_filter": return "content-filter";
+		case "stop": return {
+			raw: finishReason,
+			unified: "stop"
+		};
+		case "length": return {
+			raw: finishReason,
+			unified: "length"
+		};
+		case "content_filter": return {
+			raw: finishReason,
+			unified: "content-filter"
+		};
 		case "function_call":
-		case "tool_calls": return "tool-calls";
-		default: return "other";
+		case "tool_calls": return {
+			raw: finishReason,
+			unified: "tool-calls"
+		};
+		default: return {
+			raw: finishReason ?? void 0,
+			unified: "other"
+		};
 	}
 }
 
+//#endregion
+//#region src/fmapi-language-model/call-options-to-fmapi-args.ts
+/**
+* Converts AI SDK LanguageModelV3CallOptions to Databricks FMAPI body parameters.
+*
+* Inspired by the getArgs method in:
+* https://github.com/vercel/ai/blob/main/packages/openai/src/chat/openai-chat-language-model.ts#L71
+*
+* Complies with the API described in:
+* https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/api-reference#chat-request
+*/
+function callOptionsToFmapiArgs(options) {
+	const warnings = [];
+	const databricksOptions = options.providerOptions?.databricks;
+	if (options.presencePenalty != null) warnings.push({
+		type: "unsupported",
+		feature: "presencePenalty",
+		details: "presencePenalty is not supported by the Databricks FMAPI"
+	});
+	if (options.frequencyPenalty != null) warnings.push({
+		type: "unsupported",
+		feature: "frequencyPenalty",
+		details: "frequencyPenalty is not supported by the Databricks FMAPI"
+	});
+	if (options.seed != null) warnings.push({
+		type: "unsupported",
+		feature: "seed",
+		details: "seed is not supported by the Databricks FMAPI"
+	});
+	const args = {};
+	if (options.maxOutputTokens != null) args.max_tokens = options.maxOutputTokens;
+	if (options.temperature != null) args.temperature = options.temperature;
+	if (options.topP != null) args.top_p = options.topP;
+	if (options.topK != null) args.top_k = options.topK;
+	if (options.stopSequences != null && options.stopSequences.length > 0) args.stop = options.stopSequences;
+	if (options.responseFormat != null) switch (options.responseFormat.type) {
+		case "text":
+			args.response_format = { type: "text" };
+			break;
+		case "json":
+			if (options.responseFormat.schema != null) args.response_format = {
+				type: "json_schema",
+				json_schema: {
+					name: options.responseFormat.name ?? "response",
+					description: options.responseFormat.description,
+					schema: options.responseFormat.schema,
+					strict: true
+				}
+			};
+			else args.response_format = { type: "json_object" };
+			break;
+	}
+	if (databricksOptions?.topK != null) args.top_k = databricksOptions.topK;
+	if (databricksOptions?.n != null) args.n = databricksOptions.n;
+	if (databricksOptions?.logprobs != null) args.logprobs = databricksOptions.logprobs;
+	if (databricksOptions?.topLogprobs != null) args.top_logprobs = databricksOptions.topLogprobs;
+	if (databricksOptions?.reasoningEffort != null) args.reasoning_effort = databricksOptions.reasoningEffort;
+	return {
+		args,
+		warnings
+	};
+}
+
 //#endregion
 //#region src/fmapi-language-model/fmapi-language-model.ts
 var DatabricksFmapiLanguageModel = class {
-	specificationVersion = "v2";
+	specificationVersion = "v3";
 	modelId;
 	config;
 	constructor(modelId, config) {
@@ -1757,7 +1958,7 @@ var DatabricksFmapiLanguageModel = class {
 	}
 	supportedUrls = {};
 	async doGenerate(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings, ...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: false,
@@ -1775,18 +1976,26 @@ var DatabricksFmapiLanguageModel = class {
 		const choice = response.choices[0];
 		const finishReason = mapFmapiFinishReason(choice?.finish_reason);
 		return {
-			content: convertFmapiResponseToMessagePart(response),
+			content: convertFmapiResponseToMessagePart(response, { useRemoteToolCalling: this.config.useRemoteToolCalling ?? false }),
 			finishReason,
 			usage: {
-				inputTokens: response.usage?.prompt_tokens ?? 0,
-				outputTokens: response.usage?.completion_tokens ?? 0,
-				totalTokens: response.usage?.total_tokens ?? 0
+				inputTokens: {
+					total: response.usage?.prompt_tokens ?? 0,
+					noCache: 0,
+					cacheRead: 0,
+					cacheWrite: 0
+				},
+				outputTokens: {
+					total: response.usage?.completion_tokens ?? 0,
+					text: 0,
+					reasoning: 0
+				}
 			},
-			warnings: []
+			warnings
 		};
 	}
 	async doStream(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings, ...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: true,
@@ -1802,21 +2011,33 @@ var DatabricksFmapiLanguageModel = class {
 			successfulResponseHandler: createEventSourceResponseHandler(fmapiChunkSchema),
 			abortSignal: options.abortSignal
 		});
-		let finishReason = "unknown";
+		let finishReason = {
+			raw: void 0,
+			unified: "other"
+		};
 		let usage = {
-			inputTokens: 0,
-			outputTokens: 0,
-			totalTokens: 0
+			inputTokens: {
+				total: 0,
+				noCache: 0,
+				cacheRead: 0,
+				cacheWrite: 0
+			},
+			outputTokens: {
+				total: 0,
+				text: 0,
+				reasoning: 0
+			}
 		};
-		const toolCallIdsByIndex = new Map();
-		const toolCallNamesById = new Map();
-		const toolCallInputsById = new Map();
+		const toolCallIdsByIndex = /* @__PURE__ */ new Map();
+		const toolCallNamesById = /* @__PURE__ */ new Map();
+		const toolCallInputsById = /* @__PURE__ */ new Map();
+		const useRemoteToolCalling = this.config.useRemoteToolCalling ?? false;
 		return {
 			stream: response.pipeThrough(new TransformStream({
 				start(controller) {
 					controller.enqueue({
 						type: "stream-start",
-						warnings: []
+						warnings
 					});
 				},
 				transform(chunk, controller) {
@@ -1825,7 +2046,10 @@ var DatabricksFmapiLanguageModel = class {
 						rawValue: chunk.rawValue
 					});
 					if (!chunk.success) {
-						finishReason = "error";
+						finishReason = {
+							raw: void 0,
+							unified: "error"
+						};
 						controller.enqueue({
 							type: "error",
 							error: chunk.error
@@ -1835,9 +2059,17 @@ var DatabricksFmapiLanguageModel = class {
 					const choice = chunk.value.choices[0];
 					finishReason = mapFmapiFinishReason(choice?.finish_reason);
 					if (chunk.value.usage) usage = {
-						inputTokens: chunk.value.usage.prompt_tokens ?? 0,
-						outputTokens: chunk.value.usage.completion_tokens ?? 0,
-						totalTokens: chunk.value.usage.total_tokens ?? 0
+						inputTokens: {
+							total: chunk.value.usage.prompt_tokens ?? 0,
+							noCache: 0,
+							cacheRead: 0,
+							cacheWrite: 0
+						},
+						outputTokens: {
+							total: chunk.value.usage.completion_tokens ?? 0,
+							text: 0,
+							reasoning: 0
+						}
 					};
 					const parts = convertFmapiChunkToMessagePart(chunk.value, toolCallIdsByIndex);
 					for (const part of parts) {
@@ -1862,9 +2094,12 @@ var DatabricksFmapiLanguageModel = class {
 							controller.enqueue({
 								type: "tool-call",
 								toolCallId,
-								toolName: DATABRICKS_TOOL_CALL_ID,
+								toolName,
 								input: inputText,
-								providerMetadata: { databricks: { toolName } }
+								...useRemoteToolCalling && {
+									dynamic: true,
+									providerExecuted: true
+								}
 							});
 						}
 					}
@@ -1883,6 +2118,7 @@ var DatabricksFmapiLanguageModel = class {
 		const tools = options.tools?.map((tool) => convertToolToOpenAIFormat(tool)).filter((tool) => tool !== void 0);
 		const toolChoice = options.toolChoice ? convertToolChoiceToOpenAIFormat(options.toolChoice) : void 0;
 		const { messages } = await convertPromptToFmapiMessages(options.prompt);
+		const { args: callArgs, warnings } = callOptionsToFmapiArgs(options);
 		return {
 			url: config.url({ path: "/chat/completions" }),
 			headers: combineHeaders(config.headers(), options.headers),
@@ -1892,10 +2128,9 @@ var DatabricksFmapiLanguageModel = class {
 				model: modelId,
 				...tools && tools.length > 0 ? { tools } : {},
 				...toolChoice && tools && tools.length > 0 ? { tool_choice: toolChoice } : {},
-				...options.temperature !== void 0 ? { temperature: options.temperature } : {},
-				...options.maxOutputTokens !== void 0 ? { max_tokens: options.maxOutputTokens } : {},
-				...options.stopSequences && options.stopSequences.length > 0 ? { stop: options.stopSequences } : {}
+				...callArgs
 			},
+			warnings,
 			fetch: config.fetch
 		};
 	}
@@ -1904,7 +2139,7 @@ var DatabricksFmapiLanguageModel = class {
 * Convert AI SDK tool to OpenAI format
 */
 function convertToolToOpenAIFormat(tool) {
-	if (tool.type === "provider-defined" || tool.name === DATABRICKS_TOOL_CALL_ID) return void 0;
+	if (tool.type === "provider") return;
 	return {
 		type: "function",
 		function: {
@@ -1949,13 +2184,15 @@ const createDatabricksProvider = (settings) => {
 		url: formatUrl,
 		headers: getHeaders,
 		fetch,
-		provider
+		provider,
+		useRemoteToolCalling: settings.useRemoteToolCalling
 	});
 	const createFmapi = (modelId) => new DatabricksFmapiLanguageModel(modelId, {
 		url: formatUrl,
 		headers: getHeaders,
 		fetch,
-		provider
+		provider,
+		useRemoteToolCalling: settings.useRemoteToolCalling
 	});
 	const notImplemented = (name) => {
 		return () => {
@@ -1963,15 +2200,17 @@ const createDatabricksProvider = (settings) => {
 		};
 	};
 	return {
+		specificationVersion: "v3",
 		responses: createResponsesAgent,
 		chatCompletions: createFmapi,
 		chatAgent: createChatAgent,
 		imageModel: notImplemented("ImageModel"),
 		textEmbeddingModel: notImplemented("TextEmbeddingModel"),
+		embeddingModel: notImplemented("EmbeddingModel"),
 		languageModel: notImplemented("LanguageModel")
 	};
 };
 
 //#endregion
-export { DATABRICKS_TOOL_CALL_ID, DATABRICKS_TOOL_DEFINITION, MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
-//# sourceMappingURL=index.js.map
+export { MCP_APPROVAL_REQUEST_TYPE, MCP_APPROVAL_RESPONSE_TYPE, MCP_APPROVAL_STATUS_KEY, createApprovalStatusOutput, createDatabricksProvider, extractApprovalStatus, extractApprovalStatusFromToolResult, extractDatabricksMetadata, getMcpApprovalState, isApprovalStatusOutput, isMcpApprovalRequest, isMcpApprovalResponse };
+//# sourceMappingURL=index.mjs.map