@openrouter/ai-sdk-provider 0.4.1 → 0.4.3

package/dist/index.mjs

@@ -51,39 +51,61 @@ import { z as z2 } from "zod";
 // src/convert-to-openrouter-chat-messages.ts
 import { convertUint8ArrayToBase64 } from "@ai-sdk/provider-utils";
 function convertToOpenRouterChatMessages(prompt) {
-  var _a;
+  var _a, _b, _c;
   const messages = [];
-  for (const { role, content } of prompt) {
+  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": {
-        messages.push({ role: "system", content });
+        messages.push({
+          role: "system",
+          content,
+          cache_control: getCacheControl(providerMetadata)
+        });
         break;
       }
       case "user": {
         if (content.length === 1 && ((_a = content[0]) == null ? void 0 : _a.type) === "text") {
-          messages.push({ role: "user", content: content[0].text });
+          messages.push({
+            role: "user",
+            content: content[0].text,
+            cache_control: (_b = getCacheControl(providerMetadata)) != null ? _b : getCacheControl(content[0].providerMetadata)
+          });
           break;
         }
+        const messageCacheControl = getCacheControl(providerMetadata);
         const contentParts = content.map(
           (part) => {
-            var _a2;
+            var _a2, _b2, _c2;
             switch (part.type) {
               case "text":
                 return {
                   type: "text",
-                  text: part.text
+                  text: part.text,
+                  // For text parts, only use part-specific cache control
+                  cache_control: getCacheControl(part.providerMetadata)
                 };
               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(part.image)}`
-                  }
+                    url: part.image instanceof URL ? part.image.toString() : `data:${(_a2 = part.mimeType) != null ? _a2 : "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
                 };
               case "file":
                 return {
                   type: "text",
-                  text: part.data instanceof URL ? part.data.toString() : part.data
+                  text: part.data instanceof URL ? part.data.toString() : part.data,
+                  cache_control: (_c2 = getCacheControl(part.providerMetadata)) != null ? _c2 : messageCacheControl
                 };
               default: {
                 const _exhaustiveCheck = part;
@@ -133,7 +155,8 @@ function convertToOpenRouterChatMessages(prompt) {
         messages.push({
           role: "assistant",
           content: text,
-          tool_calls: toolCalls.length > 0 ? toolCalls : void 0
+          tool_calls: toolCalls.length > 0 ? toolCalls : void 0,
+          cache_control: getCacheControl(providerMetadata)
         });
         break;
       }
@@ -142,7 +165,8 @@ function convertToOpenRouterChatMessages(prompt) {
           messages.push({
             role: "tool",
             tool_call_id: toolResponse.toolCallId,
-            content: JSON.stringify(toolResponse.result)
+            content: JSON.stringify(toolResponse.result),
+            cache_control: (_c = getCacheControl(providerMetadata)) != null ? _c : getCacheControl(toolResponse.providerMetadata)
           });
         }
         break;
@@ -189,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(),
@@ -198,7 +222,7 @@ var openAIErrorDataSchema = z.object({
   })
 });
 var openrouterFailedResponseHandler = createJsonErrorResponseHandler({
-  errorSchema: openAIErrorDataSchema,
+  errorSchema: OpenRouterErrorResponseSchema,
   errorToMessage: (data) => data.error.message
 });
 
@@ -294,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({
@@ -305,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) => {
@@ -329,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 },
@@ -353,7 +381,7 @@ var OpenRouterChatLanguageModel = class {
       }),
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
-        openrouterChatChunkSchema
+        OpenRouterStreamChatCompletionChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
@@ -388,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,
@@ -515,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({
@@ -550,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({
@@ -594,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;
@@ -886,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({
@@ -896,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),
@@ -935,7 +977,7 @@ var OpenRouterCompletionLanguageModel = class {
       }),
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler2(
-        openrouterCompletionChunkSchema
+        OpenRouterCompletionChunkSchema
       ),
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
@@ -1002,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({
@@ -1039,7 +1066,7 @@ var openrouterCompletionChunkSchema = z3.union([
       completion_tokens: z3.number()
     }).optional().nullable()
   }),
-  openAIErrorDataSchema
+  OpenRouterErrorResponseSchema
 ]);
 
 // src/openrouter-facade.ts