@openrouter/ai-sdk-provider 0.4.2 → 0.4.4

package/dist/index.mjs

@@ -50,15 +50,15 @@ import { z as z2 } from "zod";
 
 // src/convert-to-openrouter-chat-messages.ts
 import { convertUint8ArrayToBase64 } from "@ai-sdk/provider-utils";
+function getCacheControl(providerMetadata) {
+  var _a, _b, _c;
+  const anthropic = providerMetadata == null ? void 0 : providerMetadata.anthropic;
+  const openrouter2 = providerMetadata == null ? void 0 : providerMetadata.openrouter;
+  return (_c = (_b = (_a = openrouter2 == null ? void 0 : openrouter2.cacheControl) != null ? _a : openrouter2 == null ? void 0 : openrouter2.cache_control) != null ? _b : anthropic == null ? void 0 : anthropic.cacheControl) != null ? _c : anthropic == null ? void 0 : anthropic.cache_control;
+}
 function convertToOpenRouterChatMessages(prompt) {
   var _a, _b, _c;
   const messages = [];
-  function getCacheControl(providerMetadata) {
-    var _a2;
-    const anthropic = providerMetadata == null ? void 0 : providerMetadata.anthropic;
-    const cacheControlValue = (_a2 = anthropic == null ? void 0 : anthropic.cacheControl) != null ? _a2 : anthropic == null ? void 0 : anthropic.cache_control;
-    return cacheControlValue;
-  }
   for (const { role, content, providerMetadata } of prompt) {
     switch (role) {
       case "system": {
@@ -81,31 +81,31 @@ function convertToOpenRouterChatMessages(prompt) {
         const messageCacheControl = getCacheControl(providerMetadata);
         const contentParts = content.map(
           (part) => {
-            var _a2, _b2, _c2;
+            var _a2, _b2, _c2, _d;
             switch (part.type) {
               case "text":
                 return {
                   type: "text",
                   text: part.text,
                   // For text parts, only use part-specific cache control
-                  cache_control: getCacheControl(part.providerMetadata)
+                  cache_control: (_a2 = getCacheControl(part.providerMetadata)) != null ? _a2 : messageCacheControl
                 };
               case "image":
                 return {
                   type: "image_url",
                   image_url: {
-                    url: part.image instanceof URL ? part.image.toString() : `data:${(_a2 = part.mimeType) != null ? _a2 : "image/jpeg"};base64,${convertUint8ArrayToBase64(
+                    url: part.image instanceof URL ? part.image.toString() : `data:${(_b2 = part.mimeType) != null ? _b2 : "image/jpeg"};base64,${convertUint8ArrayToBase64(
                       part.image
                     )}`
                   },
                   // For image parts, use part-specific or message-level cache control
-                  cache_control: (_b2 = getCacheControl(part.providerMetadata)) != null ? _b2 : messageCacheControl
+                  cache_control: (_c2 = getCacheControl(part.providerMetadata)) != null ? _c2 : messageCacheControl
                 };
               case "file":
                 return {
                   type: "text",
                   text: part.data instanceof URL ? part.data.toString() : part.data,
-                  cache_control: (_c2 = getCacheControl(part.providerMetadata)) != null ? _c2 : messageCacheControl
+                  cache_control: (_d = getCacheControl(part.providerMetadata)) != null ? _d : messageCacheControl
                 };
               default: {
                 const _exhaustiveCheck = part;
@@ -213,7 +213,7 @@ function mapOpenRouterFinishReason(finishReason) {
 // src/openrouter-error.ts
 import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
 import { z } from "zod";
-var openAIErrorDataSchema = z.object({
+var OpenRouterErrorResponseSchema = z.object({
   error: z.object({
     message: z.string(),
     type: z.string(),
@@ -222,7 +222,7 @@ var openAIErrorDataSchema = z.object({
   })
 });
 var openrouterFailedResponseHandler = createJsonErrorResponseHandler({
-  errorSchema: openAIErrorDataSchema,
+  errorSchema: OpenRouterErrorResponseSchema,
   errorToMessage: (data) => data.error.message
 });
 
@@ -318,7 +318,7 @@ var OpenRouterChatLanguageModel = class {
     }
   }
   async doGenerate(options) {
-    var _b, _c, _d;
+    var _b, _c, _d, _e, _f, _g, _h;
     const args = this.getArgs(options);
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.config.url({
@@ -329,17 +329,21 @@ var OpenRouterChatLanguageModel = class {
       body: args,
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler(
-        openAIChatResponseSchema
+        OpenRouterNonStreamChatCompletionResponseSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
     const _a = args, { messages: rawPrompt } = _a, rawSettings = __objRest(_a, ["messages"]);
     const choice = response.choices[0];
-    if (choice == null) {
+    if (!choice) {
       throw new Error("No choice in response");
     }
     return {
+      response: {
+        id: response.id,
+        modelId: response.model
+      },
       text: (_b = choice.message.content) != null ? _b : void 0,
       reasoning: (_c = choice.message.reasoning) != null ? _c : void 0,
       toolCalls: (_d = choice.message.tool_calls) == null ? void 0 : _d.map((toolCall) => {
@@ -353,8 +357,8 @@ var OpenRouterChatLanguageModel = class {
       }),
       finishReason: mapOpenRouterFinishReason(choice.finish_reason),
       usage: {
-        promptTokens: response.usage.prompt_tokens,
-        completionTokens: response.usage.completion_tokens
+        promptTokens: (_f = (_e = response.usage) == null ? void 0 : _e.prompt_tokens) != null ? _f : 0,
+        completionTokens: (_h = (_g = response.usage) == null ? void 0 : _g.completion_tokens) != null ? _h : 0
       },
       rawCall: { rawPrompt, rawSettings },
       rawResponse: { headers: responseHeaders },
@@ -377,7 +381,7 @@ var OpenRouterChatLanguageModel = class {
       }),
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
-        openrouterChatChunkSchema
+        OpenRouterStreamChatCompletionChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
@@ -412,6 +416,12 @@ var OpenRouterChatLanguageModel = class {
                 id: value.id
               });
             }
+            if (value.model) {
+              controller.enqueue({
+                type: "response-metadata",
+                modelId: value.model
+              });
+            }
             if (value.usage != null) {
               usage = {
                 promptTokens: value.usage.prompt_tokens,
@@ -539,7 +549,16 @@ var OpenRouterChatLanguageModel = class {
     };
   }
 };
-var openAIChatResponseSchema = z2.object({
+var OpenRouterChatCompletionBaseResponseSchema = z2.object({
+  id: z2.string().optional(),
+  model: z2.string().optional(),
+  usage: z2.object({
+    prompt_tokens: z2.number(),
+    completion_tokens: z2.number(),
+    total_tokens: z2.number()
+  }).nullish()
+});
+var OpenRouterNonStreamChatCompletionResponseSchema = OpenRouterChatCompletionBaseResponseSchema.extend({
   choices: z2.array(
     z2.object({
       message: z2.object({
@@ -574,15 +593,10 @@ var openAIChatResponseSchema = z2.object({
       }).nullable().optional(),
       finish_reason: z2.string().optional().nullable()
     })
-  ),
-  usage: z2.object({
-    prompt_tokens: z2.number(),
-    completion_tokens: z2.number()
-  })
+  )
 });
-var openrouterChatChunkSchema = z2.union([
-  z2.object({
-    id: z2.string().optional(),
+var OpenRouterStreamChatCompletionChunkSchema = z2.union([
+  OpenRouterChatCompletionBaseResponseSchema.extend({
     choices: z2.array(
       z2.object({
         delta: z2.object({
@@ -618,13 +632,9 @@ var openrouterChatChunkSchema = z2.union([
         finish_reason: z2.string().nullable().optional(),
         index: z2.number()
       })
-    ),
-    usage: z2.object({
-      prompt_tokens: z2.number(),
-      completion_tokens: z2.number()
-    }).nullish()
+    )
   }),
-  openAIErrorDataSchema
+  OpenRouterErrorResponseSchema
 ]);
 function prepareToolsAndToolChoice(mode) {
   var _a;
@@ -910,6 +920,7 @@ var OpenRouterCompletionLanguageModel = class {
     }
   }
   async doGenerate(options) {
+    var _b, _c, _d, _e, _f;
     const args = this.getArgs(options);
     const { responseHeaders, value: response } = await postJsonToApi2({
       url: this.config.url({
@@ -920,22 +931,29 @@ var OpenRouterCompletionLanguageModel = class {
       body: args,
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createJsonResponseHandler2(
-        openAICompletionResponseSchema
+        OpenRouterCompletionChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
     const _a = args, { prompt: rawPrompt } = _a, rawSettings = __objRest(_a, ["prompt"]);
+    if ("error" in response) {
+      throw new Error(`${response.error.message}`);
+    }
     const choice = response.choices[0];
     if (!choice) {
       throw new Error("No choice in OpenRouter completion response");
     }
     return {
-      text: choice.text,
+      response: {
+        id: response.id,
+        modelId: response.model
+      },
+      text: (_b = choice.text) != null ? _b : "",
       reasoning: choice.reasoning || void 0,
       usage: {
-        promptTokens: response.usage.prompt_tokens,
-        completionTokens: response.usage.completion_tokens
+        promptTokens: (_d = (_c = response.usage) == null ? void 0 : _c.prompt_tokens) != null ? _d : 0,
+        completionTokens: (_f = (_e = response.usage) == null ? void 0 : _e.completion_tokens) != null ? _f : 0
       },
       finishReason: mapOpenRouterFinishReason(choice.finish_reason),
       logprobs: mapOpenRouterCompletionLogProbs(choice.logprobs),
@@ -959,7 +977,7 @@ var OpenRouterCompletionLanguageModel = class {
       }),
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler2(
-        openrouterCompletionChunkSchema
+        OpenRouterCompletionChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
@@ -1026,29 +1044,14 @@ var OpenRouterCompletionLanguageModel = class {
     };
   }
 };
-var openAICompletionResponseSchema = z3.object({
-  choices: z3.array(
-    z3.object({
-      text: z3.string(),
-      reasoning: z3.string().nullish().optional(),
-      finish_reason: z3.string(),
-      logprobs: z3.object({
-        tokens: z3.array(z3.string()),
-        token_logprobs: z3.array(z3.number()),
-        top_logprobs: z3.array(z3.record(z3.string(), z3.number())).nullable()
-      }).nullable().optional()
-    })
-  ),
-  usage: z3.object({
-    prompt_tokens: z3.number(),
-    completion_tokens: z3.number()
-  })
-});
-var openrouterCompletionChunkSchema = z3.union([
+var OpenRouterCompletionChunkSchema = z3.union([
   z3.object({
+    id: z3.string().optional(),
+    model: z3.string().optional(),
     choices: z3.array(
       z3.object({
         text: z3.string(),
+        reasoning: z3.string().nullish().optional(),
         finish_reason: z3.string().nullish(),
         index: z3.number(),
         logprobs: z3.object({
@@ -1063,7 +1066,7 @@ var openrouterCompletionChunkSchema = z3.union([
       completion_tokens: z3.number()
     }).optional().nullable()
   }),
-  openAIErrorDataSchema
+  OpenRouterErrorResponseSchema
 ]);
 
 // src/openrouter-facade.ts