langchain 0.0.146 → 0.0.148

package/dist/util/async_caller.js

@@ -12,6 +12,32 @@ const STATUS_NO_RETRY = [
     408,
     409, // Conflict
 ];
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+const defaultFailedAttemptHandler = (error) => {
+    if (error.message.startsWith("Cancel") ||
+        error.message.startsWith("TimeoutError") ||
+        error.name === "TimeoutError" ||
+        error.message.startsWith("AbortError") ||
+        error.name === "AbortError") {
+        throw error;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    if (error?.code === "ECONNABORTED") {
+        throw error;
+    }
+    const status = 
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    error?.response?.status ?? error?.status;
+    if (status && STATUS_NO_RETRY.includes(+status)) {
+        throw error;
+    }
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    if (error?.error?.code === "insufficient_quota") {
+        const err = new Error(error?.message);
+        err.name = "InsufficientQuotaError";
+        throw err;
+    }
+};
 /**
  * A class that can be used to make async calls with concurrency and retry logic.
  *
@@ -39,6 +65,12 @@ export class AsyncCaller {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "onFailedAttempt", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "queue", {
             enumerable: true,
             configurable: true,
@@ -47,6 +79,8 @@ export class AsyncCaller {
         });
         this.maxConcurrency = params.maxConcurrency ?? Infinity;
         this.maxRetries = params.maxRetries ?? 6;
+        this.onFailedAttempt =
+            params.onFailedAttempt ?? defaultFailedAttemptHandler;
         const PQueue = "default" in PQueueMod ? PQueueMod.default : PQueueMod;
         this.queue = new PQueue({ concurrency: this.maxConcurrency });
     }
@@ -61,31 +95,7 @@ export class AsyncCaller {
                 throw new Error(error);
             }
         }), {
-            onFailedAttempt(error) {
-                if (error.message.startsWith("Cancel") ||
-                    error.message.startsWith("TimeoutError") ||
-                    error.name === "TimeoutError" ||
-                    error.message.startsWith("AbortError") ||
-                    error.name === "AbortError") {
-                    throw error;
-                }
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                if (error?.code === "ECONNABORTED") {
-                    throw error;
-                }
-                const status = 
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                error?.response?.status ?? error?.status;
-                if (status && STATUS_NO_RETRY.includes(+status)) {
-                    throw error;
-                }
-                // eslint-disable-next-line @typescript-eslint/no-explicit-any
-                if (error?.error?.code === "insufficient_quota") {
-                    const err = new Error(error?.message);
-                    err.name = "InsufficientQuotaError";
-                    throw err;
-                }
-            },
+            onFailedAttempt: this.onFailedAttempt,
             retries: this.maxRetries,
             randomize: true,
             // If needed we can change some of the defaults here,

package/dist/util/googlevertexai-connection.cjs

@@ -1,9 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.GoogleVertexAILLMConnection = exports.GoogleVertexAIConnection = void 0;
-const google_auth_library_1 = require("google-auth-library");
 class GoogleVertexAIConnection {
-    constructor(fields, caller) {
+    constructor(fields, caller, client) {
         Object.defineProperty(this, "caller", {
             enumerable: true,
             configurable: true,
@@ -28,7 +27,7 @@ class GoogleVertexAIConnection {
             writable: true,
             value: "v1"
         });
-        Object.defineProperty(this, "auth", {
+        Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -38,16 +37,12 @@ class GoogleVertexAIConnection {
         this.endpoint = fields?.endpoint ?? this.endpoint;
         this.location = fields?.location ?? this.location;
         this.apiVersion = fields?.apiVersion ?? this.apiVersion;
-        this.auth = new google_auth_library_1.GoogleAuth({
-            scopes: "https://www.googleapis.com/auth/cloud-platform",
-            ...fields?.authOptions,
-        });
+        this.client = client;
     }
     buildMethod() {
         return "POST";
     }
     async _request(data, options) {
-        const client = await this.auth.getClient();
         const url = await this.buildUrl();
         const method = this.buildMethod();
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -58,11 +53,8 @@ class GoogleVertexAIConnection {
         if (data && method === "POST") {
             opts.data = data;
         }
-        async function _request() {
-            return client.request(opts);
-        }
         try {
-            const callResponse = await this.caller.callWithOptions({ signal: options?.signal }, _request.bind(client));
+            const callResponse = await this.caller.callWithOptions({ signal: options?.signal }, async () => this.client.request(opts));
             const response = callResponse; // Done for typecast safety, I guess
             return response;
         }
@@ -74,18 +66,25 @@ class GoogleVertexAIConnection {
 }
 exports.GoogleVertexAIConnection = GoogleVertexAIConnection;
 class GoogleVertexAILLMConnection extends GoogleVertexAIConnection {
-    constructor(fields, caller) {
-        super(fields, caller);
+    constructor(fields, caller, client) {
+        super(fields, caller, client);
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = client;
         this.model = fields?.model ?? this.model;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/v1/projects/${projectId}/locations/${this.location}/publishers/google/models/${this.model}:predict`;
         return url;
     }

package/dist/util/googlevertexai-connection.d.ts

@@ -1,21 +1,21 @@
-import { GoogleAuth } from "google-auth-library";
 import { BaseLanguageModelCallOptions } from "../base_language/index.js";
 import { AsyncCaller, AsyncCallerCallOptions } from "./async_caller.js";
-import { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAIConnectionParams, GoogleVertexAILLMResponse, GoogleVertexAIModelParams, GoogleVertexAIResponse } from "../types/googlevertexai-types.js";
-export declare abstract class GoogleVertexAIConnection<CallOptions extends AsyncCallerCallOptions, ResponseType extends GoogleVertexAIResponse> implements GoogleVertexAIConnectionParams {
+import type { GoogleVertexAIBaseLLMInput, GoogleVertexAIBasePrediction, GoogleVertexAIConnectionParams, GoogleVertexAILLMResponse, GoogleVertexAIModelParams, GoogleVertexAIResponse, GoogleVertexAIAbstractedClient } from "../types/googlevertexai-types.js";
+export declare abstract class GoogleVertexAIConnection<CallOptions extends AsyncCallerCallOptions, ResponseType extends GoogleVertexAIResponse, AuthOptions> implements GoogleVertexAIConnectionParams<AuthOptions> {
     caller: AsyncCaller;
     endpoint: string;
     location: string;
     apiVersion: string;
-    auth: GoogleAuth;
-    constructor(fields: GoogleVertexAIConnectionParams | undefined, caller: AsyncCaller);
+    client: GoogleVertexAIAbstractedClient;
+    constructor(fields: GoogleVertexAIConnectionParams<AuthOptions> | undefined, caller: AsyncCaller, client: GoogleVertexAIAbstractedClient);
     abstract buildUrl(): Promise<string>;
     buildMethod(): string;
     _request(data: unknown | undefined, options: CallOptions): Promise<ResponseType>;
 }
-export declare class GoogleVertexAILLMConnection<CallOptions extends BaseLanguageModelCallOptions, InstanceType, PredictionType extends GoogleVertexAIBasePrediction> extends GoogleVertexAIConnection<CallOptions, PredictionType> implements GoogleVertexAIBaseLLMInput {
+export declare class GoogleVertexAILLMConnection<CallOptions extends BaseLanguageModelCallOptions, InstanceType, PredictionType extends GoogleVertexAIBasePrediction, AuthOptions> extends GoogleVertexAIConnection<CallOptions, PredictionType, AuthOptions> implements GoogleVertexAIBaseLLMInput<AuthOptions> {
     model: string;
-    constructor(fields: GoogleVertexAIBaseLLMInput | undefined, caller: AsyncCaller);
+    client: GoogleVertexAIAbstractedClient;
+    constructor(fields: GoogleVertexAIBaseLLMInput<AuthOptions> | undefined, caller: AsyncCaller, client: GoogleVertexAIAbstractedClient);
     buildUrl(): Promise<string>;
     request(instances: InstanceType[], parameters: GoogleVertexAIModelParams, options: CallOptions): Promise<GoogleVertexAILLMResponse<PredictionType>>;
 }

package/dist/util/googlevertexai-connection.js

@@ -1,6 +1,5 @@
-import { GoogleAuth } from "google-auth-library";
 export class GoogleVertexAIConnection {
-    constructor(fields, caller) {
+    constructor(fields, caller, client) {
         Object.defineProperty(this, "caller", {
             enumerable: true,
             configurable: true,
@@ -25,7 +24,7 @@ export class GoogleVertexAIConnection {
             writable: true,
             value: "v1"
         });
-        Object.defineProperty(this, "auth", {
+        Object.defineProperty(this, "client", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -35,16 +34,12 @@ export class GoogleVertexAIConnection {
         this.endpoint = fields?.endpoint ?? this.endpoint;
         this.location = fields?.location ?? this.location;
         this.apiVersion = fields?.apiVersion ?? this.apiVersion;
-        this.auth = new GoogleAuth({
-            scopes: "https://www.googleapis.com/auth/cloud-platform",
-            ...fields?.authOptions,
-        });
+        this.client = client;
     }
     buildMethod() {
         return "POST";
     }
     async _request(data, options) {
-        const client = await this.auth.getClient();
         const url = await this.buildUrl();
         const method = this.buildMethod();
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -55,11 +50,8 @@ export class GoogleVertexAIConnection {
         if (data && method === "POST") {
             opts.data = data;
         }
-        async function _request() {
-            return client.request(opts);
-        }
         try {
-            const callResponse = await this.caller.callWithOptions({ signal: options?.signal }, _request.bind(client));
+            const callResponse = await this.caller.callWithOptions({ signal: options?.signal }, async () => this.client.request(opts));
             const response = callResponse; // Done for typecast safety, I guess
             return response;
         }
@@ -70,18 +62,25 @@ export class GoogleVertexAIConnection {
     }
 }
 export class GoogleVertexAILLMConnection extends GoogleVertexAIConnection {
-    constructor(fields, caller) {
-        super(fields, caller);
+    constructor(fields, caller, client) {
+        super(fields, caller, client);
         Object.defineProperty(this, "model", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "client", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.client = client;
         this.model = fields?.model ?? this.model;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/v1/projects/${projectId}/locations/${this.location}/publishers/google/models/${this.model}:predict`;
         return url;
     }

package/dist/util/googlevertexai-webauth.cjs

@@ -0,0 +1,56 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WebGoogleAuth = void 0;
+const google_1 = require("web-auth-library/google");
+const env_js_1 = require("./env.cjs");
+class WebGoogleAuth {
+    constructor(options) {
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const credentials = options?.credentials ??
+            (0, env_js_1.getEnvironmentVariable)("GOOGLE_VERTEX_AI_WEB_CREDENTIALS");
+        if (credentials === undefined)
+            throw new Error(`Credentials not found. Please set the GOOGLE_VERTEX_AI_WEB_CREDENTIALS or pass credentials into "authOptions.credentials".`);
+        const scope = options?.scope ?? "https://www.googleapis.com/auth/cloud-platform";
+        this.options = { ...options, credentials, scope };
+    }
+    async getProjectId() {
+        const credentials = (0, google_1.getCredentials)(this.options.credentials);
+        return credentials.project_id;
+    }
+    async request(opts) {
+        const accessToken = await (0, google_1.getAccessToken)(this.options);
+        if (opts.url == null)
+            throw new Error("Missing URL");
+        const fetchOptions = {
+            method: opts.method,
+            headers: {
+                Authorization: `Bearer ${accessToken}`,
+                "Content-Type": "application/json",
+            },
+        };
+        if (opts.data !== undefined) {
+            fetchOptions.body = JSON.stringify(opts.data);
+        }
+        const res = await fetch(opts.url, fetchOptions);
+        if (!res.ok) {
+            const error = new Error(`Could not get access token for Vertex AI with status code: ${res.status}`);
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            error.response = res;
+            throw error;
+        }
+        return {
+            data: await res.json(),
+            config: {},
+            status: res.status,
+            statusText: res.statusText,
+            headers: res.headers,
+            request: { responseURL: res.url },
+        };
+    }
+}
+exports.WebGoogleAuth = WebGoogleAuth;

package/dist/util/googlevertexai-webauth.d.ts

@@ -0,0 +1,25 @@
+import { Credentials } from "web-auth-library/google";
+import type { GoogleVertexAIAbstractedClient } from "../types/googlevertexai-types.js";
+export type WebGoogleAuthOptions = {
+    credentials: string | Credentials;
+    scope?: string | string[];
+};
+export declare class WebGoogleAuth implements GoogleVertexAIAbstractedClient {
+    options: WebGoogleAuthOptions;
+    constructor(options?: WebGoogleAuthOptions);
+    getProjectId(): Promise<string>;
+    request(opts: {
+        url?: string;
+        method?: string;
+        data?: unknown;
+    }): Promise<{
+        data: any;
+        config: {};
+        status: number;
+        statusText: string;
+        headers: Headers;
+        request: {
+            responseURL: string;
+        };
+    }>;
+}

package/dist/util/googlevertexai-webauth.js

@@ -0,0 +1,52 @@
+import { getAccessToken, getCredentials, } from "web-auth-library/google";
+import { getEnvironmentVariable } from "./env.js";
+export class WebGoogleAuth {
+    constructor(options) {
+        Object.defineProperty(this, "options", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const credentials = options?.credentials ??
+            getEnvironmentVariable("GOOGLE_VERTEX_AI_WEB_CREDENTIALS");
+        if (credentials === undefined)
+            throw new Error(`Credentials not found. Please set the GOOGLE_VERTEX_AI_WEB_CREDENTIALS or pass credentials into "authOptions.credentials".`);
+        const scope = options?.scope ?? "https://www.googleapis.com/auth/cloud-platform";
+        this.options = { ...options, credentials, scope };
+    }
+    async getProjectId() {
+        const credentials = getCredentials(this.options.credentials);
+        return credentials.project_id;
+    }
+    async request(opts) {
+        const accessToken = await getAccessToken(this.options);
+        if (opts.url == null)
+            throw new Error("Missing URL");
+        const fetchOptions = {
+            method: opts.method,
+            headers: {
+                Authorization: `Bearer ${accessToken}`,
+                "Content-Type": "application/json",
+            },
+        };
+        if (opts.data !== undefined) {
+            fetchOptions.body = JSON.stringify(opts.data);
+        }
+        const res = await fetch(opts.url, fetchOptions);
+        if (!res.ok) {
+            const error = new Error(`Could not get access token for Vertex AI with status code: ${res.status}`);
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            error.response = res;
+            throw error;
+        }
+        return {
+            data: await res.json(),
+            config: {},
+            status: res.status,
+            statusText: res.statusText,
+            headers: res.headers,
+            request: { responseURL: res.url },
+        };
+    }
+}

package/dist/vectorstores/googlevertexai.cjs

@@ -29,6 +29,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.MatchingEngine = exports.IdDocument = void 0;
 const uuid = __importStar(require("uuid"));
 const flat_1 = __importDefault(require("flat"));
+const google_auth_library_1 = require("google-auth-library");
 const base_js_1 = require("./base.cjs");
 const document_js_1 = require("../document.cjs");
 const googlevertexai_connection_js_1 = require("../util/googlevertexai-connection.cjs");
@@ -51,7 +52,7 @@ class IdDocument extends document_js_1.Document {
 exports.IdDocument = IdDocument;
 class IndexEndpointConnection extends googlevertexai_connection_js_1.GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new google_auth_library_1.GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "indexEndpoint", {
             enumerable: true,
             configurable: true,
@@ -61,7 +62,7 @@ class IndexEndpointConnection extends googlevertexai_connection_js_1.GoogleVerte
         this.indexEndpoint = fields.indexEndpoint;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexEndpoints/${this.indexEndpoint}`;
         return url;
     }
@@ -74,7 +75,7 @@ class IndexEndpointConnection extends googlevertexai_connection_js_1.GoogleVerte
 }
 class RemoveDatapointConnection extends googlevertexai_connection_js_1.GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new google_auth_library_1.GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "index", {
             enumerable: true,
             configurable: true,
@@ -84,7 +85,7 @@ class RemoveDatapointConnection extends googlevertexai_connection_js_1.GoogleVer
         this.index = fields.index;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexes/${this.index}:removeDatapoints`;
         return url;
     }
@@ -100,7 +101,7 @@ class RemoveDatapointConnection extends googlevertexai_connection_js_1.GoogleVer
 }
 class UpsertDatapointConnection extends googlevertexai_connection_js_1.GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new google_auth_library_1.GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "index", {
             enumerable: true,
             configurable: true,
@@ -110,7 +111,7 @@ class UpsertDatapointConnection extends googlevertexai_connection_js_1.GoogleVer
         this.index = fields.index;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexes/${this.index}:upsertDatapoints`;
         return url;
     }
@@ -126,7 +127,7 @@ class UpsertDatapointConnection extends googlevertexai_connection_js_1.GoogleVer
 }
 class FindNeighborsConnection extends googlevertexai_connection_js_1.GoogleVertexAIConnection {
     constructor(params, caller) {
-        super(params, caller);
+        super(params, caller, new google_auth_library_1.GoogleAuth(params.authOptions));
         Object.defineProperty(this, "indexEndpoint", {
             enumerable: true,
             configurable: true,
@@ -143,7 +144,7 @@ class FindNeighborsConnection extends googlevertexai_connection_js_1.GoogleVerte
         this.deployedIndexId = params.deployedIndexId;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexEndpoints/${this.indexEndpoint}:findNeighbors`;
         return url;
     }

package/dist/vectorstores/googlevertexai.d.ts

@@ -1,3 +1,4 @@
+import { GoogleAuthOptions } from "google-auth-library";
 import { VectorStore } from "./base.js";
 import { Embeddings } from "../embeddings/base.js";
 import { Document, DocumentInput } from "../document.js";
@@ -18,7 +19,7 @@ export declare class IdDocument extends Document implements IdDocumentInput {
     id?: string;
     constructor(fields: IdDocumentInput);
 }
-interface IndexEndpointConnectionParams extends GoogleVertexAIConnectionParams {
+interface IndexEndpointConnectionParams extends GoogleVertexAIConnectionParams<GoogleAuthOptions> {
     indexEndpoint: string;
 }
 interface DeployedIndex {
@@ -31,7 +32,7 @@ interface IndexEndpointResponse extends GoogleVertexAIResponse {
         publicEndpointDomainName: string;
     };
 }
-declare class IndexEndpointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, IndexEndpointResponse> {
+declare class IndexEndpointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, IndexEndpointResponse, GoogleAuthOptions> {
     indexEndpoint: string;
     constructor(fields: IndexEndpointConnectionParams, caller: AsyncCaller);
     buildUrl(): Promise<string>;
@@ -45,19 +46,19 @@ declare class IndexEndpointConnection extends GoogleVertexAIConnection<AsyncCall
 export interface MatchingEngineDeleteParams {
     ids: string[];
 }
-interface RemoveDatapointParams extends GoogleVertexAIConnectionParams {
+interface RemoveDatapointParams extends GoogleVertexAIConnectionParams<GoogleAuthOptions> {
     index: string;
 }
 interface RemoveDatapointResponse extends GoogleVertexAIResponse {
 }
-declare class RemoveDatapointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, RemoveDatapointResponse> {
+declare class RemoveDatapointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, RemoveDatapointResponse, GoogleAuthOptions> {
     index: string;
     constructor(fields: RemoveDatapointParams, caller: AsyncCaller);
     buildUrl(): Promise<string>;
     buildMethod(): string;
     request(datapointIds: string[], options: AsyncCallerCallOptions): Promise<RemoveDatapointResponse>;
 }
-interface UpsertDatapointParams extends GoogleVertexAIConnectionParams {
+interface UpsertDatapointParams extends GoogleVertexAIConnectionParams<GoogleAuthOptions> {
     index: string;
 }
 export interface Restriction {
@@ -76,7 +77,7 @@ interface IndexDatapoint {
 }
 interface UpsertDatapointResponse extends GoogleVertexAIResponse {
 }
-declare class UpsertDatapointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, UpsertDatapointResponse> {
+declare class UpsertDatapointConnection extends GoogleVertexAIConnection<AsyncCallerCallOptions, UpsertDatapointResponse, GoogleAuthOptions> {
     index: string;
     constructor(fields: UpsertDatapointParams, caller: AsyncCaller);
     buildUrl(): Promise<string>;
@@ -94,7 +95,7 @@ export interface PublicAPIEndpointInfo {
 /**
  * Parameters necessary to configure the Matching Engine.
  */
-export interface MatchingEngineArgs extends GoogleVertexAIConnectionParams, IndexEndpointConnectionParams, UpsertDatapointParams {
+export interface MatchingEngineArgs extends GoogleVertexAIConnectionParams<GoogleAuthOptions>, IndexEndpointConnectionParams, UpsertDatapointParams {
     docstore: Docstore;
     callerParams?: AsyncCallerParams;
     callerOptions?: AsyncCallerCallOptions;

package/dist/vectorstores/googlevertexai.js

@@ -1,5 +1,6 @@
 import * as uuid from "uuid";
 import flatten from "flat";
+import { GoogleAuth } from "google-auth-library";
 import { VectorStore } from "./base.js";
 import { Document } from "../document.js";
 import { GoogleVertexAIConnection } from "../util/googlevertexai-connection.js";
@@ -21,7 +22,7 @@ export class IdDocument extends Document {
 }
 class IndexEndpointConnection extends GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "indexEndpoint", {
             enumerable: true,
             configurable: true,
@@ -31,7 +32,7 @@ class IndexEndpointConnection extends GoogleVertexAIConnection {
         this.indexEndpoint = fields.indexEndpoint;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexEndpoints/${this.indexEndpoint}`;
         return url;
     }
@@ -44,7 +45,7 @@ class IndexEndpointConnection extends GoogleVertexAIConnection {
 }
 class RemoveDatapointConnection extends GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "index", {
             enumerable: true,
             configurable: true,
@@ -54,7 +55,7 @@ class RemoveDatapointConnection extends GoogleVertexAIConnection {
         this.index = fields.index;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexes/${this.index}:removeDatapoints`;
         return url;
     }
@@ -70,7 +71,7 @@ class RemoveDatapointConnection extends GoogleVertexAIConnection {
 }
 class UpsertDatapointConnection extends GoogleVertexAIConnection {
     constructor(fields, caller) {
-        super(fields, caller);
+        super(fields, caller, new GoogleAuth(fields.authOptions));
         Object.defineProperty(this, "index", {
             enumerable: true,
             configurable: true,
@@ -80,7 +81,7 @@ class UpsertDatapointConnection extends GoogleVertexAIConnection {
         this.index = fields.index;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexes/${this.index}:upsertDatapoints`;
         return url;
     }
@@ -96,7 +97,7 @@ class UpsertDatapointConnection extends GoogleVertexAIConnection {
 }
 class FindNeighborsConnection extends GoogleVertexAIConnection {
     constructor(params, caller) {
-        super(params, caller);
+        super(params, caller, new GoogleAuth(params.authOptions));
         Object.defineProperty(this, "indexEndpoint", {
             enumerable: true,
             configurable: true,
@@ -113,7 +114,7 @@ class FindNeighborsConnection extends GoogleVertexAIConnection {
         this.deployedIndexId = params.deployedIndexId;
     }
     async buildUrl() {
-        const projectId = await this.auth.getProjectId();
+        const projectId = await this.client.getProjectId();
         const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/indexEndpoints/${this.indexEndpoint}:findNeighbors`;
         return url;
     }