@databricks/ai-sdk-provider 0.2.0 → 0.2.2

package/README.md

@@ -4,12 +4,12 @@ Databricks provider for the [Vercel AI SDK](https://sdk.vercel.ai/docs).
 
 ## Features
 
-- 🚀 Support for three Databricks endpoint types:
-  - **Chat Agent** (`agent/v2/chat`) - Databricks chat agent API
-  - **Responses Agent** (`agent/v1/responses`) - Databricks responses agent API
-  - **FM API** (`llm/v1/chat`) - Foundation model chat completions API
+- 🚀 Support for all Databricks endpoint types:
+  - **Responses** (`agent/v1/responses`) - Foundation model and agent responses API ([docs](https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/api-reference#responses-api))
+  - **Chat Completions** (`llm/v1/chat`) - Foundation model chat completions API ([docs](https://docs.databricks.com/aws/en/machine-learning/foundation-model-apis/api-reference#chat-completions-api))
+  - **Chat Agent** (`agent/v2/chat`) - Legacy Databricks chat agent API ([docs](https://docs.databricks.com/aws/en/generative-ai/agent-framework/agent-legacy-schema))
 - 🔄 Stream and non-stream (generate) support for all endpoint types
-- 🛠️ Custom tool calling mechanism for Databricks agents
+- 🛠️ Tool calling and agent support
 - 🔐 Flexible authentication (bring your own tokens/headers)
 - 🎯 Full TypeScript support
 

package/dist/index.cjs

@@ -545,7 +545,7 @@ const responsesAgentResponseSchema = zod_v4.z.object({
 	}).nullish(),
 	model: zod_v4.z.string().optional(),
 	output: zod_v4.z.array(responsesAgentOutputItem),
-	incomplete_details: zod_v4.z.object({ reason: zod_v4.z.enum(["max_output_tokens", "content_filter"]).optional() }).nullish(),
+	incomplete_details: zod_v4.z.object({ reason: zod_v4.z.string().nullish().optional() }).nullish(),
 	usage: zod_v4.z.object({
 		input_tokens: zod_v4.z.number(),
 		output_tokens: zod_v4.z.number(),
@@ -615,7 +615,7 @@ const responsesCompletedSchema = zod_v4.z.object({
 			"queued",
 			"incomplete"
 		]).optional(),
-		incomplete_details: zod_v4.z.object({ reason: zod_v4.z.enum(["max_output_tokens", "content_filter"]).optional() }).nullish(),
+		incomplete_details: zod_v4.z.object({ reason: zod_v4.z.string().nullish().optional() }).nullish(),
 		usage: zod_v4.z.object({
 			input_tokens: zod_v4.z.number(),
 			output_tokens: zod_v4.z.number(),
@@ -1479,11 +1479,7 @@ const fmapiChunkSchema = zod_v4.z.object({
 			]).optional(),
 			tool_calls: zod_v4.z.array(toolCallDeltaSchema).optional()
 		}),
-		finish_reason: zod_v4.z.union([
-			zod_v4.z.literal("stop"),
-			zod_v4.z.literal("tool_calls"),
-			zod_v4.z.null()
-		]).optional()
+		finish_reason: zod_v4.z.union([zod_v4.z.string(), zod_v4.z.null()]).optional()
 	}))
 });
 const fmapiResponseSchema = zod_v4.z.object({
@@ -1509,11 +1505,7 @@ const fmapiResponseSchema = zod_v4.z.object({
 			]).optional(),
 			tool_calls: zod_v4.z.array(toolCallSchema).optional()
 		}),
-		finish_reason: zod_v4.z.union([
-			zod_v4.z.literal("stop"),
-			zod_v4.z.literal("tool_calls"),
-			zod_v4.z.null()
-		]).optional()
+		finish_reason: zod_v4.z.union([zod_v4.z.string(), zod_v4.z.null()]).optional()
 	}))
 });
 
@@ -1761,6 +1753,19 @@ const getToolNameFromPart = async (part) => {
 	return providerOptions?.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 "function_call":
+		case "tool_calls": return "tool-calls";
+		default: return "other";
+	}
+}
+
 //#endregion
 //#region src/fmapi-language-model/fmapi-language-model.ts
 var DatabricksFmapiLanguageModel = class {
@@ -1792,8 +1797,7 @@ var DatabricksFmapiLanguageModel = class {
 			})
 		});
 		const choice = response.choices[0];
-		let finishReason = "stop";
-		if (choice?.finish_reason === "tool_calls") finishReason = "tool-calls";
+		const finishReason = mapFmapiFinishReason(choice?.finish_reason);
 		return {
 			content: convertFmapiResponseToMessagePart(response),
 			finishReason,
@@ -1853,8 +1857,7 @@ var DatabricksFmapiLanguageModel = class {
 						return;
 					}
 					const choice = chunk.value.choices[0];
-					if (choice?.finish_reason === "stop") finishReason = "stop";
-					else if (choice?.finish_reason === "tool_calls") finishReason = "tool-calls";
+					finishReason = mapFmapiFinishReason(choice?.finish_reason);
 					if (chunk.value.usage) usage = {
 						inputTokens: chunk.value.usage.prompt_tokens ?? 0,
 						outputTokens: chunk.value.usage.completion_tokens ?? 0,