From 77c6fd959b374a970231e8cede2f0c4eda916bcb Mon Sep 17 00:00:00 2001
From: Daniel Salazar <daniel.salazar@puter.com>
Date: Sat, 22 Nov 2025 15:51:47 -0800
Subject: [PATCH] fix: openrouter missing usage

---
 Dockerfile                                    |  1 +
 package-lock.json                             | 67 ++++++++++++++++++-
 .../src/modules/puterai/OpenRouterService.js  |  1 +
 3 files changed, 68 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index c5ab1dd2f5..785290c475 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -24,6 +24,7 @@ COPY package*.json ./
 COPY . .
 
 # Install mocha
+RUN npm i -g npm@latest
 RUN npm install -g mocha
 
 # Install node modules
diff --git a/package-lock.json b/package-lock.json
index e6e84a05c6..d6f62c3584 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -34,6 +34,7 @@
       },
       "devDependencies": {
         "@eslint/js": "^9.35.0",
+        "@playwright/test": "^1.56.1",
         "@stylistic/eslint-plugin": "^5.3.1",
         "@types/uuid": "^10.0.0",
         "@typescript-eslint/eslint-plugin": "^8.46.1",
@@ -57,7 +58,8 @@
         "vite-plugin-static-copy": "^3.1.3",
         "vitest": "^3.2.4",
         "webpack": "^5.88.2",
-        "webpack-cli": "^5.1.1"
+        "webpack-cli": "^5.1.1",
+        "yaml": "^2.8.1"
       },
       "engines": {
         "node": ">=20.19.5"
@@ -5645,6 +5647,22 @@
         "node": ">=14"
       }
     },
+    "node_modules/@playwright/test": {
+      "version": "1.56.1",
+      "resolved": "https://registry.npmjs.org/@playwright/test/-/test-1.56.1.tgz",
+      "integrity": "sha512-vSMYtL/zOcFpvJCW71Q/OEGQb7KYBPAdKh35WNSkaZA75JlAO8ED8UN6GUNTm3drWomcbcqRPFqQbLae8yBTdg==",
+      "dev": true,
+      "license": "Apache-2.0",
+      "dependencies": {
+        "playwright": "1.56.1"
+      },
+      "bin": {
+        "playwright": "cli.js"
+      },
+      "engines": {
+        "node": ">=18"
+      }
+    },
     "node_modules/@protobufjs/aspromise": {
       "version": "1.1.2",
       "resolved": "https://registry.npmjs.org/@protobufjs/aspromise/-/aspromise-1.1.2.tgz",
@@ -15377,6 +15395,53 @@
         "node": ">=8"
       }
     },
+    "node_modules/playwright": {
+      "version": "1.56.1",
+      "resolved": "https://registry.npmjs.org/playwright/-/playwright-1.56.1.tgz",
+      "integrity": "sha512-aFi5B0WovBHTEvpM3DzXTUaeN6eN0qWnTkKx4NQaH4Wvcmc153PdaY2UBdSYKaGYw+UyWXSVyxDUg5DoPEttjw==",
+      "dev": true,
+      "license": "Apache-2.0",
+      "dependencies": {
+        "playwright-core": "1.56.1"
+      },
+      "bin": {
+        "playwright": "cli.js"
+      },
+      "engines": {
+        "node": ">=18"
+      },
+      "optionalDependencies": {
+        "fsevents": "2.3.2"
+      }
+    },
+    "node_modules/playwright-core": {
+      "version": "1.56.1",
+      "resolved": "https://registry.npmjs.org/playwright-core/-/playwright-core-1.56.1.tgz",
+      "integrity": "sha512-hutraynyn31F+Bifme+Ps9Vq59hKuUCz7H1kDOcBs+2oGguKkWTU50bBWrtz34OUWmIwpBTWDxaRPXrIXkgvmQ==",
+      "dev": true,
+      "license": "Apache-2.0",
+      "bin": {
+        "playwright-core": "cli.js"
+      },
+      "engines": {
+        "node": ">=18"
+      }
+    },
+    "node_modules/playwright/node_modules/fsevents": {
+      "version": "2.3.2",
+      "resolved": "https://registry.npmjs.org/fsevents/-/fsevents-2.3.2.tgz",
+      "integrity": "sha512-xiqMQR4xAeHTuB9uWm+fFRcIOgKBMiOBP+eXiyT7jsgVCq1bkVygt00oASowB7EdtpOHaaPgKt812P9ab+DDKA==",
+      "dev": true,
+      "hasInstallScript": true,
+      "license": "MIT",
+      "optional": true,
+      "os": [
+        "darwin"
+      ],
+      "engines": {
+        "node": "^8.16.0 || ^10.6.0 || >=11.0.0"
+      }
+    },
     "node_modules/pngjs": {
       "version": "6.0.0",
       "resolved": "https://registry.npmjs.org/pngjs/-/pngjs-6.0.0.tgz",
diff --git a/src/backend/src/modules/puterai/OpenRouterService.js b/src/backend/src/modules/puterai/OpenRouterService.js
index 2128e15b6f..9db62cbc0f 100644
--- a/src/backend/src/modules/puterai/OpenRouterService.js
+++ b/src/backend/src/modules/puterai/OpenRouterService.js
@@ -138,6 +138,7 @@ class OpenRouterService extends BaseService {
             ...(stream ? {
                 stream_options: { include_usage: true },
             } : {}),
+            usage: { include: true },
         });
 
         const modelDetails =  (await this.models()).find(m => m.id === `openrouter:${ model}`);