@ax-llm/ax 10.0.45 → 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,36 +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;
-    const expiry = this.getExpiry(tokenResponse);
-    const tenMinutes = 10 * 60 * 1e3;
-    this.tokenExpiry = expiry - tenMinutes;
-    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);
-      }
+    const response = await client.getAccessToken();
+    if (!response.token) {
+      throw new Error("Failed to obtain access token");
     }
-    return expiry;
+    return response.token;
   }
 };