@databricks/ai-sdk-provider 0.2.0 → 0.2.2

package/dist/index.js

@@ -521,7 +521,7 @@ const responsesAgentResponseSchema = z.object({
 	}).nullish(),
 	model: z.string().optional(),
 	output: z.array(responsesAgentOutputItem),
-	incomplete_details: z.object({ reason: z.enum(["max_output_tokens", "content_filter"]).optional() }).nullish(),
+	incomplete_details: z.object({ reason: z.string().nullish().optional() }).nullish(),
 	usage: z.object({
 		input_tokens: z.number(),
 		output_tokens: z.number(),
@@ -591,7 +591,7 @@ const responsesCompletedSchema = z.object({
 			"queued",
 			"incomplete"
 		]).optional(),
-		incomplete_details: z.object({ reason: z.enum(["max_output_tokens", "content_filter"]).optional() }).nullish(),
+		incomplete_details: z.object({ reason: z.string().nullish().optional() }).nullish(),
 		usage: z.object({
 			input_tokens: z.number(),
 			output_tokens: z.number(),
@@ -1455,11 +1455,7 @@ const fmapiChunkSchema = z.object({
 			]).optional(),
 			tool_calls: z.array(toolCallDeltaSchema).optional()
 		}),
-		finish_reason: z.union([
-			z.literal("stop"),
-			z.literal("tool_calls"),
-			z.null()
-		]).optional()
+		finish_reason: z.union([z.string(), z.null()]).optional()
 	}))
 });
 const fmapiResponseSchema = z.object({
@@ -1485,11 +1481,7 @@ const fmapiResponseSchema = z.object({
 			]).optional(),
 			tool_calls: z.array(toolCallSchema).optional()
 		}),
-		finish_reason: z.union([
-			z.literal("stop"),
-			z.literal("tool_calls"),
-			z.null()
-		]).optional()
+		finish_reason: z.union([z.string(), z.null()]).optional()
 	}))
 });
 
@@ -1737,6 +1729,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 {
@@ -1768,8 +1773,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,
@@ -1829,8 +1833,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,