@databricks/ai-sdk-provider 0.2.2 → 0.2.3

package/dist/index.cjs

@@ -1189,6 +1189,75 @@ function convertResponsesToolChoice(toolChoice) {
 	}
 }
 
+//#endregion
+//#region src/responses-agent-language-model/call-options-to-responses-args.ts
+/**
+* Converts AI SDK LanguageModelV2CallOptions 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-setting",
+		setting: "topK",
+		details: "topK is not supported by the Databricks Responses API"
+	});
+	if (options.presencePenalty != null) warnings.push({
+		type: "unsupported-setting",
+		setting: "presencePenalty",
+		details: "presencePenalty is not supported by the Databricks Responses API"
+	});
+	if (options.frequencyPenalty != null) warnings.push({
+		type: "unsupported-setting",
+		setting: "frequencyPenalty",
+		details: "frequencyPenalty is not supported by the Databricks Responses API"
+	});
+	if (options.seed != null) warnings.push({
+		type: "unsupported-setting",
+		setting: "seed",
+		details: "seed is not supported by the Databricks Responses API"
+	});
+	if (options.stopSequences != null && options.stopSequences.length > 0) warnings.push({
+		type: "unsupported-setting",
+		setting: "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 }) {
@@ -1213,7 +1282,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,
@@ -1241,11 +1310,11 @@ var DatabricksResponsesAgentLanguageModel = class {
 				outputTokens: response.usage?.output_tokens ?? 0,
 				totalTokens: response.usage?.total_tokens ?? 0
 			},
-			warnings: []
+			warnings
 		};
 	}
 	async doStream(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings,...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: true,
@@ -1273,7 +1342,7 @@ var DatabricksResponsesAgentLanguageModel = class {
 				start(controller) {
 					controller.enqueue({
 						type: "stream-start",
-						warnings: []
+						warnings
 					});
 				},
 				transform(chunk, controller) {
@@ -1348,6 +1417,7 @@ var DatabricksResponsesAgentLanguageModel = class {
 			tools: options.tools,
 			toolChoice: options.toolChoice
 		});
+		const { args: callArgs, warnings } = callOptionsToResponsesArgs(options);
 		return {
 			url: config.url({ path: "/responses" }),
 			headers: (0, __ai_sdk_provider_utils.combineHeaders)(config.headers(), options.headers),
@@ -1356,8 +1426,10 @@ var DatabricksResponsesAgentLanguageModel = class {
 				input,
 				stream,
 				...tools ? { tools } : {},
-				...toolChoice && tools ? { tool_choice: toolChoice } : {}
+				...toolChoice && tools ? { tool_choice: toolChoice } : {},
+				...callArgs
 			},
+			warnings,
 			fetch: config.fetch
 		};
 	}
@@ -1766,6 +1838,69 @@ function mapFmapiFinishReason(finishReason) {
 	}
 }
 
+//#endregion
+//#region src/fmapi-language-model/call-options-to-fmapi-args.ts
+/**
+* Converts AI SDK LanguageModelV2CallOptions 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-setting",
+		setting: "presencePenalty",
+		details: "presencePenalty is not supported by the Databricks FMAPI"
+	});
+	if (options.frequencyPenalty != null) warnings.push({
+		type: "unsupported-setting",
+		setting: "frequencyPenalty",
+		details: "frequencyPenalty is not supported by the Databricks FMAPI"
+	});
+	if (options.seed != null) warnings.push({
+		type: "unsupported-setting",
+		setting: "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 {
@@ -1781,7 +1916,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,
@@ -1806,11 +1941,11 @@ var DatabricksFmapiLanguageModel = class {
 				outputTokens: response.usage?.completion_tokens ?? 0,
 				totalTokens: response.usage?.total_tokens ?? 0
 			},
-			warnings: []
+			warnings
 		};
 	}
 	async doStream(options) {
-		const networkArgs = await this.getArgs({
+		const { warnings,...networkArgs } = await this.getArgs({
 			config: this.config,
 			options,
 			stream: true,
@@ -1840,7 +1975,7 @@ var DatabricksFmapiLanguageModel = class {
 				start(controller) {
 					controller.enqueue({
 						type: "stream-start",
-						warnings: []
+						warnings
 					});
 				},
 				transform(chunk, controller) {
@@ -1907,6 +2042,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: (0, __ai_sdk_provider_utils.combineHeaders)(config.headers(), options.headers),
@@ -1916,10 +2052,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
 		};
 	}