@langchain/google-common 2.1.18 → 2.1.20

package/dist/experimental/media.js

@@ -12,7 +12,7 @@ var GoogleMultipartUploadConnection = class extends GoogleHostConnection {
 	async _body(separator, data, metadata) {
 		const contentType = data.mimetype;
 		const { encoded, encoding } = await data.encode();
-		const body = [
+		return [
 			`--${separator}`,
 			"Content-Type: application/json; charset=UTF-8",
 			"",
@@ -24,8 +24,7 @@ var GoogleMultipartUploadConnection = class extends GoogleHostConnection {
 			"",
 			encoded,
 			`--${separator}--`
-		];
-		return body.join("\n");
+		].join("\n");
 	}
 	async request(data, metadata, options) {
 		const separator = `separator-${Date.now()}`;
@@ -34,8 +33,7 @@ var GoogleMultipartUploadConnection = class extends GoogleHostConnection {
 			"Content-Type": `multipart/related; boundary=${separator}`,
 			"X-Goog-Upload-Protocol": "multipart"
 		};
-		const response = this._request(body, options, requestHeaders);
-		return response;
+		return this._request(body, options, requestHeaders);
 	}
 };
 var GoogleDownloadConnection = class extends GoogleHostConnection {
@@ -61,28 +59,18 @@ var BlobStoreGoogle = class extends BlobStore {
 	}
 	async _set(keyValuePair) {
 		const [, blob] = keyValuePair;
-		const setMetadata = this.buildSetMetadata(keyValuePair);
-		const metadata = setMetadata;
-		const options = {};
-		const connection = this.buildSetConnection(keyValuePair);
-		const response = await connection.request(blob, metadata, options);
-		return response;
+		const metadata = this.buildSetMetadata(keyValuePair);
+		return await this.buildSetConnection(keyValuePair).request(blob, metadata, {});
 	}
 	async mset(keyValuePairs) {
 		const ret = keyValuePairs.map((keyValue) => this._set(keyValue));
 		await Promise.all(ret);
 	}
 	async _getMetadata(key) {
-		const connection = this.buildGetMetadataConnection(key);
-		const options = {};
-		const response = await connection.request(options);
-		return response.data;
+		return (await this.buildGetMetadataConnection(key).request({})).data;
 	}
 	async _getData(key) {
-		const connection = this.buildGetDataConnection(key);
-		const options = {};
-		const response = await connection.request(options);
-		return response.data;
+		return (await this.buildGetDataConnection(key).request({})).data;
 	}
 	_getMimetypeFromMetadata(metadata) {
 		return metadata.contentType;
@@ -90,22 +78,18 @@ var BlobStoreGoogle = class extends BlobStore {
 	async _get(key) {
 		const metadata = await this._getMetadata(key);
 		const data = await this._getData(key);
-		if (data && metadata) {
-			const ret = await MediaBlob.fromBlob(data, {
-				metadata,
-				path: key
-			});
-			return ret;
-		} else return void 0;
+		if (data && metadata) return await MediaBlob.fromBlob(data, {
+			metadata,
+			path: key
+		});
+		else return;
 	}
 	async mget(keys) {
 		const ret = keys.map((key) => this._get(key));
 		return await Promise.all(ret);
 	}
 	async _del(key) {
-		const connection = this.buildDeleteConnection(key);
-		const options = {};
-		await connection.request(options);
+		await this.buildDeleteConnection(key).request({});
 	}
 	async mdelete(keys) {
 		const ret = keys.map((key) => this._del(key));
@@ -179,8 +163,7 @@ var GoogleCloudStorageRawConnection = class extends GoogleDownloadRawConnection
 	}
 	async buildUrl() {
 		const path = encodeURIComponent(this.uri.path);
-		const ret = `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}?alt=media`;
-		return ret;
+		return `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}?alt=media`;
 	}
 };
 var BlobStoreGoogleCloudStorageBase = class extends BlobStoreGoogle {
@@ -194,40 +177,34 @@ var BlobStoreGoogleCloudStorageBase = class extends BlobStoreGoogle {
 		};
 	}
 	buildSetConnection([key, _blob]) {
-		const params = {
+		return new GoogleCloudStorageUploadConnection({
 			...this.params,
 			uri: key
-		};
-		return new GoogleCloudStorageUploadConnection(params, this.caller, this.client);
+		}, this.caller, this.client);
 	}
 	buildSetMetadata([key, blob]) {
-		const uri = new GoogleCloudStorageUri(key);
-		const ret = {
-			name: uri.path,
+		return {
+			name: new GoogleCloudStorageUri(key).path,
 			metadata: blob.metadata,
 			contentType: blob.mimetype
 		};
-		return ret;
 	}
 	buildGetMetadataConnection(key) {
-		const params = {
+		return new GoogleCloudStorageDownloadConnection({
 			uri: key,
 			method: "GET",
 			alt: void 0
-		};
-		return new GoogleCloudStorageDownloadConnection(params, this.caller, this.client);
+		}, this.caller, this.client);
 	}
 	buildGetDataConnection(key) {
-		const params = { uri: key };
-		return new GoogleCloudStorageRawConnection(params, this.caller, this.client);
+		return new GoogleCloudStorageRawConnection({ uri: key }, this.caller, this.client);
 	}
 	buildDeleteConnection(key) {
-		const params = {
+		return new GoogleCloudStorageDownloadConnection({
 			uri: key,
 			method: "DELETE",
 			alt: void 0
-		};
-		return new GoogleCloudStorageDownloadConnection(params, this.caller, this.client);
+		}, this.caller, this.client);
 	}
 };
 var AIStudioMediaBlob = class extends MediaBlob {
@@ -249,8 +226,7 @@ var AIStudioMediaBlob = class extends MediaBlob {
 	}
 	get isExpired() {
 		const now = (/* @__PURE__ */ new Date()).toISOString();
-		const exp = this.metadata?.expirationTime ?? now;
-		return exp <= now;
+		return (this.metadata?.expirationTime ?? now) <= now;
 	}
 };
 var AIStudioFileUploadConnection = class extends GoogleMultipartUploadConnection {
@@ -330,38 +306,33 @@ var BlobStoreAIStudioFileBase = class extends BlobStoreGoogle {
 		return {};
 	}
 	buildGetMetadataConnection(key) {
-		const params = {
+		return new AIStudioFileDownloadConnection({
 			...this.params,
 			method: "GET",
 			name: this._pathToName(key)
-		};
-		return new AIStudioFileDownloadConnection(params, this.caller, this.client);
+		}, this.caller, this.client);
 	}
 	buildGetDataConnection(_key) {
 		throw new Error("AI Studio File API does not provide data");
 	}
 	async _get(key) {
 		const metadata = await this._getMetadata(key);
-		if (metadata) {
-			const contentType = metadata?.mimeType ?? "application/octet-stream";
-			const data = {
+		if (metadata) return new MediaBlob({
+			path: key,
+			data: {
 				value: "",
-				type: contentType
-			};
-			return new MediaBlob({
-				path: key,
-				data,
-				metadata
-			});
-		} else return void 0;
+				type: metadata?.mimeType ?? "application/octet-stream"
+			},
+			metadata
+		});
+		else return;
 	}
 	buildDeleteConnection(key) {
-		const params = {
+		return new AIStudioFileDownloadConnection({
 			...this.params,
 			method: "DELETE",
 			name: this._pathToName(key)
-		};
-		return new AIStudioFileDownloadConnection(params, this.caller, this.client);
+		}, this.caller, this.client);
 	}
 };
 

package/dist/experimental/media.js.map

@@ -1 +1 @@
-{"version":3,"file":"media.js","names":["fields: GoogleConnectionParams<AuthOptions> | undefined","caller: AsyncCaller","client: GoogleAbstractedClient","separator: string","data: MediaBlob","metadata: Record<string, unknown>","options: CallOptions","fields?: BlobStoreGoogleParams<AuthOptions>","keyValuePair: [string, MediaBlob]","keyValuePairs: [string, MediaBlob][]","key: string","keys: string[]","_prefix: string | undefined","uri: string","fields: GoogleCloudStorageUploadConnectionParams<AuthOptions>","fields: GoogleCloudStorageDownloadConnectionParams<AuthOptions>","fields: GoogleCloudStorageRawConnectionParams<AuthOptions>","fields: BlobStoreGoogleCloudStorageBaseParams<AuthOptions>","params: GoogleCloudStorageUploadConnectionParams<AuthOptions>","ret: GoogleCloudStorageObject","params: GoogleCloudStorageDownloadConnectionParams<AuthOptions>","params: GoogleCloudStorageRawConnectionParams<AuthOptions>","value: string","field: string","fields: AIStudioFileDownloadConnectionParams<AuthOptions>","fields?: BlobStoreAIStudioFileBaseParams<AuthOptions>","params: BlobStoreAIStudioFileBaseParams<AuthOptions>","path: string","apiKey: string","params: AIStudioFileDownloadConnectionParams<AuthOptions>","_key: string","data: MediaBlobData"],"sources":["../../src/experimental/media.ts"],"sourcesContent":["import {\n  AsyncCaller,\n  AsyncCallerCallOptions,\n  AsyncCallerParams,\n} from \"@langchain/core/utils/async_caller\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport {\n  MediaBlob,\n  BlobStore,\n  BlobStoreOptions,\n  MediaBlobData,\n} from \"./utils/media_core.js\";\nimport {\n  GoogleConnectionParams,\n  GoogleRawResponse,\n  GoogleResponse,\n} from \"../types.js\";\nimport { GoogleHostConnection, GoogleRawConnection } from \"../connection.js\";\nimport {\n  ApiKeyGoogleAuth,\n  GoogleAbstractedClient,\n  GoogleAbstractedClientOpsMethod,\n} from \"../auth.js\";\n\nexport interface GoogleUploadConnectionParams<\n  AuthOptions,\n> extends GoogleConnectionParams<AuthOptions> {}\n\nexport abstract class GoogleMultipartUploadConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleHostConnection<CallOptions, ResponseType, AuthOptions> {\n  constructor(\n    fields: GoogleConnectionParams<AuthOptions> | undefined,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n  }\n\n  async _body(\n    separator: string,\n    data: MediaBlob,\n    metadata: Record<string, unknown>\n  ): Promise<string> {\n    const contentType = data.mimetype;\n    const { encoded, encoding } = await data.encode();\n    const body = [\n      `--${separator}`,\n      \"Content-Type: application/json; charset=UTF-8\",\n      \"\",\n      JSON.stringify(metadata),\n      \"\",\n      `--${separator}`,\n      `Content-Type: ${contentType}`,\n      `Content-Transfer-Encoding: ${encoding}`,\n      \"\",\n      encoded,\n      `--${separator}--`,\n    ];\n    return body.join(\"\\n\");\n  }\n\n  async request(\n    data: MediaBlob,\n    metadata: Record<string, unknown>,\n    options: CallOptions\n  ): Promise<ResponseType> {\n    const separator = `separator-${Date.now()}`;\n    const body = await this._body(separator, data, metadata);\n    const requestHeaders = {\n      \"Content-Type\": `multipart/related; boundary=${separator}`,\n      \"X-Goog-Upload-Protocol\": \"multipart\",\n    };\n    const response = this._request(body, options, requestHeaders);\n    return response;\n  }\n}\n\nexport abstract class GoogleDownloadConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleHostConnection<CallOptions, ResponseType, AuthOptions> {\n  async request(options: CallOptions): Promise<ResponseType> {\n    return this._request(undefined, options);\n  }\n}\n\nexport abstract class GoogleDownloadRawConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  AuthOptions,\n> extends GoogleRawConnection<CallOptions, AuthOptions> {\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return \"GET\";\n  }\n\n  async request(options: CallOptions): Promise<GoogleRawResponse> {\n    return this._request(undefined, options);\n  }\n}\n\nexport interface BlobStoreGoogleParams<AuthOptions>\n  extends\n    GoogleConnectionParams<AuthOptions>,\n    AsyncCallerParams,\n    BlobStoreOptions {}\n\nexport abstract class BlobStoreGoogle<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends BlobStore {\n  caller: AsyncCaller;\n\n  client: GoogleAbstractedClient;\n\n  constructor(fields?: BlobStoreGoogleParams<AuthOptions>) {\n    super(fields);\n    this.caller = new AsyncCaller(fields ?? {});\n    this.client = this.buildClient(fields);\n  }\n\n  abstract buildClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  abstract buildSetMetadata([key, blob]: [string, MediaBlob]): Record<\n    string,\n    unknown\n  >;\n\n  abstract buildSetConnection([key, blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    ResponseType,\n    AuthOptions\n  >;\n\n  async _set(keyValuePair: [string, MediaBlob]): Promise<ResponseType> {\n    const [, blob] = keyValuePair;\n    const setMetadata = this.buildSetMetadata(keyValuePair);\n    const metadata = setMetadata;\n    const options = {};\n    const connection = this.buildSetConnection(keyValuePair);\n    const response = await connection.request(blob, metadata, options);\n    return response;\n  }\n\n  async mset(keyValuePairs: [string, MediaBlob][]): Promise<void> {\n    const ret = keyValuePairs.map((keyValue) => this._set(keyValue));\n    await Promise.all(ret);\n  }\n\n  abstract buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    ResponseType,\n    AuthOptions\n  >;\n\n  async _getMetadata(key: string): Promise<Record<string, unknown>> {\n    const connection = this.buildGetMetadataConnection(key);\n    const options = {};\n    const response = await connection.request(options);\n    return response.data;\n  }\n\n  abstract buildGetDataConnection(\n    key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions>;\n\n  async _getData(key: string): Promise<Blob> {\n    const connection = this.buildGetDataConnection(key);\n    const options = {};\n    const response = await connection.request(options);\n    return response.data;\n  }\n\n  _getMimetypeFromMetadata(metadata: Record<string, unknown>): string {\n    return metadata.contentType as string;\n  }\n\n  async _get(key: string): Promise<MediaBlob | undefined> {\n    const metadata = await this._getMetadata(key);\n    const data = await this._getData(key);\n    if (data && metadata) {\n      const ret = await MediaBlob.fromBlob(data, { metadata, path: key });\n      return ret;\n    } else {\n      return undefined;\n    }\n  }\n\n  async mget(keys: string[]): Promise<(MediaBlob | undefined)[]> {\n    const ret = keys.map((key) => this._get(key));\n    return await Promise.all(ret);\n  }\n\n  abstract buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleResponse,\n    AuthOptions\n  >;\n\n  async _del(key: string): Promise<void> {\n    const connection = this.buildDeleteConnection(key);\n    const options = {};\n    await connection.request(options);\n  }\n\n  async mdelete(keys: string[]): Promise<void> {\n    const ret = keys.map((key) => this._del(key));\n    await Promise.all(ret);\n  }\n\n  // eslint-disable-next-line require-yield\n  async *yieldKeys(_prefix: string | undefined): AsyncGenerator<string> {\n    // TODO: Implement. Most have an implementation that uses nextToken.\n    throw new Error(\"yieldKeys is not implemented\");\n  }\n}\n\n/**\n * Based on https://cloud.google.com/storage/docs/json_api/v1/objects#resource\n */\nexport interface GoogleCloudStorageObject extends Record<string, unknown> {\n  id?: string;\n  name?: string;\n  contentType?: string;\n  metadata?: Record<string, unknown>;\n  // This is incomplete.\n}\n\nexport interface GoogleCloudStorageResponse extends GoogleResponse {\n  data: GoogleCloudStorageObject;\n}\n\nexport type BucketAndPath = {\n  bucket: string;\n  path: string;\n};\n\nexport class GoogleCloudStorageUri {\n  static uriRegexp = /gs:\\/\\/([a-z0-9][a-z0-9._-]+[a-z0-9])\\/(.*)/;\n\n  bucket: string;\n\n  path: string;\n\n  constructor(uri: string) {\n    const bucketAndPath = GoogleCloudStorageUri.uriToBucketAndPath(uri);\n    this.bucket = bucketAndPath.bucket;\n    this.path = bucketAndPath.path;\n  }\n\n  get uri() {\n    return `gs://${this.bucket}/${this.path}`;\n  }\n\n  get isValid() {\n    return (\n      typeof this.bucket !== \"undefined\" && typeof this.path !== \"undefined\"\n    );\n  }\n\n  static uriToBucketAndPath(uri: string): BucketAndPath {\n    const match = this.uriRegexp.exec(uri);\n    if (!match) {\n      throw new Error(`Invalid gs:// URI: ${uri}`);\n    }\n    return {\n      bucket: match[1],\n      path: match[2],\n    };\n  }\n\n  static isValidUri(uri: string): boolean {\n    return this.uriRegexp.test(uri);\n  }\n}\n\nexport interface GoogleCloudStorageConnectionParams {\n  uri: string;\n}\n\nexport interface GoogleCloudStorageUploadConnectionParams<AuthOptions>\n  extends\n    GoogleUploadConnectionParams<AuthOptions>,\n    GoogleCloudStorageConnectionParams {}\n\nexport class GoogleCloudStorageUploadConnection<\n  AuthOptions,\n> extends GoogleMultipartUploadConnection<\n  AsyncCallerCallOptions,\n  GoogleCloudStorageResponse,\n  AuthOptions\n> {\n  uri: GoogleCloudStorageUri;\n\n  constructor(\n    fields: GoogleCloudStorageUploadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://storage.googleapis.com/upload/storage/${this.apiVersion}/b/${this.uri.bucket}/o?uploadType=multipart`;\n  }\n}\n\nexport interface GoogleCloudStorageDownloadConnectionParams<AuthOptions>\n  extends\n    GoogleCloudStorageConnectionParams,\n    GoogleConnectionParams<AuthOptions> {\n  method: GoogleAbstractedClientOpsMethod;\n  alt: \"media\" | undefined;\n}\n\nexport class GoogleCloudStorageDownloadConnection<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleDownloadConnection<\n  AsyncCallerCallOptions,\n  ResponseType,\n  AuthOptions\n> {\n  uri: GoogleCloudStorageUri;\n\n  method: GoogleAbstractedClientOpsMethod;\n\n  alt: \"media\" | undefined;\n\n  constructor(\n    fields: GoogleCloudStorageDownloadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n    this.method = fields.method;\n    this.alt = fields.alt;\n  }\n\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return this.method;\n  }\n\n  async buildUrl(): Promise<string> {\n    const path = encodeURIComponent(this.uri.path);\n    const ret = `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}`;\n    return this.alt ? `${ret}?alt=${this.alt}` : ret;\n  }\n}\n\nexport interface GoogleCloudStorageRawConnectionParams<AuthOptions>\n  extends\n    GoogleCloudStorageConnectionParams,\n    GoogleConnectionParams<AuthOptions> {}\n\nexport class GoogleCloudStorageRawConnection<\n  AuthOptions,\n> extends GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n  uri: GoogleCloudStorageUri;\n\n  constructor(\n    fields: GoogleCloudStorageRawConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n  }\n\n  async buildUrl(): Promise<string> {\n    const path = encodeURIComponent(this.uri.path);\n    const ret = `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}?alt=media`;\n    return ret;\n  }\n}\n\nexport interface BlobStoreGoogleCloudStorageBaseParams<\n  AuthOptions,\n> extends BlobStoreGoogleParams<AuthOptions> {\n  uriPrefix: GoogleCloudStorageUri;\n}\n\nexport abstract class BlobStoreGoogleCloudStorageBase<\n  AuthOptions,\n> extends BlobStoreGoogle<GoogleCloudStorageResponse, AuthOptions> {\n  params: BlobStoreGoogleCloudStorageBaseParams<AuthOptions>;\n\n  constructor(fields: BlobStoreGoogleCloudStorageBaseParams<AuthOptions>) {\n    super(fields);\n    this.params = fields;\n    this.defaultStoreOptions = {\n      ...this.defaultStoreOptions,\n      pathPrefix: fields.uriPrefix.uri,\n    };\n  }\n\n  buildSetConnection([key, _blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    GoogleCloudStorageResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageUploadConnectionParams<AuthOptions> = {\n      ...this.params,\n      uri: key,\n    };\n    return new GoogleCloudStorageUploadConnection<AuthOptions>(\n      params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildSetMetadata([key, blob]: [string, MediaBlob]): Record<string, unknown> {\n    const uri = new GoogleCloudStorageUri(key);\n    const ret: GoogleCloudStorageObject = {\n      name: uri.path,\n      metadata: blob.metadata,\n      contentType: blob.mimetype,\n    };\n    return ret;\n  }\n\n  buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleCloudStorageResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageDownloadConnectionParams<AuthOptions> = {\n      uri: key,\n      method: \"GET\",\n      alt: undefined,\n    };\n    return new GoogleCloudStorageDownloadConnection<\n      GoogleCloudStorageResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n\n  buildGetDataConnection(\n    key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n    const params: GoogleCloudStorageRawConnectionParams<AuthOptions> = {\n      uri: key,\n    };\n    return new GoogleCloudStorageRawConnection<AuthOptions>(\n      params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageDownloadConnectionParams<AuthOptions> = {\n      uri: key,\n      method: \"DELETE\",\n      alt: undefined,\n    };\n    return new GoogleCloudStorageDownloadConnection<\n      GoogleResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n}\n\nexport type AIStudioFileState =\n  | \"PROCESSING\"\n  | \"ACTIVE\"\n  | \"FAILED\"\n  | \"STATE_UNSPECIFIED\";\n\nexport type AIStudioFileVideoMetadata = {\n  videoMetadata: {\n    videoDuration: string; // Duration in seconds, possibly with fractional, ending in \"s\"\n  };\n};\n\nexport type AIStudioFileMetadata = AIStudioFileVideoMetadata;\n\nexport interface AIStudioFileObject {\n  name?: string;\n  displayName?: string;\n  mimeType?: string;\n  sizeBytes?: string; // int64 format\n  createTime?: string; // timestamp format\n  updateTime?: string; // timestamp format\n  expirationTime?: string; // timestamp format\n  sha256Hash?: string; // base64 encoded\n  uri?: string;\n  state?: AIStudioFileState;\n  error?: {\n    code: number;\n    message: string;\n    details: Record<string, unknown>[];\n  };\n  metadata?: AIStudioFileMetadata;\n}\n\nexport class AIStudioMediaBlob extends MediaBlob {\n  _valueAsDate(value: string): Date {\n    if (!value) {\n      return new Date(0);\n    }\n    return new Date(value);\n  }\n\n  _metadataFieldAsDate(field: string): Date {\n    return this._valueAsDate(this.metadata?.[field]);\n  }\n\n  get createDate(): Date {\n    return this._metadataFieldAsDate(\"createTime\");\n  }\n\n  get updateDate(): Date {\n    return this._metadataFieldAsDate(\"updateTime\");\n  }\n\n  get expirationDate(): Date {\n    return this._metadataFieldAsDate(\"expirationTime\");\n  }\n\n  get isExpired(): boolean {\n    const now = new Date().toISOString();\n    const exp = this.metadata?.expirationTime ?? now;\n    return exp <= now;\n  }\n}\n\nexport interface AIStudioFileGetResponse extends GoogleResponse {\n  data: AIStudioFileObject;\n}\n\nexport interface AIStudioFileSaveResponse extends GoogleResponse {\n  data: {\n    file: AIStudioFileObject;\n  };\n}\n\nexport interface AIStudioFileListResponse extends GoogleResponse {\n  data: {\n    files: AIStudioFileObject[];\n    nextPageToken: string;\n  };\n}\n\nexport type AIStudioFileResponse =\n  | AIStudioFileGetResponse\n  | AIStudioFileSaveResponse\n  | AIStudioFileListResponse;\n\nexport interface AIStudioFileConnectionParams {}\n\nexport interface AIStudioFileUploadConnectionParams<AuthOptions>\n  extends\n    GoogleUploadConnectionParams<AuthOptions>,\n    AIStudioFileConnectionParams {}\n\nexport class AIStudioFileUploadConnection<\n  AuthOptions,\n> extends GoogleMultipartUploadConnection<\n  AsyncCallerCallOptions,\n  AIStudioFileSaveResponse,\n  AuthOptions\n> {\n  get computedApiVersion(): string {\n    return \"v1beta\";\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://generativelanguage.googleapis.com/upload/${this.apiVersion}/files`;\n  }\n}\n\nexport interface AIStudioFileDownloadConnectionParams<AuthOptions>\n  extends AIStudioFileConnectionParams, GoogleConnectionParams<AuthOptions> {\n  method: GoogleAbstractedClientOpsMethod;\n  name: string;\n}\n\nexport class AIStudioFileDownloadConnection<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleDownloadConnection<\n  AsyncCallerCallOptions,\n  ResponseType,\n  AuthOptions\n> {\n  method: GoogleAbstractedClientOpsMethod;\n\n  name: string;\n\n  constructor(\n    fields: AIStudioFileDownloadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.method = fields.method;\n    this.name = fields.name;\n  }\n\n  get computedApiVersion(): string {\n    return \"v1beta\";\n  }\n\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return this.method;\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://generativelanguage.googleapis.com/${this.apiVersion}/files/${this.name}`;\n  }\n}\n\nexport interface BlobStoreAIStudioFileBaseParams<\n  AuthOptions,\n> extends BlobStoreGoogleParams<AuthOptions> {\n  retryTime?: number;\n}\n\nexport abstract class BlobStoreAIStudioFileBase<\n  AuthOptions,\n> extends BlobStoreGoogle<AIStudioFileResponse, AuthOptions> {\n  params?: BlobStoreAIStudioFileBaseParams<AuthOptions>;\n\n  retryTime: number = 1000;\n\n  constructor(fields?: BlobStoreAIStudioFileBaseParams<AuthOptions>) {\n    const params: BlobStoreAIStudioFileBaseParams<AuthOptions> = {\n      defaultStoreOptions: {\n        pathPrefix: \"https://generativelanguage.googleapis.com/v1beta/files/\",\n        actionIfInvalid: \"removePath\",\n      },\n      ...fields,\n    };\n    super(params);\n    this.params = params;\n    this.retryTime = params?.retryTime ?? this.retryTime ?? 1000;\n  }\n\n  _pathToName(path: string): string {\n    return path.split(\"/\").pop() ?? path;\n  }\n\n  abstract buildAbstractedClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: BlobStoreGoogleParams<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      // TODO: Test that you can use OAuth to access\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  async _regetMetadata(key: string): Promise<AIStudioFileObject> {\n    // Sleep for some time period\n    await new Promise((resolve) => setTimeout(resolve, this.retryTime));\n\n    // Fetch the latest metadata\n    return this._getMetadata(key);\n  }\n\n  async _set([key, blob]: [\n    string,\n    MediaBlob,\n  ]): Promise<AIStudioFileSaveResponse> {\n    const response = (await super._set([\n      key,\n      blob,\n    ])) as AIStudioFileSaveResponse;\n\n    let file = response.data?.file ?? { state: \"FAILED\" };\n    while (file.state === \"PROCESSING\" && file.uri && this.retryTime > 0) {\n      file = await this._regetMetadata(file.uri);\n    }\n\n    // The response should contain the name (and valid URI), so we need to\n    // update the blob with this. We can't return a new blob, since mset()\n    // doesn't return anything.\n    blob.path = file.uri;\n    blob.metadata = {\n      ...blob.metadata,\n      ...file,\n    };\n    return response;\n  }\n\n  buildSetConnection([_key, _blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    return new AIStudioFileUploadConnection(\n      this.params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildSetMetadata([_key, _blob]: [string, MediaBlob]): Record<\n    string,\n    unknown\n  > {\n    return {};\n  }\n\n  buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    const params: AIStudioFileDownloadConnectionParams<AuthOptions> = {\n      ...this.params,\n      method: \"GET\",\n      name: this._pathToName(key),\n    };\n    return new AIStudioFileDownloadConnection<\n      AIStudioFileResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n\n  buildGetDataConnection(\n    _key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n    throw new Error(\"AI Studio File API does not provide data\");\n  }\n\n  async _get(key: string): Promise<MediaBlob | undefined> {\n    const metadata = await this._getMetadata(key);\n    if (metadata) {\n      const contentType =\n        (metadata?.mimeType as string) ?? \"application/octet-stream\";\n      // TODO - Get the actual data (and other metadata) from an optional backing store\n      const data: MediaBlobData = {\n        value: \"\",\n        type: contentType,\n      };\n\n      return new MediaBlob({\n        path: key,\n        data,\n        metadata,\n      });\n    } else {\n      return undefined;\n    }\n  }\n\n  buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    const params: AIStudioFileDownloadConnectionParams<AuthOptions> = {\n      ...this.params,\n      method: \"DELETE\",\n      name: this._pathToName(key),\n    };\n    return new AIStudioFileDownloadConnection<\n      AIStudioFileResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n}\n"],"mappings":";;;;;;;AA4BA,IAAsB,kCAAtB,cAIU,qBAA6D;CACrE,YACEA,QACAC,QACAC,QACA;EACA,MAAM,QAAQ,QAAQ,OAAO;CAC9B;CAED,MAAM,MACJC,WACAC,MACAC,UACiB;EACjB,MAAM,cAAc,KAAK;EACzB,MAAM,EAAE,SAAS,UAAU,GAAG,MAAM,KAAK,QAAQ;EACjD,MAAM,OAAO;GACX,CAAC,EAAE,EAAE,WAAW;GAChB;GACA;GACA,KAAK,UAAU,SAAS;GACxB;GACA,CAAC,EAAE,EAAE,WAAW;GAChB,CAAC,cAAc,EAAE,aAAa;GAC9B,CAAC,2BAA2B,EAAE,UAAU;GACxC;GACA;GACA,CAAC,EAAE,EAAE,UAAU,EAAE,CAAC;EACnB;AACD,SAAO,KAAK,KAAK,KAAK;CACvB;CAED,MAAM,QACJD,MACAC,UACAC,SACuB;EACvB,MAAM,YAAY,CAAC,UAAU,EAAE,KAAK,KAAK,EAAE;EAC3C,MAAM,OAAO,MAAM,KAAK,MAAM,WAAW,MAAM,SAAS;EACxD,MAAM,iBAAiB;GACrB,gBAAgB,CAAC,4BAA4B,EAAE,WAAW;GAC1D,0BAA0B;EAC3B;EACD,MAAM,WAAW,KAAK,SAAS,MAAM,SAAS,eAAe;AAC7D,SAAO;CACR;AACF;AAED,IAAsB,2BAAtB,cAIU,qBAA6D;CACrE,MAAM,QAAQA,SAA6C;AACzD,SAAO,KAAK,SAAS,QAAW,QAAQ;CACzC;AACF;AAED,IAAsB,8BAAtB,cAGU,oBAA8C;CACtD,cAA+C;AAC7C,SAAO;CACR;CAED,MAAM,QAAQA,SAAkD;AAC9D,SAAO,KAAK,SAAS,QAAW,QAAQ;CACzC;AACF;AAQD,IAAsB,kBAAtB,cAGU,UAAU;CAClB;CAEA;CAEA,YAAYC,QAA6C;EACvD,MAAM,OAAO;EACb,KAAK,SAAS,IAAI,YAAY,UAAU,CAAE;EAC1C,KAAK,SAAS,KAAK,YAAY,OAAO;CACvC;CAoBD,MAAM,KAAKC,cAA0D;EACnE,MAAM,GAAG,KAAK,GAAG;EACjB,MAAM,cAAc,KAAK,iBAAiB,aAAa;EACvD,MAAM,WAAW;EACjB,MAAM,UAAU,CAAE;EAClB,MAAM,aAAa,KAAK,mBAAmB,aAAa;EACxD,MAAM,WAAW,MAAM,WAAW,QAAQ,MAAM,UAAU,QAAQ;AAClE,SAAO;CACR;CAED,MAAM,KAAKC,eAAqD;EAC9D,MAAM,MAAM,cAAc,IAAI,CAAC,aAAa,KAAK,KAAK,SAAS,CAAC;EAChE,MAAM,QAAQ,IAAI,IAAI;CACvB;CAUD,MAAM,aAAaC,KAA+C;EAChE,MAAM,aAAa,KAAK,2BAA2B,IAAI;EACvD,MAAM,UAAU,CAAE;EAClB,MAAM,WAAW,MAAM,WAAW,QAAQ,QAAQ;AAClD,SAAO,SAAS;CACjB;CAMD,MAAM,SAASA,KAA4B;EACzC,MAAM,aAAa,KAAK,uBAAuB,IAAI;EACnD,MAAM,UAAU,CAAE;EAClB,MAAM,WAAW,MAAM,WAAW,QAAQ,QAAQ;AAClD,SAAO,SAAS;CACjB;CAED,yBAAyBL,UAA2C;AAClE,SAAO,SAAS;CACjB;CAED,MAAM,KAAKK,KAA6C;EACtD,MAAM,WAAW,MAAM,KAAK,aAAa,IAAI;EAC7C,MAAM,OAAO,MAAM,KAAK,SAAS,IAAI;AACrC,MAAI,QAAQ,UAAU;GACpB,MAAM,MAAM,MAAM,UAAU,SAAS,MAAM;IAAE;IAAU,MAAM;GAAK,EAAC;AACnE,UAAO;EACR,MACC,QAAO;CAEV;CAED,MAAM,KAAKC,MAAoD;EAC7D,MAAM,MAAM,KAAK,IAAI,CAAC,QAAQ,KAAK,KAAK,IAAI,CAAC;AAC7C,SAAO,MAAM,QAAQ,IAAI,IAAI;CAC9B;CAUD,MAAM,KAAKD,KAA4B;EACrC,MAAM,aAAa,KAAK,sBAAsB,IAAI;EAClD,MAAM,UAAU,CAAE;EAClB,MAAM,WAAW,QAAQ,QAAQ;CAClC;CAED,MAAM,QAAQC,MAA+B;EAC3C,MAAM,MAAM,KAAK,IAAI,CAAC,QAAQ,KAAK,KAAK,IAAI,CAAC;EAC7C,MAAM,QAAQ,IAAI,IAAI;CACvB;CAGD,OAAO,UAAUC,SAAqD;AAEpE,QAAM,IAAI,MAAM;CACjB;AACF;AAsBD,IAAa,wBAAb,MAAa,sBAAsB;CACjC,OAAO,YAAY;CAEnB;CAEA;CAEA,YAAYC,KAAa;EACvB,MAAM,gBAAgB,sBAAsB,mBAAmB,IAAI;EACnE,KAAK,SAAS,cAAc;EAC5B,KAAK,OAAO,cAAc;CAC3B;CAED,IAAI,MAAM;AACR,SAAO,CAAC,KAAK,EAAE,KAAK,OAAO,CAAC,EAAE,KAAK,MAAM;CAC1C;CAED,IAAI,UAAU;AACZ,SACE,OAAO,KAAK,WAAW,eAAe,OAAO,KAAK,SAAS;CAE9D;CAED,OAAO,mBAAmBA,KAA4B;EACpD,MAAM,QAAQ,KAAK,UAAU,KAAK,IAAI;AACtC,MAAI,CAAC,MACH,OAAM,IAAI,MAAM,CAAC,mBAAmB,EAAE,KAAK;AAE7C,SAAO;GACL,QAAQ,MAAM;GACd,MAAM,MAAM;EACb;CACF;CAED,OAAO,WAAWA,KAAsB;AACtC,SAAO,KAAK,UAAU,KAAK,IAAI;CAChC;AACF;AAWD,IAAa,qCAAb,cAEU,gCAIR;CACA;CAEA,YACEC,QACAb,QACAC,QACA;EACA,MAAM,QAAQ,QAAQ,OAAO;EAC7B,KAAK,MAAM,IAAI,sBAAsB,OAAO;CAC7C;CAED,MAAM,WAA4B;AAChC,SAAO,CAAC,8CAA8C,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,IAAI,OAAO,uBAAuB,CAAC;CACtH;AACF;AAUD,IAAa,uCAAb,cAGU,yBAIR;CACA;CAEA;CAEA;CAEA,YACEa,QACAd,QACAC,QACA;EACA,MAAM,QAAQ,QAAQ,OAAO;EAC7B,KAAK,MAAM,IAAI,sBAAsB,OAAO;EAC5C,KAAK,SAAS,OAAO;EACrB,KAAK,MAAM,OAAO;CACnB;CAED,cAA+C;AAC7C,SAAO,KAAK;CACb;CAED,MAAM,WAA4B;EAChC,MAAM,OAAO,mBAAmB,KAAK,IAAI,KAAK;EAC9C,MAAM,MAAM,CAAC,uCAAuC,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,IAAI,OAAO,GAAG,EAAE,MAAM;AACtG,SAAO,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,KAAK,KAAK,GAAG;CAC9C;AACF;AAOD,IAAa,kCAAb,cAEU,4BAAiE;CACzE;CAEA,YACEc,QACAf,QACAC,QACA;EACA,MAAM,QAAQ,QAAQ,OAAO;EAC7B,KAAK,MAAM,IAAI,sBAAsB,OAAO;CAC7C;CAED,MAAM,WAA4B;EAChC,MAAM,OAAO,mBAAmB,KAAK,IAAI,KAAK;EAC9C,MAAM,MAAM,CAAC,uCAAuC,EAAE,KAAK,WAAW,GAAG,EAAE,KAAK,IAAI,OAAO,GAAG,EAAE,KAAK,UAAU,CAAC;AAChH,SAAO;CACR;AACF;AAQD,IAAsB,kCAAtB,cAEU,gBAAyD;CACjE;CAEA,YAAYe,QAA4D;EACtE,MAAM,OAAO;EACb,KAAK,SAAS;EACd,KAAK,sBAAsB;GACzB,GAAG,KAAK;GACR,YAAY,OAAO,UAAU;EAC9B;CACF;CAED,mBAAmB,CAAC,KAAK,MAGxB,EAIC;EACA,MAAMC,SAAgE;GACpE,GAAG,KAAK;GACR,KAAK;EACN;AACD,SAAO,IAAI,mCACT,QACA,KAAK,QACL,KAAK;CAER;CAED,iBAAiB,CAAC,KAAK,KAA0B,EAA2B;EAC1E,MAAM,MAAM,IAAI,sBAAsB;EACtC,MAAMC,MAAgC;GACpC,MAAM,IAAI;GACV,UAAU,KAAK;GACf,aAAa,KAAK;EACnB;AACD,SAAO;CACR;CAED,2BACET,KAKA;EACA,MAAMU,SAAkE;GACtE,KAAK;GACL,QAAQ;GACR,KAAK;EACN;AACD,SAAO,IAAI,qCAGT,QAAQ,KAAK,QAAQ,KAAK;CAC7B;CAED,uBACEV,KACkE;EAClE,MAAMW,SAA6D,EACjE,KAAK,IACN;AACD,SAAO,IAAI,gCACT,QACA,KAAK,QACL,KAAK;CAER;CAED,sBACEX,KAKA;EACA,MAAMU,SAAkE;GACtE,KAAK;GACL,QAAQ;GACR,KAAK;EACN;AACD,SAAO,IAAI,qCAGT,QAAQ,KAAK,QAAQ,KAAK;CAC7B;AACF;AAmCD,IAAa,oBAAb,cAAuC,UAAU;CAC/C,aAAaE,OAAqB;AAChC,MAAI,CAAC,MACH,wBAAO,IAAI,KAAK;AAElB,SAAO,IAAI,KAAK;CACjB;CAED,qBAAqBC,OAAqB;AACxC,SAAO,KAAK,aAAa,KAAK,WAAW,OAAO;CACjD;CAED,IAAI,aAAmB;AACrB,SAAO,KAAK,qBAAqB,aAAa;CAC/C;CAED,IAAI,aAAmB;AACrB,SAAO,KAAK,qBAAqB,aAAa;CAC/C;CAED,IAAI,iBAAuB;AACzB,SAAO,KAAK,qBAAqB,iBAAiB;CACnD;CAED,IAAI,YAAqB;EACvB,MAAM,uBAAM,IAAI,QAAO,aAAa;EACpC,MAAM,MAAM,KAAK,UAAU,kBAAkB;AAC7C,SAAO,OAAO;CACf;AACF;AA+BD,IAAa,+BAAb,cAEU,gCAIR;CACA,IAAI,qBAA6B;AAC/B,SAAO;CACR;CAED,MAAM,WAA4B;AAChC,SAAO,CAAC,iDAAiD,EAAE,KAAK,WAAW,MAAM,CAAC;CACnF;AACF;AAQD,IAAa,iCAAb,cAGU,yBAIR;CACA;CAEA;CAEA,YACEC,QACAvB,QACAC,QACA;EACA,MAAM,QAAQ,QAAQ,OAAO;EAC7B,KAAK,SAAS,OAAO;EACrB,KAAK,OAAO,OAAO;CACpB;CAED,IAAI,qBAA6B;AAC/B,SAAO;CACR;CAED,cAA+C;AAC7C,SAAO,KAAK;CACb;CAED,MAAM,WAA4B;AAChC,SAAO,CAAC,0CAA0C,EAAE,KAAK,WAAW,OAAO,EAAE,KAAK,MAAM;CACzF;AACF;AAQD,IAAsB,4BAAtB,cAEU,gBAAmD;CAC3D;CAEA,YAAoB;CAEpB,YAAYuB,QAAuD;EACjE,MAAMC,SAAuD;GAC3D,qBAAqB;IACnB,YAAY;IACZ,iBAAiB;GAClB;GACD,GAAG;EACJ;EACD,MAAM,OAAO;EACb,KAAK,SAAS;EACd,KAAK,YAAY,QAAQ,aAAa,KAAK,aAAa;CACzD;CAED,YAAYC,MAAsB;AAChC,SAAO,KAAK,MAAM,IAAI,CAAC,KAAK,IAAI;CACjC;CAMD,kBAAkBC,QAAwC;AACxD,SAAO,IAAI,iBAAiB;CAC7B;CAED,YAAYrB,QAAiE;AAC3E,SAAO,QAAQ,UAAU,uBAAuB,iBAAiB;CAClE;CAED,YACEA,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAGrC,QAAO,KAAK,sBAAsB,OAAO;CAE5C;CAED,MAAM,eAAeG,KAA0C;EAE7D,MAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,KAAK,UAAU;AAGlE,SAAO,KAAK,aAAa,IAAI;CAC9B;CAED,MAAM,KAAK,CAAC,KAAK,KAGhB,EAAqC;EACpC,MAAM,WAAY,MAAM,MAAM,KAAK,CACjC,KACA,IACD,EAAC;EAEF,IAAI,OAAO,SAAS,MAAM,QAAQ,EAAE,OAAO,SAAU;AACrD,SAAO,KAAK,UAAU,gBAAgB,KAAK,OAAO,KAAK,YAAY,GACjE,OAAO,MAAM,KAAK,eAAe,KAAK,IAAI;EAM5C,KAAK,OAAO,KAAK;EACjB,KAAK,WAAW;GACd,GAAG,KAAK;GACR,GAAG;EACJ;AACD,SAAO;CACR;CAED,mBAAmB,CAAC,MAAM,MAGzB,EAIC;AACA,SAAO,IAAI,6BACT,KAAK,QACL,KAAK,QACL,KAAK;CAER;CAED,iBAAiB,CAAC,MAAM,MAA2B,EAGjD;AACA,SAAO,CAAE;CACV;CAED,2BACEA,KAKA;EACA,MAAMmB,SAA4D;GAChE,GAAG,KAAK;GACR,QAAQ;GACR,MAAM,KAAK,YAAY,IAAI;EAC5B;AACD,SAAO,IAAI,+BAGT,QAAQ,KAAK,QAAQ,KAAK;CAC7B;CAED,uBACEC,MACkE;AAClE,QAAM,IAAI,MAAM;CACjB;CAED,MAAM,KAAKpB,KAA6C;EACtD,MAAM,WAAW,MAAM,KAAK,aAAa,IAAI;AAC7C,MAAI,UAAU;GACZ,MAAM,cACH,UAAU,YAAuB;GAEpC,MAAMqB,OAAsB;IAC1B,OAAO;IACP,MAAM;GACP;AAED,UAAO,IAAI,UAAU;IACnB,MAAM;IACN;IACA;GACD;EACF,MACC,QAAO;CAEV;CAED,sBACErB,KAKA;EACA,MAAMmB,SAA4D;GAChE,GAAG,KAAK;GACR,QAAQ;GACR,MAAM,KAAK,YAAY,IAAI;EAC5B;AACD,SAAO,IAAI,+BAGT,QAAQ,KAAK,QAAQ,KAAK;CAC7B;AACF"}
+{"version":3,"file":"media.js","names":[],"sources":["../../src/experimental/media.ts"],"sourcesContent":["import {\n  AsyncCaller,\n  AsyncCallerCallOptions,\n  AsyncCallerParams,\n} from \"@langchain/core/utils/async_caller\";\nimport { getEnvironmentVariable } from \"@langchain/core/utils/env\";\nimport {\n  MediaBlob,\n  BlobStore,\n  BlobStoreOptions,\n  MediaBlobData,\n} from \"./utils/media_core.js\";\nimport {\n  GoogleConnectionParams,\n  GoogleRawResponse,\n  GoogleResponse,\n} from \"../types.js\";\nimport { GoogleHostConnection, GoogleRawConnection } from \"../connection.js\";\nimport {\n  ApiKeyGoogleAuth,\n  GoogleAbstractedClient,\n  GoogleAbstractedClientOpsMethod,\n} from \"../auth.js\";\n\nexport interface GoogleUploadConnectionParams<\n  AuthOptions,\n> extends GoogleConnectionParams<AuthOptions> {}\n\nexport abstract class GoogleMultipartUploadConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleHostConnection<CallOptions, ResponseType, AuthOptions> {\n  constructor(\n    fields: GoogleConnectionParams<AuthOptions> | undefined,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n  }\n\n  async _body(\n    separator: string,\n    data: MediaBlob,\n    metadata: Record<string, unknown>\n  ): Promise<string> {\n    const contentType = data.mimetype;\n    const { encoded, encoding } = await data.encode();\n    const body = [\n      `--${separator}`,\n      \"Content-Type: application/json; charset=UTF-8\",\n      \"\",\n      JSON.stringify(metadata),\n      \"\",\n      `--${separator}`,\n      `Content-Type: ${contentType}`,\n      `Content-Transfer-Encoding: ${encoding}`,\n      \"\",\n      encoded,\n      `--${separator}--`,\n    ];\n    return body.join(\"\\n\");\n  }\n\n  async request(\n    data: MediaBlob,\n    metadata: Record<string, unknown>,\n    options: CallOptions\n  ): Promise<ResponseType> {\n    const separator = `separator-${Date.now()}`;\n    const body = await this._body(separator, data, metadata);\n    const requestHeaders = {\n      \"Content-Type\": `multipart/related; boundary=${separator}`,\n      \"X-Goog-Upload-Protocol\": \"multipart\",\n    };\n    const response = this._request(body, options, requestHeaders);\n    return response;\n  }\n}\n\nexport abstract class GoogleDownloadConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleHostConnection<CallOptions, ResponseType, AuthOptions> {\n  async request(options: CallOptions): Promise<ResponseType> {\n    return this._request(undefined, options);\n  }\n}\n\nexport abstract class GoogleDownloadRawConnection<\n  CallOptions extends AsyncCallerCallOptions,\n  AuthOptions,\n> extends GoogleRawConnection<CallOptions, AuthOptions> {\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return \"GET\";\n  }\n\n  async request(options: CallOptions): Promise<GoogleRawResponse> {\n    return this._request(undefined, options);\n  }\n}\n\nexport interface BlobStoreGoogleParams<AuthOptions>\n  extends\n    GoogleConnectionParams<AuthOptions>,\n    AsyncCallerParams,\n    BlobStoreOptions {}\n\nexport abstract class BlobStoreGoogle<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends BlobStore {\n  caller: AsyncCaller;\n\n  client: GoogleAbstractedClient;\n\n  constructor(fields?: BlobStoreGoogleParams<AuthOptions>) {\n    super(fields);\n    this.caller = new AsyncCaller(fields ?? {});\n    this.client = this.buildClient(fields);\n  }\n\n  abstract buildClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  abstract buildSetMetadata([key, blob]: [string, MediaBlob]): Record<\n    string,\n    unknown\n  >;\n\n  abstract buildSetConnection([key, blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    ResponseType,\n    AuthOptions\n  >;\n\n  async _set(keyValuePair: [string, MediaBlob]): Promise<ResponseType> {\n    const [, blob] = keyValuePair;\n    const setMetadata = this.buildSetMetadata(keyValuePair);\n    const metadata = setMetadata;\n    const options = {};\n    const connection = this.buildSetConnection(keyValuePair);\n    const response = await connection.request(blob, metadata, options);\n    return response;\n  }\n\n  async mset(keyValuePairs: [string, MediaBlob][]): Promise<void> {\n    const ret = keyValuePairs.map((keyValue) => this._set(keyValue));\n    await Promise.all(ret);\n  }\n\n  abstract buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    ResponseType,\n    AuthOptions\n  >;\n\n  async _getMetadata(key: string): Promise<Record<string, unknown>> {\n    const connection = this.buildGetMetadataConnection(key);\n    const options = {};\n    const response = await connection.request(options);\n    return response.data;\n  }\n\n  abstract buildGetDataConnection(\n    key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions>;\n\n  async _getData(key: string): Promise<Blob> {\n    const connection = this.buildGetDataConnection(key);\n    const options = {};\n    const response = await connection.request(options);\n    return response.data;\n  }\n\n  _getMimetypeFromMetadata(metadata: Record<string, unknown>): string {\n    return metadata.contentType as string;\n  }\n\n  async _get(key: string): Promise<MediaBlob | undefined> {\n    const metadata = await this._getMetadata(key);\n    const data = await this._getData(key);\n    if (data && metadata) {\n      const ret = await MediaBlob.fromBlob(data, { metadata, path: key });\n      return ret;\n    } else {\n      return undefined;\n    }\n  }\n\n  async mget(keys: string[]): Promise<(MediaBlob | undefined)[]> {\n    const ret = keys.map((key) => this._get(key));\n    return await Promise.all(ret);\n  }\n\n  abstract buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleResponse,\n    AuthOptions\n  >;\n\n  async _del(key: string): Promise<void> {\n    const connection = this.buildDeleteConnection(key);\n    const options = {};\n    await connection.request(options);\n  }\n\n  async mdelete(keys: string[]): Promise<void> {\n    const ret = keys.map((key) => this._del(key));\n    await Promise.all(ret);\n  }\n\n  // eslint-disable-next-line require-yield\n  async *yieldKeys(_prefix: string | undefined): AsyncGenerator<string> {\n    // TODO: Implement. Most have an implementation that uses nextToken.\n    throw new Error(\"yieldKeys is not implemented\");\n  }\n}\n\n/**\n * Based on https://cloud.google.com/storage/docs/json_api/v1/objects#resource\n */\nexport interface GoogleCloudStorageObject extends Record<string, unknown> {\n  id?: string;\n  name?: string;\n  contentType?: string;\n  metadata?: Record<string, unknown>;\n  // This is incomplete.\n}\n\nexport interface GoogleCloudStorageResponse extends GoogleResponse {\n  data: GoogleCloudStorageObject;\n}\n\nexport type BucketAndPath = {\n  bucket: string;\n  path: string;\n};\n\nexport class GoogleCloudStorageUri {\n  static uriRegexp = /gs:\\/\\/([a-z0-9][a-z0-9._-]+[a-z0-9])\\/(.*)/;\n\n  bucket: string;\n\n  path: string;\n\n  constructor(uri: string) {\n    const bucketAndPath = GoogleCloudStorageUri.uriToBucketAndPath(uri);\n    this.bucket = bucketAndPath.bucket;\n    this.path = bucketAndPath.path;\n  }\n\n  get uri() {\n    return `gs://${this.bucket}/${this.path}`;\n  }\n\n  get isValid() {\n    return (\n      typeof this.bucket !== \"undefined\" && typeof this.path !== \"undefined\"\n    );\n  }\n\n  static uriToBucketAndPath(uri: string): BucketAndPath {\n    const match = this.uriRegexp.exec(uri);\n    if (!match) {\n      throw new Error(`Invalid gs:// URI: ${uri}`);\n    }\n    return {\n      bucket: match[1],\n      path: match[2],\n    };\n  }\n\n  static isValidUri(uri: string): boolean {\n    return this.uriRegexp.test(uri);\n  }\n}\n\nexport interface GoogleCloudStorageConnectionParams {\n  uri: string;\n}\n\nexport interface GoogleCloudStorageUploadConnectionParams<AuthOptions>\n  extends\n    GoogleUploadConnectionParams<AuthOptions>,\n    GoogleCloudStorageConnectionParams {}\n\nexport class GoogleCloudStorageUploadConnection<\n  AuthOptions,\n> extends GoogleMultipartUploadConnection<\n  AsyncCallerCallOptions,\n  GoogleCloudStorageResponse,\n  AuthOptions\n> {\n  uri: GoogleCloudStorageUri;\n\n  constructor(\n    fields: GoogleCloudStorageUploadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://storage.googleapis.com/upload/storage/${this.apiVersion}/b/${this.uri.bucket}/o?uploadType=multipart`;\n  }\n}\n\nexport interface GoogleCloudStorageDownloadConnectionParams<AuthOptions>\n  extends\n    GoogleCloudStorageConnectionParams,\n    GoogleConnectionParams<AuthOptions> {\n  method: GoogleAbstractedClientOpsMethod;\n  alt: \"media\" | undefined;\n}\n\nexport class GoogleCloudStorageDownloadConnection<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleDownloadConnection<\n  AsyncCallerCallOptions,\n  ResponseType,\n  AuthOptions\n> {\n  uri: GoogleCloudStorageUri;\n\n  method: GoogleAbstractedClientOpsMethod;\n\n  alt: \"media\" | undefined;\n\n  constructor(\n    fields: GoogleCloudStorageDownloadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n    this.method = fields.method;\n    this.alt = fields.alt;\n  }\n\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return this.method;\n  }\n\n  async buildUrl(): Promise<string> {\n    const path = encodeURIComponent(this.uri.path);\n    const ret = `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}`;\n    return this.alt ? `${ret}?alt=${this.alt}` : ret;\n  }\n}\n\nexport interface GoogleCloudStorageRawConnectionParams<AuthOptions>\n  extends\n    GoogleCloudStorageConnectionParams,\n    GoogleConnectionParams<AuthOptions> {}\n\nexport class GoogleCloudStorageRawConnection<\n  AuthOptions,\n> extends GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n  uri: GoogleCloudStorageUri;\n\n  constructor(\n    fields: GoogleCloudStorageRawConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.uri = new GoogleCloudStorageUri(fields.uri);\n  }\n\n  async buildUrl(): Promise<string> {\n    const path = encodeURIComponent(this.uri.path);\n    const ret = `https://storage.googleapis.com/storage/${this.apiVersion}/b/${this.uri.bucket}/o/${path}?alt=media`;\n    return ret;\n  }\n}\n\nexport interface BlobStoreGoogleCloudStorageBaseParams<\n  AuthOptions,\n> extends BlobStoreGoogleParams<AuthOptions> {\n  uriPrefix: GoogleCloudStorageUri;\n}\n\nexport abstract class BlobStoreGoogleCloudStorageBase<\n  AuthOptions,\n> extends BlobStoreGoogle<GoogleCloudStorageResponse, AuthOptions> {\n  params: BlobStoreGoogleCloudStorageBaseParams<AuthOptions>;\n\n  constructor(fields: BlobStoreGoogleCloudStorageBaseParams<AuthOptions>) {\n    super(fields);\n    this.params = fields;\n    this.defaultStoreOptions = {\n      ...this.defaultStoreOptions,\n      pathPrefix: fields.uriPrefix.uri,\n    };\n  }\n\n  buildSetConnection([key, _blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    GoogleCloudStorageResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageUploadConnectionParams<AuthOptions> = {\n      ...this.params,\n      uri: key,\n    };\n    return new GoogleCloudStorageUploadConnection<AuthOptions>(\n      params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildSetMetadata([key, blob]: [string, MediaBlob]): Record<string, unknown> {\n    const uri = new GoogleCloudStorageUri(key);\n    const ret: GoogleCloudStorageObject = {\n      name: uri.path,\n      metadata: blob.metadata,\n      contentType: blob.mimetype,\n    };\n    return ret;\n  }\n\n  buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleCloudStorageResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageDownloadConnectionParams<AuthOptions> = {\n      uri: key,\n      method: \"GET\",\n      alt: undefined,\n    };\n    return new GoogleCloudStorageDownloadConnection<\n      GoogleCloudStorageResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n\n  buildGetDataConnection(\n    key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n    const params: GoogleCloudStorageRawConnectionParams<AuthOptions> = {\n      uri: key,\n    };\n    return new GoogleCloudStorageRawConnection<AuthOptions>(\n      params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    GoogleResponse,\n    AuthOptions\n  > {\n    const params: GoogleCloudStorageDownloadConnectionParams<AuthOptions> = {\n      uri: key,\n      method: \"DELETE\",\n      alt: undefined,\n    };\n    return new GoogleCloudStorageDownloadConnection<\n      GoogleResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n}\n\nexport type AIStudioFileState =\n  | \"PROCESSING\"\n  | \"ACTIVE\"\n  | \"FAILED\"\n  | \"STATE_UNSPECIFIED\";\n\nexport type AIStudioFileVideoMetadata = {\n  videoMetadata: {\n    videoDuration: string; // Duration in seconds, possibly with fractional, ending in \"s\"\n  };\n};\n\nexport type AIStudioFileMetadata = AIStudioFileVideoMetadata;\n\nexport interface AIStudioFileObject {\n  name?: string;\n  displayName?: string;\n  mimeType?: string;\n  sizeBytes?: string; // int64 format\n  createTime?: string; // timestamp format\n  updateTime?: string; // timestamp format\n  expirationTime?: string; // timestamp format\n  sha256Hash?: string; // base64 encoded\n  uri?: string;\n  state?: AIStudioFileState;\n  error?: {\n    code: number;\n    message: string;\n    details: Record<string, unknown>[];\n  };\n  metadata?: AIStudioFileMetadata;\n}\n\nexport class AIStudioMediaBlob extends MediaBlob {\n  _valueAsDate(value: string): Date {\n    if (!value) {\n      return new Date(0);\n    }\n    return new Date(value);\n  }\n\n  _metadataFieldAsDate(field: string): Date {\n    return this._valueAsDate(this.metadata?.[field]);\n  }\n\n  get createDate(): Date {\n    return this._metadataFieldAsDate(\"createTime\");\n  }\n\n  get updateDate(): Date {\n    return this._metadataFieldAsDate(\"updateTime\");\n  }\n\n  get expirationDate(): Date {\n    return this._metadataFieldAsDate(\"expirationTime\");\n  }\n\n  get isExpired(): boolean {\n    const now = new Date().toISOString();\n    const exp = this.metadata?.expirationTime ?? now;\n    return exp <= now;\n  }\n}\n\nexport interface AIStudioFileGetResponse extends GoogleResponse {\n  data: AIStudioFileObject;\n}\n\nexport interface AIStudioFileSaveResponse extends GoogleResponse {\n  data: {\n    file: AIStudioFileObject;\n  };\n}\n\nexport interface AIStudioFileListResponse extends GoogleResponse {\n  data: {\n    files: AIStudioFileObject[];\n    nextPageToken: string;\n  };\n}\n\nexport type AIStudioFileResponse =\n  | AIStudioFileGetResponse\n  | AIStudioFileSaveResponse\n  | AIStudioFileListResponse;\n\nexport interface AIStudioFileConnectionParams {}\n\nexport interface AIStudioFileUploadConnectionParams<AuthOptions>\n  extends\n    GoogleUploadConnectionParams<AuthOptions>,\n    AIStudioFileConnectionParams {}\n\nexport class AIStudioFileUploadConnection<\n  AuthOptions,\n> extends GoogleMultipartUploadConnection<\n  AsyncCallerCallOptions,\n  AIStudioFileSaveResponse,\n  AuthOptions\n> {\n  get computedApiVersion(): string {\n    return \"v1beta\";\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://generativelanguage.googleapis.com/upload/${this.apiVersion}/files`;\n  }\n}\n\nexport interface AIStudioFileDownloadConnectionParams<AuthOptions>\n  extends AIStudioFileConnectionParams, GoogleConnectionParams<AuthOptions> {\n  method: GoogleAbstractedClientOpsMethod;\n  name: string;\n}\n\nexport class AIStudioFileDownloadConnection<\n  ResponseType extends GoogleResponse,\n  AuthOptions,\n> extends GoogleDownloadConnection<\n  AsyncCallerCallOptions,\n  ResponseType,\n  AuthOptions\n> {\n  method: GoogleAbstractedClientOpsMethod;\n\n  name: string;\n\n  constructor(\n    fields: AIStudioFileDownloadConnectionParams<AuthOptions>,\n    caller: AsyncCaller,\n    client: GoogleAbstractedClient\n  ) {\n    super(fields, caller, client);\n    this.method = fields.method;\n    this.name = fields.name;\n  }\n\n  get computedApiVersion(): string {\n    return \"v1beta\";\n  }\n\n  buildMethod(): GoogleAbstractedClientOpsMethod {\n    return this.method;\n  }\n\n  async buildUrl(): Promise<string> {\n    return `https://generativelanguage.googleapis.com/${this.apiVersion}/files/${this.name}`;\n  }\n}\n\nexport interface BlobStoreAIStudioFileBaseParams<\n  AuthOptions,\n> extends BlobStoreGoogleParams<AuthOptions> {\n  retryTime?: number;\n}\n\nexport abstract class BlobStoreAIStudioFileBase<\n  AuthOptions,\n> extends BlobStoreGoogle<AIStudioFileResponse, AuthOptions> {\n  params?: BlobStoreAIStudioFileBaseParams<AuthOptions>;\n\n  retryTime: number = 1000;\n\n  constructor(fields?: BlobStoreAIStudioFileBaseParams<AuthOptions>) {\n    const params: BlobStoreAIStudioFileBaseParams<AuthOptions> = {\n      defaultStoreOptions: {\n        pathPrefix: \"https://generativelanguage.googleapis.com/v1beta/files/\",\n        actionIfInvalid: \"removePath\",\n      },\n      ...fields,\n    };\n    super(params);\n    this.params = params;\n    this.retryTime = params?.retryTime ?? this.retryTime ?? 1000;\n  }\n\n  _pathToName(path: string): string {\n    return path.split(\"/\").pop() ?? path;\n  }\n\n  abstract buildAbstractedClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient;\n\n  buildApiKeyClient(apiKey: string): GoogleAbstractedClient {\n    return new ApiKeyGoogleAuth(apiKey);\n  }\n\n  buildApiKey(fields?: BlobStoreGoogleParams<AuthOptions>): string | undefined {\n    return fields?.apiKey ?? getEnvironmentVariable(\"GOOGLE_API_KEY\");\n  }\n\n  buildClient(\n    fields?: BlobStoreGoogleParams<AuthOptions>\n  ): GoogleAbstractedClient {\n    const apiKey = this.buildApiKey(fields);\n    if (apiKey) {\n      return this.buildApiKeyClient(apiKey);\n    } else {\n      // TODO: Test that you can use OAuth to access\n      return this.buildAbstractedClient(fields);\n    }\n  }\n\n  async _regetMetadata(key: string): Promise<AIStudioFileObject> {\n    // Sleep for some time period\n    await new Promise((resolve) => setTimeout(resolve, this.retryTime));\n\n    // Fetch the latest metadata\n    return this._getMetadata(key);\n  }\n\n  async _set([key, blob]: [\n    string,\n    MediaBlob,\n  ]): Promise<AIStudioFileSaveResponse> {\n    const response = (await super._set([\n      key,\n      blob,\n    ])) as AIStudioFileSaveResponse;\n\n    let file = response.data?.file ?? { state: \"FAILED\" };\n    while (file.state === \"PROCESSING\" && file.uri && this.retryTime > 0) {\n      file = await this._regetMetadata(file.uri);\n    }\n\n    // The response should contain the name (and valid URI), so we need to\n    // update the blob with this. We can't return a new blob, since mset()\n    // doesn't return anything.\n    blob.path = file.uri;\n    blob.metadata = {\n      ...blob.metadata,\n      ...file,\n    };\n    return response;\n  }\n\n  buildSetConnection([_key, _blob]: [\n    string,\n    MediaBlob,\n  ]): GoogleMultipartUploadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    return new AIStudioFileUploadConnection(\n      this.params,\n      this.caller,\n      this.client\n    );\n  }\n\n  buildSetMetadata([_key, _blob]: [string, MediaBlob]): Record<\n    string,\n    unknown\n  > {\n    return {};\n  }\n\n  buildGetMetadataConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    const params: AIStudioFileDownloadConnectionParams<AuthOptions> = {\n      ...this.params,\n      method: \"GET\",\n      name: this._pathToName(key),\n    };\n    return new AIStudioFileDownloadConnection<\n      AIStudioFileResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n\n  buildGetDataConnection(\n    _key: string\n  ): GoogleDownloadRawConnection<AsyncCallerCallOptions, AuthOptions> {\n    throw new Error(\"AI Studio File API does not provide data\");\n  }\n\n  async _get(key: string): Promise<MediaBlob | undefined> {\n    const metadata = await this._getMetadata(key);\n    if (metadata) {\n      const contentType =\n        (metadata?.mimeType as string) ?? \"application/octet-stream\";\n      // TODO - Get the actual data (and other metadata) from an optional backing store\n      const data: MediaBlobData = {\n        value: \"\",\n        type: contentType,\n      };\n\n      return new MediaBlob({\n        path: key,\n        data,\n        metadata,\n      });\n    } else {\n      return undefined;\n    }\n  }\n\n  buildDeleteConnection(\n    key: string\n  ): GoogleDownloadConnection<\n    AsyncCallerCallOptions,\n    AIStudioFileResponse,\n    AuthOptions\n  > {\n    const params: AIStudioFileDownloadConnectionParams<AuthOptions> = {\n      ...this.params,\n      method: \"DELETE\",\n      name: this._pathToName(key),\n    };\n    return new AIStudioFileDownloadConnection<\n      AIStudioFileResponse,\n      AuthOptions\n    >(params, this.caller, this.client);\n  }\n}\n"],"mappings":";;;;;;;AA4BA,IAAsB,kCAAtB,cAIU,qBAA6D;CACrE,YACE,QACA,QACA,QACA;AACA,QAAM,QAAQ,QAAQ,OAAO;;CAG/B,MAAM,MACJ,WACA,MACA,UACiB;EACjB,MAAM,cAAc,KAAK;EACzB,MAAM,EAAE,SAAS,aAAa,MAAM,KAAK,QAAQ;AAcjD,SAba;GACX,KAAK;GACL;GACA;GACA,KAAK,UAAU,SAAS;GACxB;GACA,KAAK;GACL,iBAAiB;GACjB,8BAA8B;GAC9B;GACA;GACA,KAAK,UAAU;GAChB,CACW,KAAK,KAAK;;CAGxB,MAAM,QACJ,MACA,UACA,SACuB;EACvB,MAAM,YAAY,aAAa,KAAK,KAAK;EACzC,MAAM,OAAO,MAAM,KAAK,MAAM,WAAW,MAAM,SAAS;EACxD,MAAM,iBAAiB;GACrB,gBAAgB,+BAA+B;GAC/C,0BAA0B;GAC3B;AAED,SADiB,KAAK,SAAS,MAAM,SAAS,eAAe;;;AAKjE,IAAsB,2BAAtB,cAIU,qBAA6D;CACrE,MAAM,QAAQ,SAA6C;AACzD,SAAO,KAAK,SAAS,QAAW,QAAQ;;;AAI5C,IAAsB,8BAAtB,cAGU,oBAA8C;CACtD,cAA+C;AAC7C,SAAO;;CAGT,MAAM,QAAQ,SAAkD;AAC9D,SAAO,KAAK,SAAS,QAAW,QAAQ;;;AAU5C,IAAsB,kBAAtB,cAGU,UAAU;CAClB;CAEA;CAEA,YAAY,QAA6C;AACvD,QAAM,OAAO;AACb,OAAK,SAAS,IAAI,YAAY,UAAU,EAAE,CAAC;AAC3C,OAAK,SAAS,KAAK,YAAY,OAAO;;CAqBxC,MAAM,KAAK,cAA0D;EACnE,MAAM,GAAG,QAAQ;EAEjB,MAAM,WADc,KAAK,iBAAiB,aAAa;AAKvD,SADiB,MADE,KAAK,mBAAmB,aAAa,CACtB,QAAQ,MAAM,UAFhC,EAAE,CAEgD;;CAIpE,MAAM,KAAK,eAAqD;EAC9D,MAAM,MAAM,cAAc,KAAK,aAAa,KAAK,KAAK,SAAS,CAAC;AAChE,QAAM,QAAQ,IAAI,IAAI;;CAWxB,MAAM,aAAa,KAA+C;AAIhE,UADiB,MAFE,KAAK,2BAA2B,IAAI,CAErB,QADlB,EAAE,CACgC,EAClC;;CAOlB,MAAM,SAAS,KAA4B;AAIzC,UADiB,MAFE,KAAK,uBAAuB,IAAI,CAEjB,QADlB,EAAE,CACgC,EAClC;;CAGlB,yBAAyB,UAA2C;AAClE,SAAO,SAAS;;CAGlB,MAAM,KAAK,KAA6C;EACtD,MAAM,WAAW,MAAM,KAAK,aAAa,IAAI;EAC7C,MAAM,OAAO,MAAM,KAAK,SAAS,IAAI;AACrC,MAAI,QAAQ,SAEV,QADY,MAAM,UAAU,SAAS,MAAM;GAAE;GAAU,MAAM;GAAK,CAAC;MAGnE;;CAIJ,MAAM,KAAK,MAAoD;EAC7D,MAAM,MAAM,KAAK,KAAK,QAAQ,KAAK,KAAK,IAAI,CAAC;AAC7C,SAAO,MAAM,QAAQ,IAAI,IAAI;;CAW/B,MAAM,KAAK,KAA4B;AAGrC,QAFmB,KAAK,sBAAsB,IAAI,CAEjC,QADD,EAAE,CACe;;CAGnC,MAAM,QAAQ,MAA+B;EAC3C,MAAM,MAAM,KAAK,KAAK,QAAQ,KAAK,KAAK,IAAI,CAAC;AAC7C,QAAM,QAAQ,IAAI,IAAI;;CAIxB,OAAO,UAAU,SAAqD;AAEpE,QAAM,IAAI,MAAM,+BAA+B;;;AAwBnD,IAAa,wBAAb,MAAa,sBAAsB;CACjC,OAAO,YAAY;CAEnB;CAEA;CAEA,YAAY,KAAa;EACvB,MAAM,gBAAgB,sBAAsB,mBAAmB,IAAI;AACnE,OAAK,SAAS,cAAc;AAC5B,OAAK,OAAO,cAAc;;CAG5B,IAAI,MAAM;AACR,SAAO,QAAQ,KAAK,OAAO,GAAG,KAAK;;CAGrC,IAAI,UAAU;AACZ,SACE,OAAO,KAAK,WAAW,eAAe,OAAO,KAAK,SAAS;;CAI/D,OAAO,mBAAmB,KAA4B;EACpD,MAAM,QAAQ,KAAK,UAAU,KAAK,IAAI;AACtC,MAAI,CAAC,MACH,OAAM,IAAI,MAAM,sBAAsB,MAAM;AAE9C,SAAO;GACL,QAAQ,MAAM;GACd,MAAM,MAAM;GACb;;CAGH,OAAO,WAAW,KAAsB;AACtC,SAAO,KAAK,UAAU,KAAK,IAAI;;;AAanC,IAAa,qCAAb,cAEU,gCAIR;CACA;CAEA,YACE,QACA,QACA,QACA;AACA,QAAM,QAAQ,QAAQ,OAAO;AAC7B,OAAK,MAAM,IAAI,sBAAsB,OAAO,IAAI;;CAGlD,MAAM,WAA4B;AAChC,SAAO,iDAAiD,KAAK,WAAW,KAAK,KAAK,IAAI,OAAO;;;AAYjG,IAAa,uCAAb,cAGU,yBAIR;CACA;CAEA;CAEA;CAEA,YACE,QACA,QACA,QACA;AACA,QAAM,QAAQ,QAAQ,OAAO;AAC7B,OAAK,MAAM,IAAI,sBAAsB,OAAO,IAAI;AAChD,OAAK,SAAS,OAAO;AACrB,OAAK,MAAM,OAAO;;CAGpB,cAA+C;AAC7C,SAAO,KAAK;;CAGd,MAAM,WAA4B;EAChC,MAAM,OAAO,mBAAmB,KAAK,IAAI,KAAK;EAC9C,MAAM,MAAM,0CAA0C,KAAK,WAAW,KAAK,KAAK,IAAI,OAAO,KAAK;AAChG,SAAO,KAAK,MAAM,GAAG,IAAI,OAAO,KAAK,QAAQ;;;AASjD,IAAa,kCAAb,cAEU,4BAAiE;CACzE;CAEA,YACE,QACA,QACA,QACA;AACA,QAAM,QAAQ,QAAQ,OAAO;AAC7B,OAAK,MAAM,IAAI,sBAAsB,OAAO,IAAI;;CAGlD,MAAM,WAA4B;EAChC,MAAM,OAAO,mBAAmB,KAAK,IAAI,KAAK;AAE9C,SADY,0CAA0C,KAAK,WAAW,KAAK,KAAK,IAAI,OAAO,KAAK,KAAK;;;AAWzG,IAAsB,kCAAtB,cAEU,gBAAyD;CACjE;CAEA,YAAY,QAA4D;AACtE,QAAM,OAAO;AACb,OAAK,SAAS;AACd,OAAK,sBAAsB;GACzB,GAAG,KAAK;GACR,YAAY,OAAO,UAAU;GAC9B;;CAGH,mBAAmB,CAAC,KAAK,QAOvB;AAKA,SAAO,IAAI,mCAJ2D;GACpE,GAAG,KAAK;GACR,KAAK;GACN,EAGC,KAAK,QACL,KAAK,OACN;;CAGH,iBAAiB,CAAC,KAAK,OAAqD;AAO1E,SALsC;GACpC,MAFU,IAAI,sBAAsB,IAAI,CAE9B;GACV,UAAU,KAAK;GACf,aAAa,KAAK;GACnB;;CAIH,2BACE,KAKA;AAMA,SAAO,IAAI,qCAL6D;GACtE,KAAK;GACL,QAAQ;GACR,KAAK;GACN,EAIS,KAAK,QAAQ,KAAK,OAAO;;CAGrC,uBACE,KACkE;AAIlE,SAAO,IAAI,gCAHwD,EACjE,KAAK,KACN,EAGC,KAAK,QACL,KAAK,OACN;;CAGH,sBACE,KAKA;AAMA,SAAO,IAAI,qCAL6D;GACtE,KAAK;GACL,QAAQ;GACR,KAAK;GACN,EAIS,KAAK,QAAQ,KAAK,OAAO;;;AAqCvC,IAAa,oBAAb,cAAuC,UAAU;CAC/C,aAAa,OAAqB;AAChC,MAAI,CAAC,MACH,wBAAO,IAAI,KAAK,EAAE;AAEpB,SAAO,IAAI,KAAK,MAAM;;CAGxB,qBAAqB,OAAqB;AACxC,SAAO,KAAK,aAAa,KAAK,WAAW,OAAO;;CAGlD,IAAI,aAAmB;AACrB,SAAO,KAAK,qBAAqB,aAAa;;CAGhD,IAAI,aAAmB;AACrB,SAAO,KAAK,qBAAqB,aAAa;;CAGhD,IAAI,iBAAuB;AACzB,SAAO,KAAK,qBAAqB,iBAAiB;;CAGpD,IAAI,YAAqB;EACvB,MAAM,uBAAM,IAAI,MAAM,EAAC,aAAa;AAEpC,UADY,KAAK,UAAU,kBAAkB,QAC/B;;;AAiClB,IAAa,+BAAb,cAEU,gCAIR;CACA,IAAI,qBAA6B;AAC/B,SAAO;;CAGT,MAAM,WAA4B;AAChC,SAAO,oDAAoD,KAAK,WAAW;;;AAU/E,IAAa,iCAAb,cAGU,yBAIR;CACA;CAEA;CAEA,YACE,QACA,QACA,QACA;AACA,QAAM,QAAQ,QAAQ,OAAO;AAC7B,OAAK,SAAS,OAAO;AACrB,OAAK,OAAO,OAAO;;CAGrB,IAAI,qBAA6B;AAC/B,SAAO;;CAGT,cAA+C;AAC7C,SAAO,KAAK;;CAGd,MAAM,WAA4B;AAChC,SAAO,6CAA6C,KAAK,WAAW,SAAS,KAAK;;;AAUtF,IAAsB,4BAAtB,cAEU,gBAAmD;CAC3D;CAEA,YAAoB;CAEpB,YAAY,QAAuD;EACjE,MAAM,SAAuD;GAC3D,qBAAqB;IACnB,YAAY;IACZ,iBAAiB;IAClB;GACD,GAAG;GACJ;AACD,QAAM,OAAO;AACb,OAAK,SAAS;AACd,OAAK,YAAY,QAAQ,aAAa,KAAK,aAAa;;CAG1D,YAAY,MAAsB;AAChC,SAAO,KAAK,MAAM,IAAI,CAAC,KAAK,IAAI;;CAOlC,kBAAkB,QAAwC;AACxD,SAAO,IAAI,iBAAiB,OAAO;;CAGrC,YAAY,QAAiE;AAC3E,SAAO,QAAQ,UAAU,uBAAuB,iBAAiB;;CAGnE,YACE,QACwB;EACxB,MAAM,SAAS,KAAK,YAAY,OAAO;AACvC,MAAI,OACF,QAAO,KAAK,kBAAkB,OAAO;MAGrC,QAAO,KAAK,sBAAsB,OAAO;;CAI7C,MAAM,eAAe,KAA0C;AAE7D,QAAM,IAAI,SAAS,YAAY,WAAW,SAAS,KAAK,UAAU,CAAC;AAGnE,SAAO,KAAK,aAAa,IAAI;;CAG/B,MAAM,KAAK,CAAC,KAAK,OAGqB;EACpC,MAAM,WAAY,MAAM,MAAM,KAAK,CACjC,KACA,KACD,CAAC;EAEF,IAAI,OAAO,SAAS,MAAM,QAAQ,EAAE,OAAO,UAAU;AACrD,SAAO,KAAK,UAAU,gBAAgB,KAAK,OAAO,KAAK,YAAY,EACjE,QAAO,MAAM,KAAK,eAAe,KAAK,IAAI;AAM5C,OAAK,OAAO,KAAK;AACjB,OAAK,WAAW;GACd,GAAG,KAAK;GACR,GAAG;GACJ;AACD,SAAO;;CAGT,mBAAmB,CAAC,MAAM,QAOxB;AACA,SAAO,IAAI,6BACT,KAAK,QACL,KAAK,QACL,KAAK,OACN;;CAGH,iBAAiB,CAAC,MAAM,QAGtB;AACA,SAAO,EAAE;;CAGX,2BACE,KAKA;AAMA,SAAO,IAAI,+BALuD;GAChE,GAAG,KAAK;GACR,QAAQ;GACR,MAAM,KAAK,YAAY,IAAI;GAC5B,EAIS,KAAK,QAAQ,KAAK,OAAO;;CAGrC,uBACE,MACkE;AAClE,QAAM,IAAI,MAAM,2CAA2C;;CAG7D,MAAM,KAAK,KAA6C;EACtD,MAAM,WAAW,MAAM,KAAK,aAAa,IAAI;AAC7C,MAAI,SASF,QAAO,IAAI,UAAU;GACnB,MAAM;GACN,MAP0B;IAC1B,OAAO;IACP,MAJC,UAAU,YAAuB;IAKnC;GAKC;GACD,CAAC;MAEF;;CAIJ,sBACE,KAKA;AAMA,SAAO,IAAI,+BALuD;GAChE,GAAG,KAAK;GACR,QAAQ;GACR,MAAM,KAAK,YAAY,IAAI;GAC5B,EAIS,KAAK,QAAQ,KAAK,OAAO"}

package/dist/experimental/utils/media_core.cjs

@@ -1,7 +1,7 @@
-const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
-const uuid = require_rolldown_runtime.__toESM(require("uuid"));
-const __langchain_core_stores = require_rolldown_runtime.__toESM(require("@langchain/core/stores"));
-const __langchain_core_load_serializable = require_rolldown_runtime.__toESM(require("@langchain/core/load/serializable"));
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+let uuid = require("uuid");
+let _langchain_core_stores = require("@langchain/core/stores");
+let _langchain_core_load_serializable = require("@langchain/core/load/serializable");
 
 //#region src/experimental/utils/media_core.ts
 function bytesToString(dataArray) {
@@ -17,7 +17,7 @@ function bytesToString(dataArray) {
 * Represents a chunk of data that can be identified by the path where the
 * data is (or will be) located, along with optional metadata about the data.
 */
-var MediaBlob = class MediaBlob extends __langchain_core_load_serializable.Serializable {
+var MediaBlob = class MediaBlob extends _langchain_core_load_serializable.Serializable {
 	lc_serializable = true;
 	lc_namespace = [
 		"langchain",
@@ -74,11 +74,9 @@ var MediaBlob = class MediaBlob extends __langchain_core_load_serializable.Seria
 	async encode() {
 		const dataUrl = await this.asDataUrl();
 		const comma = dataUrl.indexOf(",");
-		const encoded = dataUrl.substring(comma + 1);
-		const encoding = dataUrl.indexOf("base64") > -1 ? "base64" : "8bit";
 		return {
-			encoded,
-			encoding
+			encoded: dataUrl.substring(comma + 1),
+			encoding: dataUrl.indexOf("base64") > -1 ? "base64" : "8bit"
 		};
 	}
 	static fromDataUrl(url) {
@@ -87,20 +85,17 @@ var MediaBlob = class MediaBlob extends __langchain_core_load_serializable.Seria
 		const semicolon = url.indexOf(";");
 		const mimeType = url.substring(colon + 1, semicolon);
 		const comma = url.indexOf(",");
-		const base64Data = url.substring(comma + 1);
-		const data = {
-			type: mimeType,
-			value: base64Data
-		};
 		return new MediaBlob({
-			data,
+			data: {
+				type: mimeType,
+				value: url.substring(comma + 1)
+			},
 			path: url
 		});
 	}
 	static async fromBlob(blob, other) {
 		const valueBuffer = await blob.arrayBuffer();
-		const valueArray = new Uint8Array(valueBuffer);
-		const valueStr = bytesToString(valueArray);
+		const valueStr = bytesToString(new Uint8Array(valueBuffer));
 		const value = btoa(valueStr);
 		return new MediaBlob({
 			...other,
@@ -123,7 +118,7 @@ var MediaBlob = class MediaBlob extends __langchain_core_load_serializable.Seria
 * BaseStore may be implemented (or even possible). Those that are not
 * implemented should be documented and throw an Error if called.
 */
-var BlobStore = class extends __langchain_core_stores.BaseStore {
+var BlobStore = class extends _langchain_core_stores.BaseStore {
 	lc_namespace = ["langchain", "google-common"];
 	defaultStoreOptions;
 	defaultFetchOptions;
@@ -169,9 +164,7 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 		return newBlob;
 	}
 	async _validBlobPrefixPath(blob, opts) {
-		const prefix = opts?.pathPrefix ?? "";
-		const suffix = this._blobPathSuffix(blob);
-		const newPath = `${prefix}${suffix}`;
+		const newPath = `${opts?.pathPrefix ?? ""}${this._blobPathSuffix(blob)}`;
 		return this._newBlob(blob, newPath);
 	}
 	_validBlobPrefixUuidFunction(name) {
@@ -182,9 +175,7 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 		}
 	}
 	async _validBlobPrefixUuid(blob, opts) {
-		const prefix = opts?.pathPrefix ?? "";
-		const suffix = this._validBlobPrefixUuidFunction(opts?.actionIfInvalid ?? "prefixUuid4");
-		const newPath = `${prefix}${suffix}`;
+		const newPath = `${opts?.pathPrefix ?? ""}${this._validBlobPrefixUuidFunction(opts?.actionIfInvalid ?? "prefixUuid4")}`;
 		return this._newBlob(blob, newPath);
 	}
 	async _validBlobRemovePath(blob, _opts) {
@@ -206,7 +197,7 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 			case "prefixUuid6":
 			case "prefixUuid7": return this._validBlobPrefixUuid(blob, opts);
 			case "removePath": return this._validBlobRemovePath(blob, opts);
-			default: return void 0;
+			default: return;
 		}
 	}
 	async store(blob, opts = {}) {
@@ -221,7 +212,6 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 			const savedKey = await validBlob.asUri();
 			return await this.fetch(savedKey);
 		}
-		return void 0;
 	}
 	async _missingFetchBlobEmpty(path, _opts) {
 		return new MediaBlob({ path });
@@ -229,7 +219,7 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 	async _missingFetchBlob(path, opts) {
 		switch (opts?.actionIfBlobMissing) {
 			case "emptyBlob": return this._missingFetchBlobEmpty(path, opts);
-			default: return void 0;
+			default: return;
 		}
 	}
 	async fetch(key, opts = {}) {
@@ -238,8 +228,7 @@ var BlobStore = class extends __langchain_core_stores.BaseStore {
 			...opts
 		};
 		const realKey = await this._realKey(key);
-		const ret = await this.mget([realKey]);
-		return ret?.[0] ?? await this._missingFetchBlob(realKey, allOpts);
+		return (await this.mget([realKey]))?.[0] ?? await this._missingFetchBlob(realKey, allOpts);
 	}
 };
 var BackedBlobStore = class extends BlobStore {
@@ -298,17 +287,13 @@ var SimpleWebBlobStore = class extends BlobStore {
 	async _fetch(url) {
 		const ret = new MediaBlob({ path: url });
 		const metadata = {};
-		const fetchOptions = { method: "GET" };
-		const res = await fetch(url, fetchOptions);
+		const res = await fetch(url, { method: "GET" });
 		metadata.status = res.status;
 		const headers = {};
 		for (const [key, value] of res.headers.entries()) headers[key] = value;
 		metadata.headers = headers;
 		metadata.ok = res.ok;
-		if (res.ok) {
-			const resMediaBlob = await MediaBlob.fromBlob(await res.blob());
-			ret.data = resMediaBlob.data;
-		}
+		if (res.ok) ret.data = (await MediaBlob.fromBlob(await res.blob())).data;
 		ret.metadata = metadata;
 		return ret;
 	}
@@ -342,8 +327,7 @@ var DataBlobStore = class extends BlobStore {
 		return MediaBlob.fromDataUrl(url);
 	}
 	async mget(keys) {
-		const blobMap = keys.map(this._fetch);
-		return blobMap;
+		return keys.map(this._fetch);
 	}
 	async mdelete(_keys) {
 		this._notImplementedException();
@@ -394,8 +378,7 @@ var MediaManager = class {
 	}
 	async getMediaBlob(uri) {
 		const aliasBlob = await this.store.fetch(uri);
-		const ret = await this._isInvalid(aliasBlob) ? await this._resolveAndSave(uri) : aliasBlob;
-		return ret;
+		return await this._isInvalid(aliasBlob) ? await this._resolveAndSave(uri) : aliasBlob;
 	}
 };