Add server stats command

SciPhi-AI · Aug 1, 2024 · e430da8 · e430da8
1 parent 40b33c1
commit e430da8
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/__tests__/r2rClientIntegrationSuperUser.test.ts b/__tests__/r2rClientIntegrationSuperUser.test.ts
@@ -21,6 +21,10 @@ describe("r2rClient Integration Tests", () => {
     ).resolves.not.toThrow();
   });
 
+  test("Server stats", async () => {
+    await expect(client.serverStats()).resolves.not.toThrow();
+  });
+
   test("Ingest file", async () => {
     const files = [
       { path: "examples/data/raskolnikov.txt", name: "raskolnikov.txt" },

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "r2r-js",
-  "version": "1.2.13",
+  "version": "1.2.14",
   "description": "",
   "main": "dist/index.js",
   "browser": "dist/index.browser.js",

diff --git a/src/r2rClient.ts b/src/r2rClient.ts
@@ -240,6 +240,11 @@ export class r2rClient {
     return await this._makeRequest("GET", "health");
   }
 
+  async serverStats(): Promise<any> {
+    this._ensureAuthenticated();
+    return await this._makeRequest("GET", "server_stats");
+  }
+
   @feature("updatePrompt")
   async updatePrompt(
     name: string = "default_system",