sarvam-ai-sdk 0.3.0 → 0.3.1

package/README.md

@@ -18,6 +18,15 @@ npm i sarvam-ai-sdk ai@6
 > [!WARNING]
 > This package only works with Vercel AI-SDK v6, not v7. Make sure to install `ai@6` in your project.
 
+### Version Compatibility
+
+| Sarvam AI SDK Version | Vercel AI SDK Version |
+|-----------------------|-----------------------|
+| 0.4.x (beta)			| 7.x.x (beta)			|
+| 0.3.x (current)		| 6.x.x (current)		|
+| 0.2.x					| 5.x.x					|
+| 0.1.x					| 4.x.x					|
+
 ## Provider Instance
 
 You can import the default provider instance `sarvam` from `sarvam-ai-sdk`:
@@ -160,29 +169,56 @@ console.log(result.toolResults);
 ```
 
 ## Generate JSON object
+> NB: `generateObject` is deprecated, use `generateText` with `Output` tool.
 
 ```ts
 import { z } from "zod";
 import { sarvam } from "sarvam-ai-sdk";
-import { generateObject } from 'ai';
+import { generateObject } from "ai";
 
 const { object } = await generateObject({
-  model: sarvam("sarvam-30b"),
-  schemaName: "Recipe",
-  schemaDescription: "A recipe with a name, ingredients and steps",
-  schema: z.object({
-    recipe: z.object({
-      name: z.string(),
-      ingredients: z.array(z.string()),
-      steps: z.array(z.string()),
-    }),
-  }),
-  prompt: 'Generate a South Indian recipe, in Malayalam',
+	model: sarvam("sarvam-30b"),
+	schemaName: "Recipe",
+	schemaDescription: "A recipe with a name, ingredients and steps",
+	schema: z.object({
+		recipe: z.object({
+			name: z.string(),
+			ingredients: z.array(z.string()),
+			steps: z.array(z.string()),
+		}),
+	}),
+	prompt: "Generate a South Indian recipe, in Malayalam",
 });
 
 console.log(object);
 ```
 
+## Generating Structured Outputs
+
+```ts
+import { z } from "zod";
+import { sarvam } from "sarvam-ai-sdk";
+import { generateText, Output } from "ai";
+
+const { output } = await generateText({
+	model: sarvam("sarvam-105b"),
+	output: Output.object({
+		name: "Recipe",
+		description: "A recipe with a name, ingredients and steps",
+		schema: z.object({
+			recipe: z.object({
+				name: z.string(),
+				ingredients: z.array(z.string()),
+				steps: z.array(z.string()),
+			}),
+		}),
+	}),
+	prompt: "Generate a South Indian recipe, in Malayalam",
+});
+
+console.log(output);
+```
+
 ## All APIs
 
 ```ts

package/dist/index.cjs

@@ -135,8 +135,8 @@ function prepareTools({ tools, toolChoice }) {
 	};
 	const sarvamTools = [];
 	for (const tool of finalTools) if (tool.type === "provider") toolWarnings.push({
-		type: "unsupported-tool",
-		tool
+		type: "unsupported",
+		feature: tool.name
 	});
 	else sarvamTools.push({
 		type: "function",
@@ -196,7 +196,6 @@ const chatResponseSchema = zod.default.object({
 		logprobs: zod.default.object({}).nullish(),
 		message: zod.default.object({
 			content: zod.default.string().nullish(),
-			reasoning: zod.default.string().nullish(),
 			reasoning_content: zod.default.string().nullish(),
 			refusal: zod.default.string().nullish(),
 			tool_calls: zod.default.array(zod.default.object({
@@ -341,7 +340,7 @@ var SarvamChatLanguageModel = class {
 			...options,
 			stream: false
 		});
-		const body = JSON.stringify(args);
+		const isJSON = options.responseFormat?.type === "json";
 		const { responseHeaders, value: response, rawValue: rawResponse } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
 			url: this.config.url({
 				path: "/chat/completions",
@@ -364,17 +363,16 @@ var SarvamChatLanguageModel = class {
 			type: "text",
 			text: choice.message.content
 		});
-		const reasoningText = choice.message.reasoning ?? choice.message.reasoning_content;
-		if (reasoningText) content.push({
+		if (choice.message.reasoning_content) content.push({
 			type: "reasoning",
-			text: reasoningText
+			text: choice.message.reasoning_content
 		});
-		if (choice.message.tool_calls && choice.message.tool_calls.length > 0) {
-			if (options.responseFormat?.type === "json") content.push({
+		if (choice.message.tool_calls && choice.message.tool_calls.length > 0) for (const toolCall of choice.message.tool_calls) {
+			if (isJSON) content.push({
 				type: "text",
-				text: choice.message.tool_calls[0].function.arguments
+				text: toolCall.function.arguments
 			});
-			for (const toolCall of choice.message.tool_calls) content.push({
+			content.push({
 				type: "tool-call",
 				toolCallId: toolCall.id ?? (this.config.generateId ?? __ai_sdk_provider_utils.generateId)(),
 				toolName: toolCall.function.name,
@@ -384,7 +382,7 @@ var SarvamChatLanguageModel = class {
 		return {
 			content,
 			finishReason: {
-				unified: mapFinishReason(choice.finish_reason),
+				unified: isJSON ? "stop" : mapFinishReason(choice.finish_reason),
 				raw: choice.finish_reason ?? void 0
 			},
 			usage: {
@@ -400,11 +398,18 @@ var SarvamChatLanguageModel = class {
 					reasoning: void 0
 				}
 			},
+			providerMetadata: { sarvam: {
+				system_fingerprint: response.system_fingerprint,
+				service_tier: response.service_tier
+			} },
 			warnings,
-			request: { body },
+			request: { body: args },
 			response: {
 				headers: responseHeaders,
-				body: rawResponse
+				body: rawResponse,
+				id: response.id ?? void 0,
+				modelId: response.model ?? void 0,
+				timestamp: response.created ? /* @__PURE__ */ new Date(response.created * 1e3) : void 0
 			}
 		};
 	}
@@ -413,10 +418,6 @@ var SarvamChatLanguageModel = class {
 			...options,
 			stream: true
 		});
-		const body = JSON.stringify({
-			...args,
-			stream: true
-		});
 		const { responseHeaders, value: response } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
 			url: this.config.url({
 				path: "/chat/completions",
@@ -582,7 +583,7 @@ var SarvamChatLanguageModel = class {
 					});
 				}
 			})),
-			request: { body },
+			request: { body: args },
 			response: { headers: responseHeaders }
 		};
 	}
@@ -731,7 +732,12 @@ var SarvamTranscriptionModel = class {
 			language: response.language_code ?? void 0,
 			durationInSeconds: response.timestamps?.end_time_seconds[response.timestamps.end_time_seconds.length - 1] ?? void 0,
 			warnings,
-			providerMetadata: { sarvam: { languageProbability: response.language_probability ?? void 0 } },
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				transcript: response.transcript,
+				language_code: response.language_code,
+				language_probability: response.language_probability
+			} },
 			response: {
 				timestamp: currentDate,
 				modelId: this.modelId,
@@ -880,7 +886,8 @@ var SarvamSpeechModel = class {
 		return {
 			audio,
 			warnings,
-			request: { body: JSON.stringify(requestBody) },
+			providerMetadata: { sarvam: { request_id: value.request_id } },
+			request: { body: requestBody },
 			response: {
 				timestamp: currentDate,
 				modelId: this.modelId,
@@ -931,7 +938,7 @@ var SarvamLidModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
 			url: this.config.url({
 				path: "/text-lid",
 				modelId: this.modelId
@@ -965,6 +972,17 @@ var SarvamLidModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				script_code: response.script_code,
+				language_code: response.language_code
+			} },
 			warnings: []
 		};
 	}
@@ -1052,7 +1070,7 @@ var SarvamTranslationModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
 			url: this.config.url({
 				path: "/translate",
 				modelId: this.modelId
@@ -1086,6 +1104,17 @@ var SarvamTranslationModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				source_language_code: response.source_language_code,
+				translated_text: response.translated_text
+			} },
 			warnings: []
 		};
 	}
@@ -1158,7 +1187,7 @@ var SarvamTransliterateModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await (0, __ai_sdk_provider_utils.postJsonToApi)({
 			url: this.config.url({
 				path: "/transliterate",
 				modelId: this.modelId
@@ -1192,6 +1221,17 @@ var SarvamTransliterateModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				source_language_code: response.source_language_code,
+				transliterated_text: response.transliterated_text
+			} },
 			warnings: []
 		};
 	}

package/dist/index.js

@@ -109,8 +109,8 @@ function prepareTools({ tools, toolChoice }) {
 	};
 	const sarvamTools = [];
 	for (const tool of finalTools) if (tool.type === "provider") toolWarnings.push({
-		type: "unsupported-tool",
-		tool
+		type: "unsupported",
+		feature: tool.name
 	});
 	else sarvamTools.push({
 		type: "function",
@@ -170,7 +170,6 @@ const chatResponseSchema = z$1.object({
 		logprobs: z$1.object({}).nullish(),
 		message: z$1.object({
 			content: z$1.string().nullish(),
-			reasoning: z$1.string().nullish(),
 			reasoning_content: z$1.string().nullish(),
 			refusal: z$1.string().nullish(),
 			tool_calls: z$1.array(z$1.object({
@@ -315,7 +314,7 @@ var SarvamChatLanguageModel = class {
 			...options,
 			stream: false
 		});
-		const body = JSON.stringify(args);
+		const isJSON = options.responseFormat?.type === "json";
 		const { responseHeaders, value: response, rawValue: rawResponse } = await postJsonToApi({
 			url: this.config.url({
 				path: "/chat/completions",
@@ -338,17 +337,16 @@ var SarvamChatLanguageModel = class {
 			type: "text",
 			text: choice.message.content
 		});
-		const reasoningText = choice.message.reasoning ?? choice.message.reasoning_content;
-		if (reasoningText) content.push({
+		if (choice.message.reasoning_content) content.push({
 			type: "reasoning",
-			text: reasoningText
+			text: choice.message.reasoning_content
 		});
-		if (choice.message.tool_calls && choice.message.tool_calls.length > 0) {
-			if (options.responseFormat?.type === "json") content.push({
+		if (choice.message.tool_calls && choice.message.tool_calls.length > 0) for (const toolCall of choice.message.tool_calls) {
+			if (isJSON) content.push({
 				type: "text",
-				text: choice.message.tool_calls[0].function.arguments
+				text: toolCall.function.arguments
 			});
-			for (const toolCall of choice.message.tool_calls) content.push({
+			content.push({
 				type: "tool-call",
 				toolCallId: toolCall.id ?? (this.config.generateId ?? generateId)(),
 				toolName: toolCall.function.name,
@@ -358,7 +356,7 @@ var SarvamChatLanguageModel = class {
 		return {
 			content,
 			finishReason: {
-				unified: mapFinishReason(choice.finish_reason),
+				unified: isJSON ? "stop" : mapFinishReason(choice.finish_reason),
 				raw: choice.finish_reason ?? void 0
 			},
 			usage: {
@@ -374,11 +372,18 @@ var SarvamChatLanguageModel = class {
 					reasoning: void 0
 				}
 			},
+			providerMetadata: { sarvam: {
+				system_fingerprint: response.system_fingerprint,
+				service_tier: response.service_tier
+			} },
 			warnings,
-			request: { body },
+			request: { body: args },
 			response: {
 				headers: responseHeaders,
-				body: rawResponse
+				body: rawResponse,
+				id: response.id ?? void 0,
+				modelId: response.model ?? void 0,
+				timestamp: response.created ? /* @__PURE__ */ new Date(response.created * 1e3) : void 0
 			}
 		};
 	}
@@ -387,10 +392,6 @@ var SarvamChatLanguageModel = class {
 			...options,
 			stream: true
 		});
-		const body = JSON.stringify({
-			...args,
-			stream: true
-		});
 		const { responseHeaders, value: response } = await postJsonToApi({
 			url: this.config.url({
 				path: "/chat/completions",
@@ -556,7 +557,7 @@ var SarvamChatLanguageModel = class {
 					});
 				}
 			})),
-			request: { body },
+			request: { body: args },
 			response: { headers: responseHeaders }
 		};
 	}
@@ -705,7 +706,12 @@ var SarvamTranscriptionModel = class {
 			language: response.language_code ?? void 0,
 			durationInSeconds: response.timestamps?.end_time_seconds[response.timestamps.end_time_seconds.length - 1] ?? void 0,
 			warnings,
-			providerMetadata: { sarvam: { languageProbability: response.language_probability ?? void 0 } },
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				transcript: response.transcript,
+				language_code: response.language_code,
+				language_probability: response.language_probability
+			} },
 			response: {
 				timestamp: currentDate,
 				modelId: this.modelId,
@@ -854,7 +860,8 @@ var SarvamSpeechModel = class {
 		return {
 			audio,
 			warnings,
-			request: { body: JSON.stringify(requestBody) },
+			providerMetadata: { sarvam: { request_id: value.request_id } },
+			request: { body: requestBody },
 			response: {
 				timestamp: currentDate,
 				modelId: this.modelId,
@@ -905,7 +912,7 @@ var SarvamLidModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await postJsonToApi({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await postJsonToApi({
 			url: this.config.url({
 				path: "/text-lid",
 				modelId: this.modelId
@@ -939,6 +946,17 @@ var SarvamLidModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				script_code: response.script_code,
+				language_code: response.language_code
+			} },
 			warnings: []
 		};
 	}
@@ -1026,7 +1044,7 @@ var SarvamTranslationModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await postJsonToApi({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await postJsonToApi({
 			url: this.config.url({
 				path: "/translate",
 				modelId: this.modelId
@@ -1060,6 +1078,17 @@ var SarvamTranslationModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				source_language_code: response.source_language_code,
+				translated_text: response.translated_text
+			} },
 			warnings: []
 		};
 	}
@@ -1132,7 +1161,7 @@ var SarvamTransliterateModel = class {
 			...options,
 			stream: false
 		});
-		const { value: response } = await postJsonToApi({
+		const { responseHeaders, value: response, rawValue: rawResponse } = await postJsonToApi({
 			url: this.config.url({
 				path: "/transliterate",
 				modelId: this.modelId
@@ -1166,6 +1195,17 @@ var SarvamTransliterateModel = class {
 					reasoning: void 0
 				}
 			},
+			request: { body: args },
+			response: {
+				id: response.request_id ?? void 0,
+				headers: responseHeaders,
+				body: rawResponse
+			},
+			providerMetadata: { sarvam: {
+				request_id: response.request_id,
+				source_language_code: response.source_language_code,
+				transliterated_text: response.transliterated_text
+			} },
 			warnings: []
 		};
 	}

package/package.json

@@ -2,7 +2,7 @@
 	"name": "sarvam-ai-sdk",
 	"type": "module",
 	"description": "Sarvam provider support for Vercel's AI-SDK",
-	"version": "0.3.0",
+	"version": "0.3.1",
 	"main": "./dist/index.cjs",
 	"module": "./dist/index.js",
 	"types": "./dist/index.d.ts",