@ax-llm/ax 10.0.44 → 10.0.46

package/index.js

@@ -1057,8 +1057,6 @@ import { GoogleAuth } from "google-auth-library";
 var GoogleVertexAuth = class {
   auth;
   client;
-  currentToken;
-  tokenExpiry;
   constructor(config = {}) {
     this.auth = new GoogleAuth({
       scopes: ["https://www.googleapis.com/auth/cloud-platform"],
@@ -1072,38 +1070,12 @@ var GoogleVertexAuth = class {
     return this.client;
   }
   async getAccessToken() {
-    if (this.currentToken && this.tokenExpiry && Date.now() < this.tokenExpiry) {
-      return this.currentToken;
-    }
     const client = await this.getAuthenticatedClient();
-    const tokenResponse = await client.getAccessToken();
-    this.currentToken = tokenResponse.token ?? void 0;
-    const expiry = this.getExpiry(tokenResponse);
-    const fiveMinutes = 5 * 60 * 1e3;
-    this.tokenExpiry = expiry - fiveMinutes;
-    return this.currentToken;
-  }
-  /**
-   * Get the expiry date from the token response.
-   */
-  getExpiry(tokenResponse) {
-    const oneHour = 3600 * 1e3;
-    let expiry = Date.now() + oneHour;
-    let responseExpiry = tokenResponse.res?.data?.expiry_date;
-    if (responseExpiry) {
-      if (typeof responseExpiry === "number") {
-        expiry = responseExpiry;
-      } else if (responseExpiry instanceof Date) {
-        expiry = responseExpiry.getTime();
-      } else if (typeof responseExpiry === "string") {
-        expiry = new Date(responseExpiry).getTime();
-      } else {
-        console.warn("Unknown expiry type", responseExpiry);
-      }
-    } else {
-      console.warn("No expiry date found in response", tokenResponse.res?.data);
+    const response = await client.getAccessToken();
+    if (!response.token) {
+      throw new Error("Failed to obtain access token");
     }
-    return expiry;
+    return response.token;
   }
 };
 
@@ -6084,11 +6056,16 @@ var AxBalancer = class _AxBalancer {
           default:
             throw e;
         }
-        console.warn(`Service ${this.currentService.getName()} failed`);
+        console.warn(
+          `AxBalancer: Service ${this.currentService.getName()} failed`,
+          e
+        );
         if (!this.getNextService()) {
           throw e;
         }
-        console.warn(`Switching to service ${this.currentService.getName()}`);
+        console.warn(
+          `AxBalancer: Switching to service ${this.currentService.getName()}`
+        );
       }
     }
   }