@azure/ai-projects 1.0.0-beta.4 → 1.0.0-beta.6

package/dist/esm/api/datasets/operations.js

@@ -0,0 +1,286 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+/* eslint-disable tsdoc/syntax */
+import * as fs from "fs";
+import * as nodePath from "path";
+import { _pagedDatasetVersionDeserializer, datasetVersionUnionSerializer, datasetVersionUnionDeserializer, pendingUploadRequestSerializer, pendingUploadResponseDeserializer, _getCredentialsRequestSerializer, assetCredentialResponseDeserializer, } from "../../models/models.js";
+import { expandUrlTemplate } from "../../static-helpers/urlTemplate.js";
+import { buildPagedAsyncIterator, } from "../../static-helpers/pagingHelpers.js";
+import { createRestError, operationOptionsToRequestParameters, } from "@azure-rest/core-client";
+import { ContainerClient } from "@azure/storage-blob";
+export function _getCredentialsSend(context, name, version, body, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions/{version}/credentials{?api-version}", {
+        name: name,
+        version: version,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).post(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { contentType: "application/json", headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers), body: _getCredentialsRequestSerializer(body) }));
+}
+export async function _getCredentialsDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return assetCredentialResponseDeserializer(result.body);
+}
+/** Get the SAS credential to access the storage account associated with a Dataset version. */
+export async function getCredentials(context, name, version, body, options = { requestOptions: {} }) {
+    const result = await _getCredentialsSend(context, name, version, body, options);
+    return _getCredentialsDeserialize(result);
+}
+export function _pendingUploadSend(context, name, version, body, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions/{version}/startPendingUpload{?api-version}", {
+        name: name,
+        version: version,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).post(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { contentType: "application/json", headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers), body: pendingUploadRequestSerializer(body) }));
+}
+export async function _pendingUploadDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return pendingUploadResponseDeserializer(result.body);
+}
+/** Start a new or get an existing pending upload of a dataset for a specific version. */
+export async function pendingUpload(context, name, version, body, options = { requestOptions: {} }) {
+    const result = await _pendingUploadSend(context, name, version, body, options);
+    return _pendingUploadDeserialize(result);
+}
+// Internal helper method to create a new dataset and return a ContainerClient from azure-storage-blob package, to the dataset's blob storage.
+async function createDatasetAndGetItsContainer(context, name, version) {
+    var _a;
+    // Start a pending upload to get the container URL with SAS token
+    const pendingUploadResponse = await pendingUpload(context, name, version, {
+        pendingUploadType: "BlobReference",
+    });
+    const blobReference = pendingUploadResponse.blobReference;
+    // Validate the response
+    if (!blobReference) {
+        throw new Error("Blob reference for consumption is not present");
+    }
+    if (!((_a = blobReference.credential) === null || _a === void 0 ? void 0 : _a.type)) {
+        throw new Error("Credential type is not present");
+    }
+    if (blobReference.credential.type !== "SAS") {
+        throw new Error("Credential type is not SAS");
+    }
+    if (!blobReference.blobUri) {
+        throw new Error("Blob URI is not present or empty");
+    }
+    // Optional debug logging
+    console.debug(`[createDatasetAndGetItsContainer] pendingUploadResponse.pendingUploadId = ${pendingUploadResponse.pendingUploadId}`);
+    console.debug(`[createDatasetAndGetItsContainer] pendingUploadResponse.pendingUploadType = ${pendingUploadResponse.pendingUploadType}`);
+    console.debug(`[createDatasetAndGetItsContainer] blobReference.blobUri = ${blobReference.blobUri}`);
+    console.debug(`[createDatasetAndGetItsContainer] blobReference.storageAccountArmId = ${blobReference.storageAccountArmId}`);
+    console.debug(`[createDatasetAndGetItsContainer] blobReference.credential.sasUri = ${blobReference.credential.sasUri}`);
+    console.debug(`[createDatasetAndGetItsContainer] blobReference.credential.type = ${blobReference.credential.type}`);
+    // Create container client from the blob URI (which includes the SAS token)
+    const containerClient = new ContainerClient(blobReference.blobUri);
+    return {
+        containerClient,
+        version,
+    };
+}
+export async function uploadFile(context, name, version, filePath) {
+    // if file does not exist
+    const fileExists = fs.existsSync(filePath);
+    if (!fileExists) {
+        throw new Error(`File does not exist at path: ${filePath}`);
+    }
+    // Check if the file is a directory
+    const isDirectory = fs.lstatSync(filePath).isDirectory();
+    if (isDirectory) {
+        throw new Error(`The provided file is actually a folder. Use method uploadFolder instead`);
+    }
+    const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(context, name, version);
+    // file name as blob name
+    const blobName = nodePath.basename(filePath);
+    const blockBlobClient = containerClient.getBlockBlobClient(blobName);
+    await blockBlobClient.uploadFile(filePath);
+    const datasetVersion = await createOrUpdate(context, name, outputVersion, {
+        name: name,
+        version: outputVersion,
+        type: "uri_file",
+        dataUri: blockBlobClient.url,
+    });
+    return datasetVersion;
+}
+export async function uploadFolder(context, name, version, folderPath) {
+    // Check if the folder exists
+    const folderExists = fs.existsSync(folderPath);
+    if (!folderExists) {
+        throw new Error(`Folder does not exist at path: ${folderPath}`);
+    }
+    // Check if the folder is a file
+    const isFile = fs.lstatSync(folderPath).isFile();
+    if (isFile) {
+        throw new Error(`The provided path is actually a file. Use method uploadFile instead`);
+    }
+    const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(context, name, version);
+    // Helper function to recursively get all files in a directory
+    async function getAllFiles(dir, fileList = []) {
+        const files = await fs.promises.readdir(dir);
+        for (const file of files) {
+            const filePath = `${dir}/${file}`;
+            const stat = await fs.promises.lstat(filePath);
+            if (stat.isDirectory()) {
+                await getAllFiles(filePath, fileList);
+            }
+            else {
+                fileList.push(filePath);
+            }
+        }
+        return fileList;
+    }
+    // Get all files in the folder
+    const allFiles = await getAllFiles(folderPath);
+    if (allFiles.length === 0) {
+        throw new Error("The provided folder is empty.");
+    }
+    // Upload each file to blob storage while maintaining relative paths
+    for (const filePath of allFiles) {
+        // Create blob name as relative path from the base folder
+        const relativePath = nodePath.relative(folderPath, filePath).split(nodePath.sep).join("/");
+        console.debug(`[uploadFolderAndCreate] Start uploading file '${filePath}' as blob '${relativePath}'`);
+        // Get a block blob client for the relative path
+        const blobClient = containerClient.getBlockBlobClient(relativePath);
+        // Upload the file using a readable stream for better performance
+        const fileStream = fs.createReadStream(filePath);
+        await blobClient.uploadStream(fileStream);
+        console.debug(`[uploadFolderAndCreate] Done uploading file '${filePath}' as blob '${relativePath}'`);
+    }
+    // Create dataset version that references this folder
+    const datasetVersion = await createOrUpdate(context, name, outputVersion, {
+        name: name,
+        version: outputVersion,
+        type: "uri_folder",
+        dataUri: containerClient.url,
+    });
+    return datasetVersion;
+}
+export function _createOrUpdateSend(context, name, version, body, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions/{version}{?api-version}", {
+        name: name,
+        version: version,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).patch(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { contentType: "application/json", headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers), body: datasetVersionUnionSerializer(body) }));
+}
+export async function _createOrUpdateDeserialize(result) {
+    const expectedStatuses = ["201", "200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return datasetVersionUnionDeserializer(result.body);
+}
+/** Create a new or update an existing DatasetVersion with the given version id */
+export async function createOrUpdate(context, name, version, body, options = { requestOptions: {} }) {
+    const result = await _createOrUpdateSend(context, name, version, body, options);
+    return _createOrUpdateDeserialize(result);
+}
+export function _$deleteSend(context, name, version, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions/{version}{?api-version}", {
+        name: name,
+        version: version,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).delete(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _$deleteDeserialize(result) {
+    const expectedStatuses = ["204"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return;
+}
+/** Delete the specific version of the DatasetVersion */
+/**
+ *  @fixme delete is a reserved word that cannot be used as an operation name.
+ *         Please add @clientName("clientName") or @clientName("<JS-Specific-Name>", "javascript")
+ *         to the operation to override the generated name.
+ */
+export async function $delete(context, name, version, options = { requestOptions: {} }) {
+    const result = await _$deleteSend(context, name, version, options);
+    return _$deleteDeserialize(result);
+}
+export function _getSend(context, name, version, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions/{version}{?api-version}", {
+        name: name,
+        version: version,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).get(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _getDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return datasetVersionUnionDeserializer(result.body);
+}
+/** Get the specific version of the DatasetVersion */
+export async function get(context, name, version, options = { requestOptions: {} }) {
+    const result = await _getSend(context, name, version, options);
+    return _getDeserialize(result);
+}
+export function _listSend(context, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets{?api-version,continuationToken}", {
+        "api-version": context.apiVersion,
+        continuationToken: options === null || options === void 0 ? void 0 : options.continuationToken,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).get(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _listDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return _pagedDatasetVersionDeserializer(result.body);
+}
+/** List the latest version of each DatasetVersion */
+export function list(context, options = { requestOptions: {} }) {
+    return buildPagedAsyncIterator(context, () => _listSend(context, options), _listDeserialize, ["200"], { itemName: "value", nextLinkName: "nextLink" });
+}
+export function _listVersionsSend(context, name, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/datasets/{name}/versions{?api-version,continuationToken}", {
+        name: name,
+        "api-version": context.apiVersion,
+        continuationToken: options === null || options === void 0 ? void 0 : options.continuationToken,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).get(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign({ accept: "application/json" }, (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _listVersionsDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return _pagedDatasetVersionDeserializer(result.body);
+}
+/** List all versions of the given DatasetVersion */
+export function listVersions(context, name, options = { requestOptions: {} }) {
+    return buildPagedAsyncIterator(context, () => _listVersionsSend(context, name, options), _listVersionsDeserialize, ["200"], { itemName: "value", nextLinkName: "nextLink" });
+}
+//# sourceMappingURL=operations.js.map

package/dist/esm/api/datasets/operations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"operations.js","sourceRoot":"","sources":["../../../../src/api/datasets/operations.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;AAEjC,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,QAAQ,MAAM,MAAM,CAAC;AAEjC,OAAO,EAEL,gCAAgC,EAChC,6BAA6B,EAC7B,+BAA+B,EAG/B,8BAA8B,EAE9B,iCAAiC,EACjC,gCAAgC,EAEhC,mCAAmC,GACpC,MAAM,wBAAwB,CAAC;AAUhC,OAAO,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AACxE,OAAO,EAEL,uBAAuB,GACxB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAGL,eAAe,EACf,mCAAmC,GACpC,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,qBAAqB,CAAC;AAEtD,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAAyB,EACzB,UAAgD,EAAE,cAAc,EAAE,EAAE,EAAE;;IAEtE,MAAM,IAAI,GAAG,iBAAiB,CAC5B,+DAA+D,EAC/D;QACE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,iCACzB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,WAAW,EAAE,kBAAkB,EAC/B,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,GAEpC,IAAI,EAAE,gCAAgC,CAAC,IAAI,CAAC,IAC5C,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAA6B;IAE7B,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,mCAAmC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC1D,CAAC;AAED,8FAA8F;AAC9F,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAAyB,EACzB,UAAgD,EAAE,cAAc,EAAE,EAAE,EAAE;IAEtE,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAChF,OAAO,0BAA0B,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,kBAAkB,CAChC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAA0B,EAC1B,UAA+C,EAAE,cAAc,EAAE,EAAE,EAAE;;IAErE,MAAM,IAAI,GAAG,iBAAiB,CAC5B,sEAAsE,EACtE;QACE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,iCACzB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,WAAW,EAAE,kBAAkB,EAC/B,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,GAEpC,IAAI,EAAE,8BAA8B,CAAC,IAAI,CAAC,IAC1C,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,MAA6B;IAE7B,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,iCAAiC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACxD,CAAC;AAED,yFAAyF;AACzF,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAA0B,EAC1B,UAA+C,EAAE,cAAc,EAAE,EAAE,EAAE;IAErE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC/E,OAAO,yBAAyB,CAAC,MAAM,CAAC,CAAC;AAC3C,CAAC;AAED,8IAA8I;AAC9I,KAAK,UAAU,+BAA+B,CAC5C,OAAe,EACf,IAAY,EACZ,OAAe;;IAEf,iEAAiE;IACjE,MAAM,qBAAqB,GAAG,MAAM,aAAa,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE;QACxE,iBAAiB,EAAE,eAAe;KACX,CAAC,CAAC;IAE3B,MAAM,aAAa,GAAG,qBAAqB,CAAC,aAAa,CAAC;IAC1D,wBAAwB;IACxB,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,CAAA,MAAA,aAAa,CAAC,UAAU,0CAAE,IAAI,CAAA,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;IACpD,CAAC;IAED,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QAC5C,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACtD,CAAC;IAED,yBAAyB;IACzB,OAAO,CAAC,KAAK,CACX,6EAA6E,qBAAqB,CAAC,eAAe,EAAE,CACrH,CAAC;IACF,OAAO,CAAC,KAAK,CACX,+EAA+E,qBAAqB,CAAC,iBAAiB,EAAE,CACzH,CAAC;IACF,OAAO,CAAC,KAAK,CACX,6DAA6D,aAAa,CAAC,OAAO,EAAE,CACrF,CAAC;IACF,OAAO,CAAC,KAAK,CACX,yEAAyE,aAAa,CAAC,mBAAmB,EAAE,CAC7G,CAAC;IACF,OAAO,CAAC,KAAK,CACX,uEAAuE,aAAa,CAAC,UAAU,CAAC,MAAM,EAAE,CACzG,CAAC;IACF,OAAO,CAAC,KAAK,CACX,qEAAqE,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CACrG,CAAC;IAEF,2EAA2E;IAC3E,MAAM,eAAe,GAAG,IAAI,eAAe,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IAEnE,OAAO;QACL,eAAe;QACf,OAAO;KACR,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,IAAY,EACZ,OAAe,EACf,QAAgB;IAEhB,yBAAyB;IAEzB,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAC3C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,EAAE,CAAC,CAAC;IAC9D,CAAC;IACD,mCAAmC;IACnC,MAAM,WAAW,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACzD,IAAI,WAAW,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,yEAAyE,CAAC,CAAC;IAC7F,CAAC;IAED,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,+BAA+B,CACvF,OAAO,EACP,IAAI,EACJ,OAAO,CACR,CAAC;IACF,yBAAyB;IACzB,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IAC7C,MAAM,eAAe,GAAG,eAAe,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACrE,MAAM,eAAe,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAE3C,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE;QACxE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,aAAa;QACtB,IAAI,EAAE,UAAU;QAChB,OAAO,EAAE,eAAe,CAAC,GAAG;KAC7B,CAAC,CAAC;IACH,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,UAAkB;IAElB,6BAA6B;IAC7B,MAAM,YAAY,GAAG,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAC/C,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,kCAAkC,UAAU,EAAE,CAAC,CAAC;IAClE,CAAC;IACD,gCAAgC;IAChC,MAAM,MAAM,GAAG,EAAE,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,MAAM,EAAE,CAAC;IACjD,IAAI,MAAM,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,qEAAqE,CAAC,CAAC;IACzF,CAAC;IAED,MAAM,EAAE,eAAe,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,+BAA+B,CACvF,OAAO,EACP,IAAI,EACJ,OAAO,CACR,CAAC;IAEF,8DAA8D;IAC9D,KAAK,UAAU,WAAW,CAAC,GAAW,EAAE,WAAqB,EAAE;QAC7D,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QAE7C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;YAE/C,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBACvB,MAAM,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACxC,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,8BAA8B;IAC9B,MAAM,QAAQ,GAAG,MAAM,WAAW,CAAC,UAAU,CAAC,CAAC;IAC/C,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IAED,oEAAoE;IACpE,KAAK,MAAM,QAAQ,IAAI,QAAQ,EAAE,CAAC;QAChC,yDAAyD;QACzD,MAAM,YAAY,GAAG,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3F,OAAO,CAAC,KAAK,CACX,iDAAiD,QAAQ,cAAc,YAAY,GAAG,CACvF,CAAC;QAEF,gDAAgD;QAChD,MAAM,UAAU,GAAG,eAAe,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC;QAEpE,iEAAiE;QACjE,MAAM,UAAU,GAAG,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QACjD,MAAM,UAAU,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;QAC1C,OAAO,CAAC,KAAK,CACX,gDAAgD,QAAQ,cAAc,YAAY,GAAG,CACtF,CAAC;IACJ,CAAC;IAED,qDAAqD;IACrD,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE;QACxE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,aAAa;QACtB,IAAI,EAAE,YAAY;QAClB,OAAO,EAAE,eAAe,CAAC,GAAG;KAC7B,CAAC,CAAC;IAEH,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAAyB,EACzB,UAAgD,EAAE,cAAc,EAAE,EAAE,EAAE;;IAEtE,MAAM,IAAI,GAAG,iBAAiB,CAC5B,mDAAmD,EACnD;QACE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,iCAC1B,mCAAmC,CAAC,OAAO,CAAC,KAC/C,WAAW,EAAE,kBAAkB,EAC/B,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,GAEpC,IAAI,EAAE,6BAA6B,CAAC,IAAI,CAAC,IACzC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,0BAA0B,CAC9C,MAA6B;IAE7B,MAAM,gBAAgB,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACxC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,+BAA+B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACtD,CAAC;AAED,kFAAkF;AAClF,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,OAAe,EACf,IAAY,EACZ,OAAe,EACf,IAAyB,EACzB,UAAgD,EAAE,cAAc,EAAE,EAAE,EAAE;IAEtE,MAAM,MAAM,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAChF,OAAO,0BAA0B,CAAC,MAAM,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,YAAY,CAC1B,OAAe,EACf,IAAY,EACZ,OAAe,EACf,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE9D,MAAM,IAAI,GAAG,iBAAiB,CAC5B,mDAAmD,EACnD;QACE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,iCAC3B,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,mBAAmB,CAAC,MAA6B;IACrE,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO;AACT,CAAC;AAED,wDAAwD;AACxD;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,OAAO,CAC3B,OAAe,EACf,IAAY,EACZ,OAAe,EACf,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE9D,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IACnE,OAAO,mBAAmB,CAAC,MAAM,CAAC,CAAC;AACrC,CAAC;AAED,MAAM,UAAU,QAAQ,CACtB,OAAe,EACf,IAAY,EACZ,OAAe,EACf,UAAqC,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE3D,MAAM,IAAI,GAAG,iBAAiB,CAC5B,mDAAmD,EACnD;QACE,IAAI,EAAE,IAAI;QACV,OAAO,EAAE,OAAO;QAChB,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,iCACxB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,MAA6B;IACjE,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,+BAA+B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACtD,CAAC;AAED,qDAAqD;AACrD,MAAM,CAAC,KAAK,UAAU,GAAG,CACvB,OAAe,EACf,IAAY,EACZ,OAAe,EACf,UAAqC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE3D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;IAC/D,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,SAAS,CACvB,OAAe,EACf,UAAsC,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE5D,MAAM,IAAI,GAAG,iBAAiB,CAC5B,2CAA2C,EAC3C;QACE,aAAa,EAAE,OAAO,CAAC,UAAU;QACjC,iBAAiB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,iBAAiB;KAC9C,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,iCACxB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,MAA6B;IAE7B,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,gCAAgC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACvD,CAAC;AAED,qDAAqD;AACrD,MAAM,UAAU,IAAI,CAClB,OAAe,EACf,UAAsC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE5D,OAAO,uBAAuB,CAC5B,OAAO,EACP,GAAG,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,EACjC,gBAAgB,EAChB,CAAC,KAAK,CAAC,EACP,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,CAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,OAAe,EACf,IAAY,EACZ,UAA8C,EAAE,cAAc,EAAE,EAAE,EAAE;;IAEpE,MAAM,IAAI,GAAG,iBAAiB,CAC5B,2DAA2D,EAC3D;QACE,IAAI,EAAE,IAAI;QACV,aAAa,EAAE,OAAO,CAAC,UAAU;QACjC,iBAAiB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,iBAAiB;KAC9C,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,iCACxB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,kBACL,MAAM,EAAE,kBAAkB,IACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,MAA6B;IAE7B,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,gCAAgC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACvD,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,YAAY,CAC1B,OAAe,EACf,IAAY,EACZ,UAA8C,EAAE,cAAc,EAAE,EAAE,EAAE;IAEpE,OAAO,uBAAuB,CAC5B,OAAO,EACP,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,EAC/C,wBAAwB,EACxB,CAAC,KAAK,CAAC,EACP,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,CAChD,CAAC;AACJ,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport * as fs from \"fs\";\nimport * as nodePath from \"path\";\nimport { AIProjectContext as Client } from \"../index.js\";\nimport {\n  _PagedDatasetVersion,\n  _pagedDatasetVersionDeserializer,\n  datasetVersionUnionSerializer,\n  datasetVersionUnionDeserializer,\n  DatasetVersionUnion,\n  PendingUploadRequest,\n  pendingUploadRequestSerializer,\n  PendingUploadResponse,\n  pendingUploadResponseDeserializer,\n  _getCredentialsRequestSerializer,\n  AssetCredentialResponse,\n  assetCredentialResponseDeserializer,\n} from \"../../models/models.js\";\nimport {\n  DatasetsGetCredentialsOptionalParams,\n  DatasetsPendingUploadOptionalParams,\n  DatasetsCreateOrUpdateOptionalParams,\n  DatasetsDeleteOptionalParams,\n  DatasetsGetOptionalParams,\n  DatasetsListOptionalParams,\n  DatasetsListVersionsOptionalParams,\n} from \"./options.js\";\nimport { expandUrlTemplate } from \"../../static-helpers/urlTemplate.js\";\nimport {\n  PagedAsyncIterableIterator,\n  buildPagedAsyncIterator,\n} from \"../../static-helpers/pagingHelpers.js\";\nimport {\n  StreamableMethod,\n  PathUncheckedResponse,\n  createRestError,\n  operationOptionsToRequestParameters,\n} from \"@azure-rest/core-client\";\nimport { ContainerClient } from \"@azure/storage-blob\";\n\nexport function _getCredentialsSend(\n  context: Client,\n  name: string,\n  version: string,\n  body: Record<string, any>,\n  options: DatasetsGetCredentialsOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions/{version}/credentials{?api-version}\",\n    {\n      name: name,\n      version: version,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).post({\n    ...operationOptionsToRequestParameters(options),\n    contentType: \"application/json\",\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n    body: _getCredentialsRequestSerializer(body),\n  });\n}\n\nexport async function _getCredentialsDeserialize(\n  result: PathUncheckedResponse,\n): Promise<AssetCredentialResponse> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return assetCredentialResponseDeserializer(result.body);\n}\n\n/** Get the SAS credential to access the storage account associated with a Dataset version. */\nexport async function getCredentials(\n  context: Client,\n  name: string,\n  version: string,\n  body: Record<string, any>,\n  options: DatasetsGetCredentialsOptionalParams = { requestOptions: {} },\n): Promise<AssetCredentialResponse> {\n  const result = await _getCredentialsSend(context, name, version, body, options);\n  return _getCredentialsDeserialize(result);\n}\n\nexport function _pendingUploadSend(\n  context: Client,\n  name: string,\n  version: string,\n  body: PendingUploadRequest,\n  options: DatasetsPendingUploadOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions/{version}/startPendingUpload{?api-version}\",\n    {\n      name: name,\n      version: version,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).post({\n    ...operationOptionsToRequestParameters(options),\n    contentType: \"application/json\",\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n    body: pendingUploadRequestSerializer(body),\n  });\n}\n\nexport async function _pendingUploadDeserialize(\n  result: PathUncheckedResponse,\n): Promise<PendingUploadResponse> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return pendingUploadResponseDeserializer(result.body);\n}\n\n/** Start a new or get an existing pending upload of a dataset for a specific version. */\nexport async function pendingUpload(\n  context: Client,\n  name: string,\n  version: string,\n  body: PendingUploadRequest,\n  options: DatasetsPendingUploadOptionalParams = { requestOptions: {} },\n): Promise<PendingUploadResponse> {\n  const result = await _pendingUploadSend(context, name, version, body, options);\n  return _pendingUploadDeserialize(result);\n}\n\n// Internal helper method to create a new dataset and return a ContainerClient from azure-storage-blob package, to the dataset's blob storage.\nasync function createDatasetAndGetItsContainer(\n  context: Client,\n  name: string,\n  version: string,\n): Promise<{ containerClient: ContainerClient; version: string }> {\n  // Start a pending upload to get the container URL with SAS token\n  const pendingUploadResponse = await pendingUpload(context, name, version, {\n    pendingUploadType: \"BlobReference\",\n  } as PendingUploadRequest);\n\n  const blobReference = pendingUploadResponse.blobReference;\n  // Validate the response\n  if (!blobReference) {\n    throw new Error(\"Blob reference for consumption is not present\");\n  }\n\n  if (!blobReference.credential?.type) {\n    throw new Error(\"Credential type is not present\");\n  }\n\n  if (blobReference.credential.type !== \"SAS\") {\n    throw new Error(\"Credential type is not SAS\");\n  }\n\n  if (!blobReference.blobUri) {\n    throw new Error(\"Blob URI is not present or empty\");\n  }\n\n  // Optional debug logging\n  console.debug(\n    `[createDatasetAndGetItsContainer] pendingUploadResponse.pendingUploadId = ${pendingUploadResponse.pendingUploadId}`,\n  );\n  console.debug(\n    `[createDatasetAndGetItsContainer] pendingUploadResponse.pendingUploadType = ${pendingUploadResponse.pendingUploadType}`,\n  );\n  console.debug(\n    `[createDatasetAndGetItsContainer] blobReference.blobUri = ${blobReference.blobUri}`,\n  );\n  console.debug(\n    `[createDatasetAndGetItsContainer] blobReference.storageAccountArmId = ${blobReference.storageAccountArmId}`,\n  );\n  console.debug(\n    `[createDatasetAndGetItsContainer] blobReference.credential.sasUri = ${blobReference.credential.sasUri}`,\n  );\n  console.debug(\n    `[createDatasetAndGetItsContainer] blobReference.credential.type = ${blobReference.credential.type}`,\n  );\n\n  // Create container client from the blob URI (which includes the SAS token)\n  const containerClient = new ContainerClient(blobReference.blobUri);\n\n  return {\n    containerClient,\n    version,\n  };\n}\n\nexport async function uploadFile(\n  context: Client,\n  name: string,\n  version: string,\n  filePath: string,\n): Promise<DatasetVersionUnion> {\n  // if file does not exist\n\n  const fileExists = fs.existsSync(filePath);\n  if (!fileExists) {\n    throw new Error(`File does not exist at path: ${filePath}`);\n  }\n  // Check if the file is a directory\n  const isDirectory = fs.lstatSync(filePath).isDirectory();\n  if (isDirectory) {\n    throw new Error(`The provided file is actually a folder. Use method uploadFolder instead`);\n  }\n\n  const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(\n    context,\n    name,\n    version,\n  );\n  // file name as blob name\n  const blobName = nodePath.basename(filePath);\n  const blockBlobClient = containerClient.getBlockBlobClient(blobName);\n  await blockBlobClient.uploadFile(filePath);\n\n  const datasetVersion = await createOrUpdate(context, name, outputVersion, {\n    name: name,\n    version: outputVersion,\n    type: \"uri_file\",\n    dataUri: blockBlobClient.url,\n  });\n  return datasetVersion;\n}\n\nexport async function uploadFolder(\n  context: Client,\n  name: string,\n  version: string,\n  folderPath: string,\n): Promise<DatasetVersionUnion> {\n  // Check if the folder exists\n  const folderExists = fs.existsSync(folderPath);\n  if (!folderExists) {\n    throw new Error(`Folder does not exist at path: ${folderPath}`);\n  }\n  // Check if the folder is a file\n  const isFile = fs.lstatSync(folderPath).isFile();\n  if (isFile) {\n    throw new Error(`The provided path is actually a file. Use method uploadFile instead`);\n  }\n\n  const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(\n    context,\n    name,\n    version,\n  );\n\n  // Helper function to recursively get all files in a directory\n  async function getAllFiles(dir: string, fileList: string[] = []): Promise<string[]> {\n    const files = await fs.promises.readdir(dir);\n\n    for (const file of files) {\n      const filePath = `${dir}/${file}`;\n      const stat = await fs.promises.lstat(filePath);\n\n      if (stat.isDirectory()) {\n        await getAllFiles(filePath, fileList);\n      } else {\n        fileList.push(filePath);\n      }\n    }\n\n    return fileList;\n  }\n\n  // Get all files in the folder\n  const allFiles = await getAllFiles(folderPath);\n  if (allFiles.length === 0) {\n    throw new Error(\"The provided folder is empty.\");\n  }\n\n  // Upload each file to blob storage while maintaining relative paths\n  for (const filePath of allFiles) {\n    // Create blob name as relative path from the base folder\n    const relativePath = nodePath.relative(folderPath, filePath).split(nodePath.sep).join(\"/\");\n\n    console.debug(\n      `[uploadFolderAndCreate] Start uploading file '${filePath}' as blob '${relativePath}'`,\n    );\n\n    // Get a block blob client for the relative path\n    const blobClient = containerClient.getBlockBlobClient(relativePath);\n\n    // Upload the file using a readable stream for better performance\n    const fileStream = fs.createReadStream(filePath);\n    await blobClient.uploadStream(fileStream);\n    console.debug(\n      `[uploadFolderAndCreate] Done uploading file '${filePath}' as blob '${relativePath}'`,\n    );\n  }\n\n  // Create dataset version that references this folder\n  const datasetVersion = await createOrUpdate(context, name, outputVersion, {\n    name: name,\n    version: outputVersion,\n    type: \"uri_folder\",\n    dataUri: containerClient.url,\n  });\n\n  return datasetVersion;\n}\n\nexport function _createOrUpdateSend(\n  context: Client,\n  name: string,\n  version: string,\n  body: DatasetVersionUnion,\n  options: DatasetsCreateOrUpdateOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions/{version}{?api-version}\",\n    {\n      name: name,\n      version: version,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).patch({\n    ...operationOptionsToRequestParameters(options),\n    contentType: \"application/json\",\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n    body: datasetVersionUnionSerializer(body),\n  });\n}\n\nexport async function _createOrUpdateDeserialize(\n  result: PathUncheckedResponse,\n): Promise<DatasetVersionUnion> {\n  const expectedStatuses = [\"201\", \"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return datasetVersionUnionDeserializer(result.body);\n}\n\n/** Create a new or update an existing DatasetVersion with the given version id */\nexport async function createOrUpdate(\n  context: Client,\n  name: string,\n  version: string,\n  body: DatasetVersionUnion,\n  options: DatasetsCreateOrUpdateOptionalParams = { requestOptions: {} },\n): Promise<DatasetVersionUnion> {\n  const result = await _createOrUpdateSend(context, name, version, body, options);\n  return _createOrUpdateDeserialize(result);\n}\n\nexport function _$deleteSend(\n  context: Client,\n  name: string,\n  version: string,\n  options: DatasetsDeleteOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions/{version}{?api-version}\",\n    {\n      name: name,\n      version: version,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).delete({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _$deleteDeserialize(result: PathUncheckedResponse): Promise<void> {\n  const expectedStatuses = [\"204\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return;\n}\n\n/** Delete the specific version of the DatasetVersion */\n/**\n *  @fixme delete is a reserved word that cannot be used as an operation name.\n *         Please add @clientName(\"clientName\") or @clientName(\"<JS-Specific-Name>\", \"javascript\")\n *         to the operation to override the generated name.\n */\nexport async function $delete(\n  context: Client,\n  name: string,\n  version: string,\n  options: DatasetsDeleteOptionalParams = { requestOptions: {} },\n): Promise<void> {\n  const result = await _$deleteSend(context, name, version, options);\n  return _$deleteDeserialize(result);\n}\n\nexport function _getSend(\n  context: Client,\n  name: string,\n  version: string,\n  options: DatasetsGetOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions/{version}{?api-version}\",\n    {\n      name: name,\n      version: version,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).get({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _getDeserialize(result: PathUncheckedResponse): Promise<DatasetVersionUnion> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return datasetVersionUnionDeserializer(result.body);\n}\n\n/** Get the specific version of the DatasetVersion */\nexport async function get(\n  context: Client,\n  name: string,\n  version: string,\n  options: DatasetsGetOptionalParams = { requestOptions: {} },\n): Promise<DatasetVersionUnion> {\n  const result = await _getSend(context, name, version, options);\n  return _getDeserialize(result);\n}\n\nexport function _listSend(\n  context: Client,\n  options: DatasetsListOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets{?api-version,continuationToken}\",\n    {\n      \"api-version\": context.apiVersion,\n      continuationToken: options?.continuationToken,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).get({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _listDeserialize(\n  result: PathUncheckedResponse,\n): Promise<_PagedDatasetVersion> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return _pagedDatasetVersionDeserializer(result.body);\n}\n\n/** List the latest version of each DatasetVersion */\nexport function list(\n  context: Client,\n  options: DatasetsListOptionalParams = { requestOptions: {} },\n): PagedAsyncIterableIterator<DatasetVersionUnion> {\n  return buildPagedAsyncIterator(\n    context,\n    () => _listSend(context, options),\n    _listDeserialize,\n    [\"200\"],\n    { itemName: \"value\", nextLinkName: \"nextLink\" },\n  );\n}\n\nexport function _listVersionsSend(\n  context: Client,\n  name: string,\n  options: DatasetsListVersionsOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/datasets/{name}/versions{?api-version,continuationToken}\",\n    {\n      name: name,\n      \"api-version\": context.apiVersion,\n      continuationToken: options?.continuationToken,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).get({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _listVersionsDeserialize(\n  result: PathUncheckedResponse,\n): Promise<_PagedDatasetVersion> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return _pagedDatasetVersionDeserializer(result.body);\n}\n\n/** List all versions of the given DatasetVersion */\nexport function listVersions(\n  context: Client,\n  name: string,\n  options: DatasetsListVersionsOptionalParams = { requestOptions: {} },\n): PagedAsyncIterableIterator<DatasetVersionUnion> {\n  return buildPagedAsyncIterator(\n    context,\n    () => _listVersionsSend(context, name, options),\n    _listVersionsDeserialize,\n    [\"200\"],\n    { itemName: \"value\", nextLinkName: \"nextLink\" },\n  );\n}\n"]}

package/dist/esm/api/datasets/options.d.ts

@@ -0,0 +1,27 @@
+import { OperationOptions } from "@azure-rest/core-client";
+/** Optional parameters. */
+export interface DatasetsGetCredentialsOptionalParams extends OperationOptions {
+}
+/** Optional parameters. */
+export interface DatasetsPendingUploadOptionalParams extends OperationOptions {
+}
+/** Optional parameters. */
+export interface DatasetsCreateOrUpdateOptionalParams extends OperationOptions {
+}
+/** Optional parameters. */
+export interface DatasetsDeleteOptionalParams extends OperationOptions {
+}
+/** Optional parameters. */
+export interface DatasetsGetOptionalParams extends OperationOptions {
+}
+/** Optional parameters. */
+export interface DatasetsListOptionalParams extends OperationOptions {
+    /** Continuation token for pagination. */
+    continuationToken?: string;
+}
+/** Optional parameters. */
+export interface DatasetsListVersionsOptionalParams extends OperationOptions {
+    /** Continuation token for pagination. */
+    continuationToken?: string;
+}
+//# sourceMappingURL=options.d.ts.map

package/dist/esm/api/datasets/options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.d.ts","sourceRoot":"","sources":["../../../../src/api/datasets/options.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAE3D,2BAA2B;AAC3B,MAAM,WAAW,oCAAqC,SAAQ,gBAAgB;CAAG;AAEjF,2BAA2B;AAC3B,MAAM,WAAW,mCAAoC,SAAQ,gBAAgB;CAAG;AAEhF,2BAA2B;AAC3B,MAAM,WAAW,oCAAqC,SAAQ,gBAAgB;CAAG;AAEjF,2BAA2B;AAC3B,MAAM,WAAW,4BAA6B,SAAQ,gBAAgB;CAAG;AAEzE,2BAA2B;AAC3B,MAAM,WAAW,yBAA0B,SAAQ,gBAAgB;CAAG;AAEtE,2BAA2B;AAC3B,MAAM,WAAW,0BAA2B,SAAQ,gBAAgB;IAClE,yCAAyC;IACzC,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,2BAA2B;AAC3B,MAAM,WAAW,kCAAmC,SAAQ,gBAAgB;IAC1E,yCAAyC;IACzC,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B"}

package/dist/esm/api/datasets/options.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=options.js.map

package/dist/esm/api/datasets/options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.js","sourceRoot":"","sources":["../../../../src/api/datasets/options.ts"],"names":[],"mappings":"","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\nimport { OperationOptions } from \"@azure-rest/core-client\";\n\n/** Optional parameters. */\nexport interface DatasetsGetCredentialsOptionalParams extends OperationOptions {}\n\n/** Optional parameters. */\nexport interface DatasetsPendingUploadOptionalParams extends OperationOptions {}\n\n/** Optional parameters. */\nexport interface DatasetsCreateOrUpdateOptionalParams extends OperationOptions {}\n\n/** Optional parameters. */\nexport interface DatasetsDeleteOptionalParams extends OperationOptions {}\n\n/** Optional parameters. */\nexport interface DatasetsGetOptionalParams extends OperationOptions {}\n\n/** Optional parameters. */\nexport interface DatasetsListOptionalParams extends OperationOptions {\n  /** Continuation token for pagination. */\n  continuationToken?: string;\n}\n\n/** Optional parameters. */\nexport interface DatasetsListVersionsOptionalParams extends OperationOptions {\n  /** Continuation token for pagination. */\n  continuationToken?: string;\n}\n"]}

package/dist/esm/api/deployments/index.d.ts

@@ -0,0 +1,3 @@
+export { list, get } from "./operations.js";
+export { DeploymentsListOptionalParams, DeploymentsGetOptionalParams } from "./options.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/esm/api/deployments/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/deployments/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC"}

package/dist/esm/api/deployments/index.js

@@ -0,0 +1,4 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+export { list, get } from "./operations.js";
+//# sourceMappingURL=index.js.map

package/dist/esm/api/deployments/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/deployments/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport { list, get } from \"./operations.js\";\nexport { DeploymentsListOptionalParams, DeploymentsGetOptionalParams } from \"./options.js\";\n"]}

package/dist/esm/api/deployments/operations.d.ts

@@ -0,0 +1,14 @@
+import { AIProjectContext as Client } from "../index.js";
+import { DeploymentUnion, _PagedDeployment } from "../../models/models.js";
+import { DeploymentsListOptionalParams, DeploymentsGetOptionalParams } from "./options.js";
+import { PagedAsyncIterableIterator } from "../../static-helpers/pagingHelpers.js";
+import { StreamableMethod, PathUncheckedResponse } from "@azure-rest/core-client";
+export declare function _listSend(context: Client, options?: DeploymentsListOptionalParams): StreamableMethod;
+export declare function _listDeserialize(result: PathUncheckedResponse): Promise<_PagedDeployment>;
+/** List all deployed models in the project */
+export declare function list(context: Client, options?: DeploymentsListOptionalParams): PagedAsyncIterableIterator<DeploymentUnion>;
+export declare function _getSend(context: Client, name: string, options?: DeploymentsGetOptionalParams): StreamableMethod;
+export declare function _getDeserialize(result: PathUncheckedResponse): Promise<DeploymentUnion>;
+/** Get a deployed model. */
+export declare function get(context: Client, name: string, options?: DeploymentsGetOptionalParams): Promise<DeploymentUnion>;
+//# sourceMappingURL=operations.d.ts.map

package/dist/esm/api/deployments/operations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/deployments/operations.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAEL,eAAe,EACf,gBAAgB,EAEjB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,6BAA6B,EAAE,4BAA4B,EAAE,MAAM,cAAc,CAAC;AAE3F,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AAEjC,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,6BAAsD,GAC9D,gBAAgB,CA0BlB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAO/F;AAED,8CAA8C;AAC9C,wBAAgB,IAAI,CAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,6BAAsD,GAC9D,0BAA0B,CAAC,eAAe,CAAC,CAQ7C;AAED,wBAAgB,QAAQ,CACtB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,4BAAqD,GAC7D,gBAAgB,CAqBlB;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,eAAe,CAAC,CAO7F;AAED,4BAA4B;AAC5B,wBAAsB,GAAG,CACvB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,4BAAqD,GAC7D,OAAO,CAAC,eAAe,CAAC,CAG1B"}

package/dist/esm/api/deployments/operations.js

@@ -0,0 +1,59 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+import { deploymentUnionDeserializer, _pagedDeploymentDeserializer, } from "../../models/models.js";
+import { expandUrlTemplate } from "../../static-helpers/urlTemplate.js";
+import { buildPagedAsyncIterator, } from "../../static-helpers/pagingHelpers.js";
+import { createRestError, operationOptionsToRequestParameters, } from "@azure-rest/core-client";
+export function _listSend(context, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/deployments{?api-version,modelPublisher,modelName,deploymentType,top,skip,maxpagesize}", {
+        "api-version": context.apiVersion,
+        modelPublisher: options === null || options === void 0 ? void 0 : options.modelPublisher,
+        modelName: options === null || options === void 0 ? void 0 : options.modelName,
+        deploymentType: options === null || options === void 0 ? void 0 : options.deploymentType,
+        top: options === null || options === void 0 ? void 0 : options.top,
+        skip: options === null || options === void 0 ? void 0 : options.skip,
+        maxpagesize: options === null || options === void 0 ? void 0 : options.maxpagesize,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).get(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign(Object.assign(Object.assign({}, ((options === null || options === void 0 ? void 0 : options.clientRequestId) !== undefined
+            ? { "x-ms-client-request-id": options === null || options === void 0 ? void 0 : options.clientRequestId }
+            : {})), { accept: "application/json" }), (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _listDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return _pagedDeploymentDeserializer(result.body);
+}
+/** List all deployed models in the project */
+export function list(context, options = { requestOptions: {} }) {
+    return buildPagedAsyncIterator(context, () => _listSend(context, options), _listDeserialize, ["200"], { itemName: "value", nextLinkName: "nextLink" });
+}
+export function _getSend(context, name, options = { requestOptions: {} }) {
+    var _a, _b;
+    const path = expandUrlTemplate("/deployments/{name}{?api-version}", {
+        name: name,
+        "api-version": context.apiVersion,
+    }, {
+        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+    });
+    return context.path(path).get(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { headers: Object.assign(Object.assign(Object.assign({}, ((options === null || options === void 0 ? void 0 : options.clientRequestId) !== undefined
+            ? { "x-ms-client-request-id": options === null || options === void 0 ? void 0 : options.clientRequestId }
+            : {})), { accept: "application/json" }), (_b = options.requestOptions) === null || _b === void 0 ? void 0 : _b.headers) }));
+}
+export async function _getDeserialize(result) {
+    const expectedStatuses = ["200"];
+    if (!expectedStatuses.includes(result.status)) {
+        throw createRestError(result);
+    }
+    return deploymentUnionDeserializer(result.body);
+}
+/** Get a deployed model. */
+export async function get(context, name, options = { requestOptions: {} }) {
+    const result = await _getSend(context, name, options);
+    return _getDeserialize(result);
+}
+//# sourceMappingURL=operations.js.map

package/dist/esm/api/deployments/operations.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"operations.js","sourceRoot":"","sources":["../../../../src/api/deployments/operations.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EACL,2BAA2B,EAG3B,4BAA4B,GAC7B,MAAM,wBAAwB,CAAC;AAEhC,OAAO,EAAE,iBAAiB,EAAE,MAAM,qCAAqC,CAAC;AACxE,OAAO,EAEL,uBAAuB,GACxB,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAGL,eAAe,EACf,mCAAmC,GACpC,MAAM,yBAAyB,CAAC;AAEjC,MAAM,UAAU,SAAS,CACvB,OAAe,EACf,UAAyC,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE/D,MAAM,IAAI,GAAG,iBAAiB,CAC5B,yFAAyF,EACzF;QACE,aAAa,EAAE,OAAO,CAAC,UAAU;QACjC,cAAc,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc;QACvC,SAAS,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,SAAS;QAC7B,cAAc,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc;QACvC,GAAG,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,GAAG;QACjB,IAAI,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,IAAI;QACnB,WAAW,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,WAAW;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,iCACxB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,gDACF,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,MAAK,SAAS;YACxC,CAAC,CAAC,EAAE,wBAAwB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,EAAE;YACxD,CAAC,CAAC,EAAE,CAAC,KACP,MAAM,EAAE,kBAAkB,KACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,MAA6B;IAClE,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,4BAA4B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AACnD,CAAC;AAED,8CAA8C;AAC9C,MAAM,UAAU,IAAI,CAClB,OAAe,EACf,UAAyC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE/D,OAAO,uBAAuB,CAC5B,OAAO,EACP,GAAG,EAAE,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,EACjC,gBAAgB,EAChB,CAAC,KAAK,CAAC,EACP,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,CAChD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,QAAQ,CACtB,OAAe,EACf,IAAY,EACZ,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE9D,MAAM,IAAI,GAAG,iBAAiB,CAC5B,mCAAmC,EACnC;QACE,IAAI,EAAE,IAAI;QACV,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,MAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc,0CAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,iCACxB,mCAAmC,CAAC,OAAO,CAAC,KAC/C,OAAO,gDACF,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,MAAK,SAAS;YACxC,CAAC,CAAC,EAAE,wBAAwB,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,eAAe,EAAE;YACxD,CAAC,CAAC,EAAE,CAAC,KACP,MAAM,EAAE,kBAAkB,KACvB,MAAA,OAAO,CAAC,cAAc,0CAAE,OAAO,KAEpC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CAAC,MAA6B;IACjE,MAAM,gBAAgB,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9C,MAAM,eAAe,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,2BAA2B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAClD,CAAC;AAED,4BAA4B;AAC5B,MAAM,CAAC,KAAK,UAAU,GAAG,CACvB,OAAe,EACf,IAAY,EACZ,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE9D,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACtD,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectContext as Client } from \"../index.js\";\nimport {\n  deploymentUnionDeserializer,\n  DeploymentUnion,\n  _PagedDeployment,\n  _pagedDeploymentDeserializer,\n} from \"../../models/models.js\";\nimport { DeploymentsListOptionalParams, DeploymentsGetOptionalParams } from \"./options.js\";\nimport { expandUrlTemplate } from \"../../static-helpers/urlTemplate.js\";\nimport {\n  PagedAsyncIterableIterator,\n  buildPagedAsyncIterator,\n} from \"../../static-helpers/pagingHelpers.js\";\nimport {\n  StreamableMethod,\n  PathUncheckedResponse,\n  createRestError,\n  operationOptionsToRequestParameters,\n} from \"@azure-rest/core-client\";\n\nexport function _listSend(\n  context: Client,\n  options: DeploymentsListOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/deployments{?api-version,modelPublisher,modelName,deploymentType,top,skip,maxpagesize}\",\n    {\n      \"api-version\": context.apiVersion,\n      modelPublisher: options?.modelPublisher,\n      modelName: options?.modelName,\n      deploymentType: options?.deploymentType,\n      top: options?.top,\n      skip: options?.skip,\n      maxpagesize: options?.maxpagesize,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).get({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      ...(options?.clientRequestId !== undefined\n        ? { \"x-ms-client-request-id\": options?.clientRequestId }\n        : {}),\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _listDeserialize(result: PathUncheckedResponse): Promise<_PagedDeployment> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return _pagedDeploymentDeserializer(result.body);\n}\n\n/** List all deployed models in the project */\nexport function list(\n  context: Client,\n  options: DeploymentsListOptionalParams = { requestOptions: {} },\n): PagedAsyncIterableIterator<DeploymentUnion> {\n  return buildPagedAsyncIterator(\n    context,\n    () => _listSend(context, options),\n    _listDeserialize,\n    [\"200\"],\n    { itemName: \"value\", nextLinkName: \"nextLink\" },\n  );\n}\n\nexport function _getSend(\n  context: Client,\n  name: string,\n  options: DeploymentsGetOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/deployments/{name}{?api-version}\",\n    {\n      name: name,\n      \"api-version\": context.apiVersion,\n    },\n    {\n      allowReserved: options?.requestOptions?.skipUrlEncoding,\n    },\n  );\n  return context.path(path).get({\n    ...operationOptionsToRequestParameters(options),\n    headers: {\n      ...(options?.clientRequestId !== undefined\n        ? { \"x-ms-client-request-id\": options?.clientRequestId }\n        : {}),\n      accept: \"application/json\",\n      ...options.requestOptions?.headers,\n    },\n  });\n}\n\nexport async function _getDeserialize(result: PathUncheckedResponse): Promise<DeploymentUnion> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return deploymentUnionDeserializer(result.body);\n}\n\n/** Get a deployed model. */\nexport async function get(\n  context: Client,\n  name: string,\n  options: DeploymentsGetOptionalParams = { requestOptions: {} },\n): Promise<DeploymentUnion> {\n  const result = await _getSend(context, name, options);\n  return _getDeserialize(result);\n}\n"]}

package/dist/esm/api/deployments/options.d.ts

@@ -0,0 +1,25 @@
+import { DeploymentType } from "../../models/models.js";
+import { OperationOptions } from "@azure-rest/core-client";
+/** Optional parameters. */
+export interface DeploymentsListOptionalParams extends OperationOptions {
+    /** Model publisher to filter models by */
+    modelPublisher?: string;
+    /** Model name (the publisher specific name) to filter models by */
+    modelName?: string;
+    /** Type of deployment to filter list by */
+    deploymentType?: DeploymentType;
+    /** The number of result items to return. */
+    top?: number;
+    /** The number of result items to skip. */
+    skip?: number;
+    /** The maximum number of result items per page. */
+    maxpagesize?: number;
+    /** An opaque, globally-unique, client-generated string identifier for the request. */
+    clientRequestId?: string;
+}
+/** Optional parameters. */
+export interface DeploymentsGetOptionalParams extends OperationOptions {
+    /** An opaque, globally-unique, client-generated string identifier for the request. */
+    clientRequestId?: string;
+}
+//# sourceMappingURL=options.d.ts.map

package/dist/esm/api/deployments/options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.d.ts","sourceRoot":"","sources":["../../../../src/api/deployments/options.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAE3D,2BAA2B;AAC3B,MAAM,WAAW,6BAA8B,SAAQ,gBAAgB;IACrE,0CAA0C;IAC1C,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,mEAAmE;IACnE,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,2CAA2C;IAC3C,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,4CAA4C;IAC5C,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,0CAA0C;IAC1C,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,mDAAmD;IACnD,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,sFAAsF;IACtF,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,2BAA2B;AAC3B,MAAM,WAAW,4BAA6B,SAAQ,gBAAgB;IACpE,sFAAsF;IACtF,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B"}

package/dist/esm/api/deployments/options.js

@@ -0,0 +1,4 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+export {};
+//# sourceMappingURL=options.js.map

package/dist/esm/api/deployments/options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.js","sourceRoot":"","sources":["../../../../src/api/deployments/options.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { DeploymentType } from \"../../models/models.js\";\nimport { OperationOptions } from \"@azure-rest/core-client\";\n\n/** Optional parameters. */\nexport interface DeploymentsListOptionalParams extends OperationOptions {\n  /** Model publisher to filter models by */\n  modelPublisher?: string;\n  /** Model name (the publisher specific name) to filter models by */\n  modelName?: string;\n  /** Type of deployment to filter list by */\n  deploymentType?: DeploymentType;\n  /** The number of result items to return. */\n  top?: number;\n  /** The number of result items to skip. */\n  skip?: number;\n  /** The maximum number of result items per page. */\n  maxpagesize?: number;\n  /** An opaque, globally-unique, client-generated string identifier for the request. */\n  clientRequestId?: string;\n}\n\n/** Optional parameters. */\nexport interface DeploymentsGetOptionalParams extends OperationOptions {\n  /** An opaque, globally-unique, client-generated string identifier for the request. */\n  clientRequestId?: string;\n}\n"]}

package/dist/esm/api/evaluations/index.d.ts

@@ -0,0 +1,3 @@
+export { createAgentEvaluation, createRun, list, get } from "./operations.js";
+export { EvaluationsCreateAgentEvaluationOptionalParams, EvaluationsCreateRunOptionalParams, EvaluationsListOptionalParams, EvaluationsGetOptionalParams, } from "./options.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/esm/api/evaluations/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/evaluations/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAC;AAC9E,OAAO,EACL,8CAA8C,EAC9C,kCAAkC,EAClC,6BAA6B,EAC7B,4BAA4B,GAC7B,MAAM,cAAc,CAAC"}

package/dist/esm/api/evaluations/index.js

@@ -0,0 +1,4 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+export { createAgentEvaluation, createRun, list, get } from "./operations.js";
+//# sourceMappingURL=index.js.map

package/dist/esm/api/evaluations/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/evaluations/index.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAElC,OAAO,EAAE,qBAAqB,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,MAAM,iBAAiB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport { createAgentEvaluation, createRun, list, get } from \"./operations.js\";\nexport {\n  EvaluationsCreateAgentEvaluationOptionalParams,\n  EvaluationsCreateRunOptionalParams,\n  EvaluationsListOptionalParams,\n  EvaluationsGetOptionalParams,\n} from \"./options.js\";\n"]}

package/dist/esm/api/evaluations/operations.d.ts

@@ -0,0 +1,22 @@
+import { AIProjectContext as Client } from "../index.js";
+import { Evaluation, EvaluationWithOptionalName, _PagedEvaluation, AgentEvaluationRequest, AgentEvaluation } from "../../models/models.js";
+import { EvaluationsCreateAgentEvaluationOptionalParams, EvaluationsCreateRunOptionalParams, EvaluationsListOptionalParams, EvaluationsGetOptionalParams } from "./options.js";
+import { PagedAsyncIterableIterator } from "../../static-helpers/pagingHelpers.js";
+import { StreamableMethod, PathUncheckedResponse } from "@azure-rest/core-client";
+export declare function _createAgentEvaluationSend(context: Client, evaluation: AgentEvaluationRequest, options?: EvaluationsCreateAgentEvaluationOptionalParams): StreamableMethod;
+export declare function _createAgentEvaluationDeserialize(result: PathUncheckedResponse): Promise<AgentEvaluation>;
+/** Creates an agent evaluation run. */
+export declare function createAgentEvaluation(context: Client, evaluation: AgentEvaluationRequest, options?: EvaluationsCreateAgentEvaluationOptionalParams): Promise<AgentEvaluation>;
+export declare function _createRunSend(context: Client, evaluation: EvaluationWithOptionalName, options?: EvaluationsCreateRunOptionalParams): StreamableMethod;
+export declare function _createRunDeserialize(result: PathUncheckedResponse): Promise<Evaluation>;
+/** Creates an evaluation run. */
+export declare function createRun(context: Client, evaluation: EvaluationWithOptionalName, options?: EvaluationsCreateRunOptionalParams): Promise<Evaluation>;
+export declare function _listSend(context: Client, options?: EvaluationsListOptionalParams): StreamableMethod;
+export declare function _listDeserialize(result: PathUncheckedResponse): Promise<_PagedEvaluation>;
+/** List evaluation runs */
+export declare function list(context: Client, options?: EvaluationsListOptionalParams): PagedAsyncIterableIterator<Evaluation>;
+export declare function _getSend(context: Client, name: string, options?: EvaluationsGetOptionalParams): StreamableMethod;
+export declare function _getDeserialize(result: PathUncheckedResponse): Promise<Evaluation>;
+/** Get an evaluation run by name. */
+export declare function get(context: Client, name: string, options?: EvaluationsGetOptionalParams): Promise<Evaluation>;
+//# sourceMappingURL=operations.d.ts.map

package/dist/esm/api/evaluations/operations.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/evaluations/operations.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EACL,UAAU,EACV,0BAA0B,EAG1B,gBAAgB,EAEhB,sBAAsB,EAEtB,eAAe,EAEhB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,8CAA8C,EAC9C,kCAAkC,EAClC,6BAA6B,EAC7B,4BAA4B,EAC7B,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AAEjC,wBAAgB,0BAA0B,CACxC,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,sBAAsB,EAClC,OAAO,GAAE,8CAER,GACA,gBAAgB,CAmBlB;AAED,wBAAsB,iCAAiC,CACrD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,eAAe,CAAC,CAO1B;AAED,uCAAuC;AACvC,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,sBAAsB,EAClC,OAAO,GAAE,8CAER,GACA,OAAO,CAAC,eAAe,CAAC,CAG1B;AAED,wBAAgB,cAAc,CAC5B,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,kCAA2D,GACnE,gBAAgB,CAmBlB;AAED,wBAAsB,qBAAqB,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC,CAO9F;AAED,iCAAiC;AACjC,wBAAsB,SAAS,CAC7B,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,kCAA2D,GACnE,OAAO,CAAC,UAAU,CAAC,CAGrB;AAED,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,6BAAsD,GAC9D,gBAAgB,CAuBlB;AAED,wBAAsB,gBAAgB,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAO/F;AAED,2BAA2B;AAC3B,wBAAgB,IAAI,CAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,6BAAsD,GAC9D,0BAA0B,CAAC,UAAU,CAAC,CAQxC;AAED,wBAAgB,QAAQ,CACtB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,4BAAqD,GAC7D,gBAAgB,CAqBlB;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,UAAU,CAAC,CAOxF;AAED,qCAAqC;AACrC,wBAAsB,GAAG,CACvB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,4BAAqD,GAC7D,OAAO,CAAC,UAAU,CAAC,CAGrB"}