@langchain/google-common 0.2.17 → 1.0.0

package/dist/connection.js

@@ -1,461 +1,319 @@
+import { getGeminiAPI } from "./utils/gemini.js";
+import { getAnthropicAPI } from "./utils/anthropic.js";
+import { modelToFamily, modelToPublisher } from "./utils/common.js";
+import "./utils/index.js";
 import { getRuntimeEnvironment } from "@langchain/core/utils/env";
 import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
-import { getGeminiAPI, modelToFamily, modelToPublisher, } from "./utils/index.js";
-import { getAnthropicAPI } from "./utils/anthropic.js";
-export class GoogleConnection {
-    constructor(caller, client, streaming) {
-        Object.defineProperty(this, "caller", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "streaming", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.caller = caller;
-        this.client = client;
-        this.streaming = streaming ?? false;
-    }
-    async _clientInfoHeaders() {
-        const { userAgent, clientLibraryVersion } = await this._getClientInfo();
-        return {
-            "User-Agent": userAgent,
-            "Client-Info": clientLibraryVersion,
-        };
-    }
-    async _getClientInfo() {
-        const env = await getRuntimeEnvironment();
-        const langchain = env?.library ?? "langchain-js";
-        // TODO: Add an API for getting the current LangChain version
-        const langchainVersion = "0";
-        const moduleName = await this._moduleName();
-        let clientLibraryVersion = `${langchain}/${langchainVersion}`;
-        if (moduleName && moduleName.length) {
-            clientLibraryVersion = `${clientLibraryVersion}-${moduleName}`;
-        }
-        return {
-            userAgent: clientLibraryVersion,
-            clientLibraryVersion: `${langchainVersion}-${moduleName}`,
-        };
-    }
-    async _moduleName() {
-        return this.constructor.name;
-    }
-    async additionalHeaders() {
-        return {};
-    }
-    async _buildOpts(data, _options, requestHeaders = {}) {
-        const url = await this.buildUrl();
-        const method = this.buildMethod();
-        const infoHeaders = (await this._clientInfoHeaders()) ?? {};
-        const additionalHeaders = (await this.additionalHeaders()) ?? {};
-        const headers = {
-            ...infoHeaders,
-            ...additionalHeaders,
-            ...requestHeaders,
-        };
-        const opts = {
-            url,
-            method,
-            headers,
-        };
-        if (data && method === "POST") {
-            opts.data = data;
-        }
-        if (this.streaming) {
-            opts.responseType = "stream";
-        }
-        else {
-            opts.responseType = "json";
-        }
-        return opts;
-    }
-    async _request(data, options, requestHeaders = {}) {
-        const opts = await this._buildOpts(data, options, requestHeaders);
-        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;
-    }
-}
-export class GoogleHostConnection extends GoogleConnection {
-    constructor(fields, caller, client, streaming) {
-        super(caller, client, streaming);
-        // This does not default to a value intentionally.
-        // Use the "platform" getter if you need this.
-        Object.defineProperty(this, "platformType", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_endpoint", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_location", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.caller = caller;
-        this.platformType = this.fieldPlatformType(fields);
-        this._endpoint = fields?.endpoint;
-        this._location = fields?.location;
-        this._apiVersion = fields?.apiVersion;
-        this.client = client;
-    }
-    fieldPlatformType(fields) {
-        if (typeof fields === "undefined") {
-            return undefined;
-        }
-        if (typeof fields.platformType !== "undefined") {
-            return fields.platformType;
-        }
-        if (fields.vertexai === true) {
-            return "gcp";
-        }
-        return undefined;
-    }
-    get platform() {
-        return this.platformType ?? this.computedPlatformType;
-    }
-    get computedPlatformType() {
-        return "gcp";
-    }
-    get computedApiVersion() {
-        return "v1";
-    }
-    get apiVersion() {
-        return this._apiVersion ?? this.computedApiVersion;
-    }
-    get location() {
-        return this._location ?? this.computedLocation;
-    }
-    get computedLocation() {
-        return "us-central1";
-    }
-    get endpoint() {
-        return this._endpoint ?? this.computedEndpoint;
-    }
-    get computedEndpoint() {
-        if (this.location === "global") {
-            return "aiplatform.googleapis.com";
-        }
-        else {
-            return `${this.location}-aiplatform.googleapis.com`;
-        }
-    }
-    buildMethod() {
-        return "POST";
-    }
-}
-export class GoogleRawConnection extends GoogleHostConnection {
-    async _buildOpts(data, _options, requestHeaders = {}) {
-        const opts = await super._buildOpts(data, _options, requestHeaders);
-        opts.responseType = "blob";
-        return opts;
-    }
-}
-export class GoogleAIConnection extends GoogleHostConnection {
-    constructor(fields, caller, client, streaming) {
-        super(fields, caller, client, streaming);
-        Object.defineProperty(this, "model", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "modelName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "client", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "_apiName", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "apiConfig", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.client = client;
-        this.modelName = fields?.model ?? fields?.modelName ?? this.model;
-        this.model = this.modelName;
-        this._apiName = fields?.apiName;
-        this.apiConfig = {
-            safetyHandler: fields?.safetyHandler, // For backwards compatibility
-            ...fields?.apiConfig,
-        };
-    }
-    get modelFamily() {
-        return modelToFamily(this.model);
-    }
-    get modelPublisher() {
-        return modelToPublisher(this.model);
-    }
-    get computedAPIName() {
-        // At least at the moment, model publishers and APIs map the same
-        return this.modelPublisher;
-    }
-    get apiName() {
-        return this._apiName ?? this.computedAPIName;
-    }
-    get api() {
-        switch (this.apiName) {
-            case "google":
-            case "gemma": // TODO: Is this true?
-                return getGeminiAPI(this.apiConfig);
-            case "anthropic":
-                return getAnthropicAPI(this.apiConfig);
-            default:
-                throw new Error(`Unknown API: ${this.apiName}`);
-        }
-    }
-    get isApiKey() {
-        return this.client.clientType === "apiKey";
-    }
-    fieldPlatformType(fields) {
-        const ret = super.fieldPlatformType(fields);
-        if (typeof ret !== "undefined") {
-            return ret;
-        }
-        if (fields?.vertexai === false) {
-            return "gai";
-        }
-        return undefined;
-    }
-    get computedPlatformType() {
-        // This is not a completely correct assumption, since GCP can
-        // have an API Key. But if so, then people need to set the platform
-        // type explicitly.
-        if (this.isApiKey) {
-            return "gai";
-        }
-        else {
-            return "gcp";
-        }
-    }
-    get computedApiVersion() {
-        switch (this.platform) {
-            case "gai":
-                return "v1beta";
-            default:
-                return "v1";
-        }
-    }
-    get computedLocation() {
-        switch (this.apiName) {
-            case "google":
-                return super.computedLocation;
-            case "anthropic":
-                return "us-east5";
-            default:
-                throw new Error(`Unknown apiName: ${this.apiName}. Can't get location.`);
-        }
-    }
-    async buildUrlGenerativeLanguage() {
-        const method = await this.buildUrlMethod();
-        const url = `https://generativelanguage.googleapis.com/${this.apiVersion}/models/${this.model}:${method}`;
-        return url;
-    }
-    async buildUrlVertexExpress() {
-        const method = await this.buildUrlMethod();
-        const publisher = this.modelPublisher;
-        const url = `https://aiplatform.googleapis.com/${this.apiVersion}/publishers/${publisher}/models/${this.model}:${method}`;
-        return url;
-    }
-    async buildUrlVertexLocation() {
-        const projectId = await this.client.getProjectId();
-        const method = await this.buildUrlMethod();
-        const publisher = this.modelPublisher;
-        const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/publishers/${publisher}/models/${this.model}:${method}`;
-        return url;
-    }
-    async buildUrlVertex() {
-        if (this.isApiKey) {
-            return this.buildUrlVertexExpress();
-        }
-        else {
-            return this.buildUrlVertexLocation();
-        }
-    }
-    async buildUrl() {
-        switch (this.platform) {
-            case "gai":
-                return this.buildUrlGenerativeLanguage();
-            default:
-                return this.buildUrlVertex();
-        }
-    }
-    async request(input, parameters, options, runManager) {
-        const moduleName = this.constructor.name;
-        const streamingParameters = {
-            ...parameters,
-            streaming: this.streaming,
-        };
-        const data = await this.formatData(input, streamingParameters);
-        await runManager?.handleCustomEvent(`google-request-${moduleName}`, {
-            data,
-            parameters: streamingParameters,
-            options,
-            connection: {
-                ...this,
-                url: await this.buildUrl(),
-                urlMethod: await this.buildUrlMethod(),
-                modelFamily: this.modelFamily,
-                modelPublisher: this.modelPublisher,
-                computedPlatformType: this.computedPlatformType,
-            },
-        });
-        const response = await this._request(data, options);
-        await runManager?.handleCustomEvent(`google-response-${moduleName}`, {
-            response,
-        });
-        return response;
-    }
-}
-export class AbstractGoogleLLMConnection extends GoogleAIConnection {
-    async buildUrlMethodGemini() {
-        return this.streaming ? "streamGenerateContent" : "generateContent";
-    }
-    async buildUrlMethodClaude() {
-        return this.streaming ? "streamRawPredict" : "rawPredict";
-    }
-    async buildUrlMethod() {
-        switch (this.modelFamily) {
-            case "gemini":
-            case "gemma": // TODO: Is this true?
-                return this.buildUrlMethodGemini();
-            case "claude":
-                return this.buildUrlMethodClaude();
-            default:
-                throw new Error(`Unknown model family: ${this.modelFamily}`);
-        }
-    }
-    async formatData(input, parameters) {
-        // Filter out labels for non-Vertex AI platforms (labels are only supported on Vertex AI)
-        let filteredParameters = parameters;
-        if (parameters.labels && this.platform !== "gcp") {
-            const { labels, ...paramsWithoutLabels } = parameters;
-            filteredParameters = paramsWithoutLabels;
-        }
-        return this.api.formatData(input, filteredParameters);
-    }
-}
-export class GoogleRequestCallbackHandler extends BaseCallbackHandler {
-    customEventInfo(eventName) {
-        const names = eventName.split("-");
-        return {
-            subEvent: names[1],
-            module: names[2],
-        };
-    }
-    handleCustomEvent(eventName, data, runId, tags, metadata) {
-        if (!eventName) {
-            return undefined;
-        }
-        const eventInfo = this.customEventInfo(eventName);
-        switch (eventInfo.subEvent) {
-            case "request":
-                return this.handleCustomRequestEvent(eventName, eventInfo, data, runId, tags, metadata);
-            case "response":
-                return this.handleCustomResponseEvent(eventName, eventInfo, data, runId, tags, metadata);
-            case "chunk":
-                return this.handleCustomChunkEvent(eventName, eventInfo, data, runId, tags, metadata);
-            default:
-                console.error(`Unexpected eventInfo for ${eventName} ${JSON.stringify(eventInfo, null, 1)}`);
-        }
-    }
-}
-export class GoogleRequestLogger extends GoogleRequestCallbackHandler {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "GoogleRequestLogger"
-        });
-    }
-    log(eventName, data, tags) {
-        const tagStr = tags ? `[${tags}]` : "[]";
-        console.log(`${eventName} ${tagStr} ${JSON.stringify(data, null, 1)}`);
-    }
-    handleCustomRequestEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
-        this.log(eventName, data, tags);
-    }
-    handleCustomResponseEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
-        this.log(eventName, data, tags);
-    }
-    handleCustomChunkEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
-        this.log(eventName, data, tags);
-    }
-}
-export class GoogleRequestRecorder extends GoogleRequestCallbackHandler {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "GoogleRequestRecorder"
-        });
-        Object.defineProperty(this, "request", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "response", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: {}
-        });
-        Object.defineProperty(this, "chunk", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: []
-        });
-    }
-    handleCustomRequestEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
-        this.request = data;
-    }
-    handleCustomResponseEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
-        this.response = data;
-    }
-    handleCustomChunkEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
-        this.chunk.push(data);
-    }
-}
+
+//#region src/connection.ts
+var GoogleConnection = class {
+	caller;
+	client;
+	streaming;
+	constructor(caller, client, streaming) {
+		this.caller = caller;
+		this.client = client;
+		this.streaming = streaming ?? false;
+	}
+	async _clientInfoHeaders() {
+		const { userAgent, clientLibraryVersion } = await this._getClientInfo();
+		return {
+			"User-Agent": userAgent,
+			"Client-Info": clientLibraryVersion
+		};
+	}
+	async _getClientInfo() {
+		const env = await getRuntimeEnvironment();
+		const langchain = env?.library ?? "langchain-js";
+		const langchainVersion = "0";
+		const moduleName = await this._moduleName();
+		let clientLibraryVersion = `${langchain}/${langchainVersion}`;
+		if (moduleName && moduleName.length) clientLibraryVersion = `${clientLibraryVersion}-${moduleName}`;
+		return {
+			userAgent: clientLibraryVersion,
+			clientLibraryVersion: `${langchainVersion}-${moduleName}`
+		};
+	}
+	async _moduleName() {
+		return this.constructor.name;
+	}
+	async additionalHeaders() {
+		return {};
+	}
+	async _buildOpts(data, _options, requestHeaders = {}) {
+		const url = await this.buildUrl();
+		const method = this.buildMethod();
+		const infoHeaders = await this._clientInfoHeaders() ?? {};
+		const additionalHeaders = await this.additionalHeaders() ?? {};
+		const headers = {
+			...infoHeaders,
+			...additionalHeaders,
+			...requestHeaders
+		};
+		const opts = {
+			url,
+			method,
+			headers
+		};
+		if (data && method === "POST") opts.data = data;
+		if (this.streaming) opts.responseType = "stream";
+		else opts.responseType = "json";
+		return opts;
+	}
+	async _request(data, options, requestHeaders = {}) {
+		const opts = await this._buildOpts(data, options, requestHeaders);
+		const callResponse = await this.caller.callWithOptions({ signal: options?.signal }, async () => this.client.request(opts));
+		const response = callResponse;
+		return response;
+	}
+};
+var GoogleHostConnection = class extends GoogleConnection {
+	platformType;
+	_endpoint;
+	_location;
+	_apiVersion;
+	constructor(fields, caller, client, streaming) {
+		super(caller, client, streaming);
+		this.caller = caller;
+		this.platformType = this.fieldPlatformType(fields);
+		this._endpoint = fields?.endpoint;
+		this._location = fields?.location;
+		this._apiVersion = fields?.apiVersion;
+		this.client = client;
+	}
+	fieldPlatformType(fields) {
+		if (typeof fields === "undefined") return void 0;
+		if (typeof fields.platformType !== "undefined") return fields.platformType;
+		if (fields.vertexai === true) return "gcp";
+		return void 0;
+	}
+	get platform() {
+		return this.platformType ?? this.computedPlatformType;
+	}
+	get computedPlatformType() {
+		return "gcp";
+	}
+	get computedApiVersion() {
+		return "v1";
+	}
+	get apiVersion() {
+		return this._apiVersion ?? this.computedApiVersion;
+	}
+	get location() {
+		return this._location ?? this.computedLocation;
+	}
+	get computedLocation() {
+		return "us-central1";
+	}
+	get endpoint() {
+		return this._endpoint ?? this.computedEndpoint;
+	}
+	get computedEndpoint() {
+		if (this.location === "global") return "aiplatform.googleapis.com";
+		else return `${this.location}-aiplatform.googleapis.com`;
+	}
+	buildMethod() {
+		return "POST";
+	}
+};
+var GoogleRawConnection = class extends GoogleHostConnection {
+	async _buildOpts(data, _options, requestHeaders = {}) {
+		const opts = await super._buildOpts(data, _options, requestHeaders);
+		opts.responseType = "blob";
+		return opts;
+	}
+};
+var GoogleAIConnection = class extends GoogleHostConnection {
+	model;
+	modelName;
+	client;
+	_apiName;
+	apiConfig;
+	constructor(fields, caller, client, streaming) {
+		super(fields, caller, client, streaming);
+		this.client = client;
+		this.modelName = fields?.model ?? fields?.modelName ?? this.model;
+		this.model = this.modelName;
+		this._apiName = fields?.apiName;
+		this.apiConfig = {
+			safetyHandler: fields?.safetyHandler,
+			...fields?.apiConfig
+		};
+	}
+	get modelFamily() {
+		return modelToFamily(this.model);
+	}
+	get modelPublisher() {
+		return modelToPublisher(this.model);
+	}
+	get computedAPIName() {
+		return this.modelPublisher;
+	}
+	get apiName() {
+		return this._apiName ?? this.computedAPIName;
+	}
+	get api() {
+		switch (this.apiName) {
+			case "google":
+			case "gemma": return getGeminiAPI(this.apiConfig);
+			case "anthropic": return getAnthropicAPI(this.apiConfig);
+			default: throw new Error(`Unknown API: ${this.apiName}`);
+		}
+	}
+	get isApiKey() {
+		return this.client.clientType === "apiKey";
+	}
+	fieldPlatformType(fields) {
+		const ret = super.fieldPlatformType(fields);
+		if (typeof ret !== "undefined") return ret;
+		if (fields?.vertexai === false) return "gai";
+		return void 0;
+	}
+	get computedPlatformType() {
+		if (this.isApiKey) return "gai";
+		else return "gcp";
+	}
+	get computedApiVersion() {
+		switch (this.platform) {
+			case "gai": return "v1beta";
+			default: return "v1";
+		}
+	}
+	get computedLocation() {
+		switch (this.apiName) {
+			case "google": return super.computedLocation;
+			case "anthropic": return "us-east5";
+			default: throw new Error(`Unknown apiName: ${this.apiName}. Can't get location.`);
+		}
+	}
+	async buildUrlGenerativeLanguage() {
+		const method = await this.buildUrlMethod();
+		const url = `https://generativelanguage.googleapis.com/${this.apiVersion}/models/${this.model}:${method}`;
+		return url;
+	}
+	async buildUrlVertexExpress() {
+		const method = await this.buildUrlMethod();
+		const publisher = this.modelPublisher;
+		const url = `https://aiplatform.googleapis.com/${this.apiVersion}/publishers/${publisher}/models/${this.model}:${method}`;
+		return url;
+	}
+	async buildUrlVertexLocation() {
+		const projectId = await this.client.getProjectId();
+		const method = await this.buildUrlMethod();
+		const publisher = this.modelPublisher;
+		const url = `https://${this.endpoint}/${this.apiVersion}/projects/${projectId}/locations/${this.location}/publishers/${publisher}/models/${this.model}:${method}`;
+		return url;
+	}
+	async buildUrlVertex() {
+		if (this.isApiKey) return this.buildUrlVertexExpress();
+		else return this.buildUrlVertexLocation();
+	}
+	async buildUrl() {
+		switch (this.platform) {
+			case "gai": return this.buildUrlGenerativeLanguage();
+			default: return this.buildUrlVertex();
+		}
+	}
+	async request(input, parameters, options, runManager) {
+		const moduleName = this.constructor.name;
+		const streamingParameters = {
+			...parameters,
+			streaming: this.streaming
+		};
+		const data = await this.formatData(input, streamingParameters);
+		await runManager?.handleCustomEvent(`google-request-${moduleName}`, {
+			data,
+			parameters: streamingParameters,
+			options,
+			connection: {
+				...this,
+				url: await this.buildUrl(),
+				urlMethod: await this.buildUrlMethod(),
+				modelFamily: this.modelFamily,
+				modelPublisher: this.modelPublisher,
+				computedPlatformType: this.computedPlatformType
+			}
+		});
+		const response = await this._request(data, options);
+		await runManager?.handleCustomEvent(`google-response-${moduleName}`, { response });
+		return response;
+	}
+};
+var AbstractGoogleLLMConnection = class extends GoogleAIConnection {
+	async buildUrlMethodGemini() {
+		return this.streaming ? "streamGenerateContent" : "generateContent";
+	}
+	async buildUrlMethodClaude() {
+		return this.streaming ? "streamRawPredict" : "rawPredict";
+	}
+	async buildUrlMethod() {
+		switch (this.modelFamily) {
+			case "gemini":
+			case "gemma": return this.buildUrlMethodGemini();
+			case "claude": return this.buildUrlMethodClaude();
+			default: throw new Error(`Unknown model family: ${this.modelFamily}`);
+		}
+	}
+	async formatData(input, parameters) {
+		let filteredParameters = parameters;
+		if (parameters.labels && this.platform !== "gcp") {
+			const { labels,...paramsWithoutLabels } = parameters;
+			filteredParameters = paramsWithoutLabels;
+		}
+		return this.api.formatData(input, filteredParameters);
+	}
+};
+var GoogleRequestCallbackHandler = class extends BaseCallbackHandler {
+	customEventInfo(eventName) {
+		const names = eventName.split("-");
+		return {
+			subEvent: names[1],
+			module: names[2]
+		};
+	}
+	handleCustomEvent(eventName, data, runId, tags, metadata) {
+		if (!eventName) return void 0;
+		const eventInfo = this.customEventInfo(eventName);
+		switch (eventInfo.subEvent) {
+			case "request": return this.handleCustomRequestEvent(eventName, eventInfo, data, runId, tags, metadata);
+			case "response": return this.handleCustomResponseEvent(eventName, eventInfo, data, runId, tags, metadata);
+			case "chunk": return this.handleCustomChunkEvent(eventName, eventInfo, data, runId, tags, metadata);
+			default: console.error(`Unexpected eventInfo for ${eventName} ${JSON.stringify(eventInfo, null, 1)}`);
+		}
+	}
+};
+var GoogleRequestLogger = class extends GoogleRequestCallbackHandler {
+	name = "GoogleRequestLogger";
+	log(eventName, data, tags) {
+		const tagStr = tags ? `[${tags}]` : "[]";
+		console.log(`${eventName} ${tagStr} ${JSON.stringify(data, null, 1)}`);
+	}
+	handleCustomRequestEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
+		this.log(eventName, data, tags);
+	}
+	handleCustomResponseEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
+		this.log(eventName, data, tags);
+	}
+	handleCustomChunkEvent(eventName, _eventInfo, data, _runId, tags, _metadata) {
+		this.log(eventName, data, tags);
+	}
+};
+var GoogleRequestRecorder = class extends GoogleRequestCallbackHandler {
+	name = "GoogleRequestRecorder";
+	request = {};
+	response = {};
+	chunk = [];
+	handleCustomRequestEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
+		this.request = data;
+	}
+	handleCustomResponseEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
+		this.response = data;
+	}
+	handleCustomChunkEvent(_eventName, _eventInfo, data, _runId, _tags, _metadata) {
+		this.chunk.push(data);
+	}
+};
+
+//#endregion
+export { AbstractGoogleLLMConnection, GoogleAIConnection, GoogleConnection, GoogleHostConnection, GoogleRawConnection, GoogleRequestCallbackHandler, GoogleRequestLogger, GoogleRequestRecorder };
+//# sourceMappingURL=connection.js.map