@openrouter/ai-sdk-provider 0.4.6 → 0.5.0

package/dist/index.mjs

@@ -262,7 +262,7 @@ var OpenRouterChatLanguageModel = class {
   }) {
     var _a;
     const type = mode.type;
-    const extraCallingBody = (_a = providerMetadata == null ? void 0 : providerMetadata["openrouter"]) != null ? _a : {};
+    const extraCallingBody = (_a = providerMetadata == null ? void 0 : providerMetadata.openrouter) != null ? _a : {};
     const baseArgs = __spreadValues(__spreadValues(__spreadValues({
       // model id:
       model: this.modelId,
@@ -287,7 +287,8 @@ var OpenRouterChatLanguageModel = class {
       messages: convertToOpenRouterChatMessages(prompt),
       // OpenRouter specific settings:
       include_reasoning: this.settings.includeReasoning,
-      reasoning: this.settings.reasoning
+      reasoning: this.settings.reasoning,
+      usage: this.settings.usage
     }, this.config.extraBody), this.settings.extraBody), extraCallingBody);
     switch (type) {
       case "regular": {
@@ -323,7 +324,7 @@ var OpenRouterChatLanguageModel = class {
     }
   }
   async doGenerate(options) {
-    var _b, _c, _d, _e, _f, _g, _h;
+    var _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const args = this.getArgs(options);
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.config.url({
@@ -344,14 +345,39 @@ var OpenRouterChatLanguageModel = class {
     if (!choice) {
       throw new Error("No choice in response");
     }
-    return {
+    const usageInfo = response.usage ? {
+      promptTokens: (_b = response.usage.prompt_tokens) != null ? _b : 0,
+      completionTokens: (_c = response.usage.completion_tokens) != null ? _c : 0
+    } : {
+      promptTokens: 0,
+      completionTokens: 0
+    };
+    const providerMetadata = {};
+    if (response.usage && ((_d = this.settings.usage) == null ? void 0 : _d.include)) {
+      providerMetadata.openrouter = {
+        usage: {
+          promptTokens: response.usage.prompt_tokens,
+          promptTokensDetails: response.usage.prompt_tokens_details ? {
+            cachedTokens: (_e = response.usage.prompt_tokens_details.cached_tokens) != null ? _e : 0
+          } : void 0,
+          completionTokens: response.usage.completion_tokens,
+          completionTokensDetails: response.usage.completion_tokens_details ? {
+            reasoningTokens: (_f = response.usage.completion_tokens_details.reasoning_tokens) != null ? _f : 0
+          } : void 0,
+          cost: response.usage.cost,
+          totalTokens: (_g = response.usage.total_tokens) != null ? _g : 0
+        }
+      };
+    }
+    const hasProviderMetadata = Object.keys(providerMetadata).length > 0;
+    return __spreadValues({
       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) => {
+      text: (_h = choice.message.content) != null ? _h : void 0,
+      reasoning: (_i = choice.message.reasoning) != null ? _i : void 0,
+      toolCalls: (_j = choice.message.tool_calls) == null ? void 0 : _j.map((toolCall) => {
         var _a2;
         return {
           toolCallType: "function",
@@ -361,17 +387,15 @@ var OpenRouterChatLanguageModel = class {
         };
       }),
       finishReason: mapOpenRouterFinishReason(choice.finish_reason),
-      usage: {
-        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
-      },
+      usage: usageInfo,
       rawCall: { rawPrompt, rawSettings },
       rawResponse: { headers: responseHeaders },
       warnings: [],
       logprobs: mapOpenRouterChatLogProbsOutput(choice.logprobs)
-    };
+    }, hasProviderMetadata ? { providerMetadata } : {});
   }
   async doStream(options) {
+    var _a, _c;
     const args = this.getArgs(options);
     const { responseHeaders, value: response } = await postJsonToApi({
       url: this.config.url({
@@ -382,7 +406,9 @@ var OpenRouterChatLanguageModel = class {
       body: __spreadProps(__spreadValues({}, args), {
         stream: true,
         // only include stream_options when in strict compatibility mode:
-        stream_options: this.config.compatibility === "strict" ? { include_usage: true } : void 0
+        stream_options: this.config.compatibility === "strict" ? __spreadValues({
+          include_usage: true
+        }, ((_a = this.settings.usage) == null ? void 0 : _a.include) ? { include_usage: true } : {}) : void 0
       }),
       failedResponseHandler: openrouterFailedResponseHandler,
       successfulResponseHandler: createEventSourceResponseHandler(
@@ -391,7 +417,7 @@ var OpenRouterChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
-    const _a = args, { messages: rawPrompt } = _a, rawSettings = __objRest(_a, ["messages"]);
+    const _b = args, { messages: rawPrompt } = _b, rawSettings = __objRest(_b, ["messages"]);
     const toolCalls = [];
     let finishReason = "other";
     let usage = {
@@ -399,11 +425,13 @@ var OpenRouterChatLanguageModel = class {
       completionTokens: Number.NaN
     };
     let logprobs;
+    const openrouterUsage = {};
+    const shouldIncludeUsageAccounting = !!((_c = this.settings.usage) == null ? void 0 : _c.include);
     return {
       stream: response.pipeThrough(
         new TransformStream({
           transform(chunk, controller) {
-            var _a2, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l;
+            var _a2, _b2, _c2, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
             if (!chunk.success) {
               finishReason = "error";
               controller.enqueue({ type: "error", error: chunk.error });
@@ -432,6 +460,20 @@ var OpenRouterChatLanguageModel = class {
                 promptTokens: value.usage.prompt_tokens,
                 completionTokens: value.usage.completion_tokens
               };
+              openrouterUsage.promptTokens = value.usage.prompt_tokens;
+              if (value.usage.prompt_tokens_details) {
+                openrouterUsage.promptTokensDetails = {
+                  cachedTokens: (_a2 = value.usage.prompt_tokens_details.cached_tokens) != null ? _a2 : 0
+                };
+              }
+              openrouterUsage.completionTokens = value.usage.completion_tokens;
+              if (value.usage.completion_tokens_details) {
+                openrouterUsage.completionTokensDetails = {
+                  reasoningTokens: (_b2 = value.usage.completion_tokens_details.reasoning_tokens) != null ? _b2 : 0
+                };
+              }
+              openrouterUsage.cost = value.usage.cost;
+              openrouterUsage.totalTokens = value.usage.total_tokens;
             }
             const choice = value.choices[0];
             if ((choice == null ? void 0 : choice.finish_reason) != null) {
@@ -476,7 +518,7 @@ var OpenRouterChatLanguageModel = class {
                       message: `Expected 'id' to be a string.`
                     });
                   }
-                  if (((_a2 = toolCallDelta.function) == null ? void 0 : _a2.name) == null) {
+                  if (((_c2 = toolCallDelta.function) == null ? void 0 : _c2.name) == null) {
                     throw new InvalidResponseDataError({
                       data: toolCallDelta,
                       message: `Expected 'function.name' to be a string.`
@@ -487,7 +529,7 @@ var OpenRouterChatLanguageModel = class {
                     type: "function",
                     function: {
                       name: toolCallDelta.function.name,
-                      arguments: (_b = toolCallDelta.function.arguments) != null ? _b : ""
+                      arguments: (_d = toolCallDelta.function.arguments) != null ? _d : ""
                     },
                     sent: false
                   };
@@ -495,7 +537,7 @@ var OpenRouterChatLanguageModel = class {
                   if (toolCall2 == null) {
                     throw new Error("Tool call is missing");
                   }
-                  if (((_c = toolCall2.function) == null ? void 0 : _c.name) != null && ((_d = toolCall2.function) == null ? void 0 : _d.arguments) != null && isParsableJson(toolCall2.function.arguments)) {
+                  if (((_e = toolCall2.function) == null ? void 0 : _e.name) != null && ((_f = toolCall2.function) == null ? void 0 : _f.arguments) != null && isParsableJson(toolCall2.function.arguments)) {
                     controller.enqueue({
                       type: "tool-call-delta",
                       toolCallType: "function",
@@ -506,7 +548,7 @@ var OpenRouterChatLanguageModel = class {
                     controller.enqueue({
                       type: "tool-call",
                       toolCallType: "function",
-                      toolCallId: (_e = toolCall2.id) != null ? _e : generateId(),
+                      toolCallId: (_g = toolCall2.id) != null ? _g : generateId(),
                       toolName: toolCall2.function.name,
                       args: toolCall2.function.arguments
                     });
@@ -518,21 +560,21 @@ var OpenRouterChatLanguageModel = class {
                 if (toolCall == null) {
                   throw new Error("Tool call is missing");
                 }
-                if (((_f = toolCallDelta.function) == null ? void 0 : _f.arguments) != null) {
-                  toolCall.function.arguments += (_h = (_g = toolCallDelta.function) == null ? void 0 : _g.arguments) != null ? _h : "";
+                if (((_h = toolCallDelta.function) == null ? void 0 : _h.arguments) != null) {
+                  toolCall.function.arguments += (_j = (_i = toolCallDelta.function) == null ? void 0 : _i.arguments) != null ? _j : "";
                 }
                 controller.enqueue({
                   type: "tool-call-delta",
                   toolCallType: "function",
                   toolCallId: toolCall.id,
                   toolName: toolCall.function.name,
-                  argsTextDelta: (_i = toolCallDelta.function.arguments) != null ? _i : ""
+                  argsTextDelta: (_k = toolCallDelta.function.arguments) != null ? _k : ""
                 });
-                if (((_j = toolCall.function) == null ? void 0 : _j.name) != null && ((_k = toolCall.function) == null ? void 0 : _k.arguments) != null && isParsableJson(toolCall.function.arguments)) {
+                if (((_l = toolCall.function) == null ? void 0 : _l.name) != null && ((_m = toolCall.function) == null ? void 0 : _m.arguments) != null && isParsableJson(toolCall.function.arguments)) {
                   controller.enqueue({
                     type: "tool-call",
                     toolCallType: "function",
-                    toolCallId: (_l = toolCall.id) != null ? _l : generateId(),
+                    toolCallId: (_n = toolCall.id) != null ? _n : generateId(),
                     toolName: toolCall.function.name,
                     args: toolCall.function.arguments
                   });
@@ -558,12 +600,19 @@ var OpenRouterChatLanguageModel = class {
                 }
               }
             }
-            controller.enqueue({
+            const providerMetadata = {};
+            if (shouldIncludeUsageAccounting && (openrouterUsage.totalTokens !== void 0 || openrouterUsage.cost !== void 0 || openrouterUsage.promptTokensDetails !== void 0 || openrouterUsage.completionTokensDetails !== void 0)) {
+              providerMetadata.openrouter = {
+                usage: openrouterUsage
+              };
+            }
+            const hasProviderMetadata = Object.keys(providerMetadata).length > 0 && shouldIncludeUsageAccounting;
+            controller.enqueue(__spreadValues({
               type: "finish",
               finishReason,
               logprobs,
               usage
-            });
+            }, hasProviderMetadata ? { providerMetadata } : {}));
           }
         })
       ),
@@ -578,8 +627,15 @@ var OpenRouterChatCompletionBaseResponseSchema = z2.object({
   model: z2.string().optional(),
   usage: z2.object({
     prompt_tokens: z2.number(),
+    prompt_tokens_details: z2.object({
+      cached_tokens: z2.number()
+    }).optional(),
     completion_tokens: z2.number(),
-    total_tokens: z2.number()
+    completion_tokens_details: z2.object({
+      reasoning_tokens: z2.number()
+    }).optional(),
+    total_tokens: z2.number(),
+    cost: z2.number().optional()
   }).nullish()
 });
 var OpenRouterNonStreamChatCompletionResponseSchema = OpenRouterChatCompletionBaseResponseSchema.extend({
@@ -676,14 +732,13 @@ function prepareToolsAndToolChoice(mode) {
           parameters: tool.parameters
         }
       };
-    } else {
-      return {
-        type: "function",
-        function: {
-          name: tool.name
-        }
-      };
     }
+    return {
+      type: "function",
+      function: {
+        name: tool.name
+      }
+    };
   });
   const toolChoice = mode.toolChoice;
   if (toolChoice == null) {
@@ -880,7 +935,7 @@ var OpenRouterCompletionLanguageModel = class {
   }) {
     var _a, _b;
     const type = mode.type;
-    const extraCallingBody = (_a = providerMetadata == null ? void 0 : providerMetadata["openrouter"]) != null ? _a : {};
+    const extraCallingBody = (_a = providerMetadata == null ? void 0 : providerMetadata.openrouter) != null ? _a : {};
     const { prompt: completionPrompt } = convertToOpenRouterCompletionPrompt({
       prompt,
       inputFormat
@@ -1177,9 +1232,7 @@ function createOpenRouter(options = {}) {
     }
     return createChatModel(modelId, settings);
   };
-  const provider = function(modelId, settings) {
-    return createLanguageModel(modelId, settings);
-  };
+  const provider = (modelId, settings) => createLanguageModel(modelId, settings);
   provider.languageModel = createLanguageModel;
   provider.chat = createChatModel;
   provider.completion = createCompletionModel;