@azure/ai-projects 1.0.0-beta.8 → 1.0.0

package/dist/browser/api/connections/operations.js

@@ -1,22 +1,27 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-import { __asyncValues } from "tslib";
 import { connectionDeserializer, _pagedConnectionDeserializer, } 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("/connections{?api-version,connectionType,defaultConnection}", {
         "api-version": context.apiVersion,
-        connectionType: options === null || options === void 0 ? void 0 : options.connectionType,
-        defaultConnection: options === null || options === void 0 ? void 0 : options.defaultConnection,
+        connectionType: options?.connectionType,
+        defaultConnection: options?.defaultConnection,
     }, {
-        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            ...(options?.clientRequestId !== undefined
+                ? { "x-ms-client-request-id": options?.clientRequestId }
+                : {}),
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];
@@ -30,16 +35,22 @@ export function list(context, options = { requestOptions: {} }) {
     return buildPagedAsyncIterator(context, () => _listSend(context, options), _listDeserialize, ["200"], { itemName: "value", nextLinkName: "nextLink" });
 }
 export function _getWithCredentialsSend(context, name, options = { requestOptions: {} }) {
-    var _a, _b;
     const path = expandUrlTemplate("/connections/{name}/getConnectionWithCredentials{?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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            ...(options?.clientRequestId !== undefined
+                ? { "x-ms-client-request-id": options?.clientRequestId }
+                : {}),
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    return context.path(path).post(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 _getWithCredentialsDeserialize(result) {
     const expectedStatuses = ["200"];
@@ -54,16 +65,22 @@ export async function getWithCredentials(context, name, options = { requestOptio
     return _getWithCredentialsDeserialize(result);
 }
 export function _getSend(context, name, options = { requestOptions: {} }) {
-    var _a, _b;
     const path = expandUrlTemplate("/connections/{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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            ...(options?.clientRequestId !== undefined
+                ? { "x-ms-client-request-id": options?.clientRequestId }
+                : {}),
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];
@@ -90,32 +107,19 @@ export async function get(context, name, includeCredentials, options = { request
  * @throws Error if no default connection is found for the given type.
  */
 export async function getDefault(context, connectionType, includeCredentials = false) {
-    var _a, e_1, _b, _c;
     const listOptions = {
         connectionType,
         defaultConnection: true,
     };
     // Use the list function to find default connections of the specified type
     const connections = list(context, listOptions);
-    try {
-        // Find the first default connection
-        for (var _d = true, connections_1 = __asyncValues(connections), connections_1_1; connections_1_1 = await connections_1.next(), _a = connections_1_1.done, !_a; _d = true) {
-            _c = connections_1_1.value;
-            _d = false;
-            const connection = _c;
-            if (includeCredentials) {
-                // If credentials are requested, get the connection with credentials
-                return getWithCredentials(context, connection.name);
-            }
-            return connection;
-        }
-    }
-    catch (e_1_1) { e_1 = { error: e_1_1 }; }
-    finally {
-        try {
-            if (!_d && !_a && (_b = connections_1.return)) await _b.call(connections_1);
+    // Find the first default connection
+    for await (const connection of connections) {
+        if (includeCredentials) {
+            // If credentials are requested, get the connection with credentials
+            return getWithCredentials(context, connection.name);
         }
-        finally { if (e_1) throw e_1.error; }
+        return connection;
     }
     throw new Error(`No default connection found for type: ${connectionType}.`);
 }

package/dist/browser/api/connections/operations.js.map

@@ -1 +1 @@
-{"version":3,"file":"operations.js","sourceRoot":"","sources":["../../../../src/api/connections/operations.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;;AAGlC,OAAO,EAEL,sBAAsB,EAEtB,4BAA4B,GAE7B,MAAM,wBAAwB,CAAC;AAMhC,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,6DAA6D,EAC7D;QACE,aAAa,EAAE,OAAO,CAAC,UAAU;QACjC,cAAc,EAAE,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,cAAc;QACvC,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,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,qFAAqF;AACrF,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,uBAAuB,CACrC,OAAe,EACf,IAAY,EACZ,UAAuD,EAAE,cAAc,EAAE,EAAE,EAAE;;IAE7E,MAAM,IAAI,GAAG,iBAAiB,CAC5B,gEAAgE,EAChE;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,IAAI,iCACzB,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,8BAA8B,CAClD,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,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC;AAED,gEAAgE;AAChE,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,IAAY,EACZ,UAAuD,EAAE,cAAc,EAAE,EAAE,EAAE;IAE7E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACrE,OAAO,8BAA8B,CAAC,MAAM,CAAC,CAAC;AAChD,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,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC;AAED,0EAA0E;AAC1E,MAAM,CAAC,KAAK,UAAU,GAAG,CACvB,OAAe,EACf,IAAY,EACZ,kBAA4B,EAC5B,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE9D,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACtD,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,cAA8B,EAC9B,qBAA8B,KAAK;;IAEnC,MAAM,WAAW,GAAG;QAClB,cAAc;QACd,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;;QAE/C,oCAAoC;QACpC,KAA+B,eAAA,gBAAA,cAAA,WAAW,CAAA,iBAAA,2FAAE,CAAC;YAAd,2BAAW;YAAX,WAAW;YAA/B,MAAM,UAAU,KAAA,CAAA;YACzB,IAAI,kBAAkB,EAAE,CAAC;gBACvB,oEAAoE;gBACpE,OAAO,kBAAkB,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,UAAU,CAAC;QACpB,CAAC;;;;;;;;;IAED,MAAM,IAAI,KAAK,CAAC,yCAAyC,cAAc,GAAG,CAAC,CAAC;AAC9E,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectContext as Client } from \"../index.js\";\nimport {\n  Connection,\n  connectionDeserializer,\n  _PagedConnection,\n  _pagedConnectionDeserializer,\n  ConnectionType,\n} from \"../../models/models.js\";\nimport {\n  ConnectionsListOptionalParams,\n  ConnectionsGetWithCredentialsOptionalParams,\n  ConnectionsGetOptionalParams,\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\";\n\nexport function _listSend(\n  context: Client,\n  options: ConnectionsListOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections{?api-version,connectionType,defaultConnection}\",\n    {\n      \"api-version\": context.apiVersion,\n      connectionType: options?.connectionType,\n      defaultConnection: options?.defaultConnection,\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<_PagedConnection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return _pagedConnectionDeserializer(result.body);\n}\n\n/** List all connections in the project, without populating connection credentials */\nexport function list(\n  context: Client,\n  options: ConnectionsListOptionalParams = { requestOptions: {} },\n): PagedAsyncIterableIterator<Connection> {\n  return buildPagedAsyncIterator(\n    context,\n    () => _listSend(context, options),\n    _listDeserialize,\n    [\"200\"],\n    { itemName: \"value\", nextLinkName: \"nextLink\" },\n  );\n}\n\nexport function _getWithCredentialsSend(\n  context: Client,\n  name: string,\n  options: ConnectionsGetWithCredentialsOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections/{name}/getConnectionWithCredentials{?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).post({\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 _getWithCredentialsDeserialize(\n  result: PathUncheckedResponse,\n): Promise<Connection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return connectionDeserializer(result.body);\n}\n\n/** Get a connection by name, with its connection credentials */\nexport async function getWithCredentials(\n  context: Client,\n  name: string,\n  options: ConnectionsGetWithCredentialsOptionalParams = { requestOptions: {} },\n): Promise<Connection> {\n  const result = await _getWithCredentialsSend(context, name, options);\n  return _getWithCredentialsDeserialize(result);\n}\n\nexport function _getSend(\n  context: Client,\n  name: string,\n  options: ConnectionsGetOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections/{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<Connection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return connectionDeserializer(result.body);\n}\n\n/** Get a connection by name, without populating connection credentials */\nexport async function get(\n  context: Client,\n  name: string,\n  includeCredentials?: boolean,\n  options: ConnectionsGetOptionalParams = { requestOptions: {} },\n): Promise<Connection> {\n  if (includeCredentials) {\n    return getWithCredentials(context, name, options);\n  }\n  const result = await _getSend(context, name, options);\n  return _getDeserialize(result);\n}\n\n/**\n * Get the default connection for a given connection type.\n *\n * @param context - The AIProjectContext client\n * @param connectionType - The type of the connection. Required.\n * @param includeCredentials - Whether to include credentials in the response. Default is false.\n * @returns A Connection object\n * @throws Error if no default connection is found for the given type.\n */\nexport async function getDefault(\n  context: Client,\n  connectionType: ConnectionType,\n  includeCredentials: boolean = false,\n): Promise<Connection> {\n  const listOptions = {\n    connectionType,\n    defaultConnection: true,\n  };\n\n  // Use the list function to find default connections of the specified type\n  const connections = list(context, listOptions);\n\n  // Find the first default connection\n  for await (const connection of connections) {\n    if (includeCredentials) {\n      // If credentials are requested, get the connection with credentials\n      return getWithCredentials(context, connection.name);\n    }\n    return connection;\n  }\n\n  throw new Error(`No default connection found for type: ${connectionType}.`);\n}\n"]}
+{"version":3,"file":"operations.js","sourceRoot":"","sources":["../../../../src/api/connections/operations.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAEL,sBAAsB,EAEtB,4BAA4B,GAE7B,MAAM,wBAAwB,CAAC;AAMhC,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,6DAA6D,EAC7D;QACE,aAAa,EAAE,OAAO,CAAC,UAAU;QACjC,cAAc,EAAE,OAAO,EAAE,cAAc;QACvC,iBAAiB,EAAE,OAAO,EAAE,iBAAiB;KAC9C,EACD;QACE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC;QAC5B,GAAG,mCAAmC,CAAC,OAAO,CAAC;QAC/C,OAAO,EAAE;YACP,GAAG,CAAC,OAAO,EAAE,eAAe,KAAK,SAAS;gBACxC,CAAC,CAAC,EAAE,wBAAwB,EAAE,OAAO,EAAE,eAAe,EAAE;gBACxD,CAAC,CAAC,EAAE,CAAC;YACP,MAAM,EAAE,kBAAkB;YAC1B,GAAG,OAAO,CAAC,cAAc,EAAE,OAAO;SACnC;KACF,CAAC,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,qFAAqF;AACrF,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,uBAAuB,CACrC,OAAe,EACf,IAAY,EACZ,UAAuD,EAAE,cAAc,EAAE,EAAE,EAAE;IAE7E,MAAM,IAAI,GAAG,iBAAiB,CAC5B,gEAAgE,EAChE;QACE,IAAI,EAAE,IAAI;QACV,aAAa,EAAE,OAAO,CAAC,UAAU;KAClC,EACD;QACE,aAAa,EAAE,OAAO,EAAE,cAAc,EAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QAC7B,GAAG,mCAAmC,CAAC,OAAO,CAAC;QAC/C,OAAO,EAAE;YACP,GAAG,CAAC,OAAO,EAAE,eAAe,KAAK,SAAS;gBACxC,CAAC,CAAC,EAAE,wBAAwB,EAAE,OAAO,EAAE,eAAe,EAAE;gBACxD,CAAC,CAAC,EAAE,CAAC;YACP,MAAM,EAAE,kBAAkB;YAC1B,GAAG,OAAO,CAAC,cAAc,EAAE,OAAO;SACnC;KACF,CAAC,CAAC;AACL,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,8BAA8B,CAClD,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,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC;AAED,gEAAgE;AAChE,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAe,EACf,IAAY,EACZ,UAAuD,EAAE,cAAc,EAAE,EAAE,EAAE;IAE7E,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACrE,OAAO,8BAA8B,CAAC,MAAM,CAAC,CAAC;AAChD,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,OAAO,EAAE,cAAc,EAAE,eAAe;KACxD,CACF,CAAC;IACF,OAAO,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC;QAC5B,GAAG,mCAAmC,CAAC,OAAO,CAAC;QAC/C,OAAO,EAAE;YACP,GAAG,CAAC,OAAO,EAAE,eAAe,KAAK,SAAS;gBACxC,CAAC,CAAC,EAAE,wBAAwB,EAAE,OAAO,EAAE,eAAe,EAAE;gBACxD,CAAC,CAAC,EAAE,CAAC;YACP,MAAM,EAAE,kBAAkB;YAC1B,GAAG,OAAO,CAAC,cAAc,EAAE,OAAO;SACnC;KACF,CAAC,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,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;AAC7C,CAAC;AAED,0EAA0E;AAC1E,MAAM,CAAC,KAAK,UAAU,GAAG,CACvB,OAAe,EACf,IAAY,EACZ,kBAA4B,EAC5B,UAAwC,EAAE,cAAc,EAAE,EAAE,EAAE;IAE9D,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,kBAAkB,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACtD,OAAO,eAAe,CAAC,MAAM,CAAC,CAAC;AACjC,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,OAAe,EACf,cAA8B,EAC9B,qBAA8B,KAAK;IAEnC,MAAM,WAAW,GAAG;QAClB,cAAc;QACd,iBAAiB,EAAE,IAAI;KACxB,CAAC;IAEF,0EAA0E;IAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAE/C,oCAAoC;IACpC,IAAI,KAAK,EAAE,MAAM,UAAU,IAAI,WAAW,EAAE,CAAC;QAC3C,IAAI,kBAAkB,EAAE,CAAC;YACvB,oEAAoE;YACpE,OAAO,kBAAkB,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,yCAAyC,cAAc,GAAG,CAAC,CAAC;AAC9E,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectContext as Client } from \"../index.js\";\nimport {\n  Connection,\n  connectionDeserializer,\n  _PagedConnection,\n  _pagedConnectionDeserializer,\n  ConnectionType,\n} from \"../../models/models.js\";\nimport {\n  ConnectionsListOptionalParams,\n  ConnectionsGetWithCredentialsOptionalParams,\n  ConnectionsGetOptionalParams,\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\";\n\nexport function _listSend(\n  context: Client,\n  options: ConnectionsListOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections{?api-version,connectionType,defaultConnection}\",\n    {\n      \"api-version\": context.apiVersion,\n      connectionType: options?.connectionType,\n      defaultConnection: options?.defaultConnection,\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<_PagedConnection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return _pagedConnectionDeserializer(result.body);\n}\n\n/** List all connections in the project, without populating connection credentials */\nexport function list(\n  context: Client,\n  options: ConnectionsListOptionalParams = { requestOptions: {} },\n): PagedAsyncIterableIterator<Connection> {\n  return buildPagedAsyncIterator(\n    context,\n    () => _listSend(context, options),\n    _listDeserialize,\n    [\"200\"],\n    { itemName: \"value\", nextLinkName: \"nextLink\" },\n  );\n}\n\nexport function _getWithCredentialsSend(\n  context: Client,\n  name: string,\n  options: ConnectionsGetWithCredentialsOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections/{name}/getConnectionWithCredentials{?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).post({\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 _getWithCredentialsDeserialize(\n  result: PathUncheckedResponse,\n): Promise<Connection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return connectionDeserializer(result.body);\n}\n\n/** Get a connection by name, with its connection credentials */\nexport async function getWithCredentials(\n  context: Client,\n  name: string,\n  options: ConnectionsGetWithCredentialsOptionalParams = { requestOptions: {} },\n): Promise<Connection> {\n  const result = await _getWithCredentialsSend(context, name, options);\n  return _getWithCredentialsDeserialize(result);\n}\n\nexport function _getSend(\n  context: Client,\n  name: string,\n  options: ConnectionsGetOptionalParams = { requestOptions: {} },\n): StreamableMethod {\n  const path = expandUrlTemplate(\n    \"/connections/{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<Connection> {\n  const expectedStatuses = [\"200\"];\n  if (!expectedStatuses.includes(result.status)) {\n    throw createRestError(result);\n  }\n\n  return connectionDeserializer(result.body);\n}\n\n/** Get a connection by name, without populating connection credentials */\nexport async function get(\n  context: Client,\n  name: string,\n  includeCredentials?: boolean,\n  options: ConnectionsGetOptionalParams = { requestOptions: {} },\n): Promise<Connection> {\n  if (includeCredentials) {\n    return getWithCredentials(context, name, options);\n  }\n  const result = await _getSend(context, name, options);\n  return _getDeserialize(result);\n}\n\n/**\n * Get the default connection for a given connection type.\n *\n * @param context - The AIProjectContext client\n * @param connectionType - The type of the connection. Required.\n * @param includeCredentials - Whether to include credentials in the response. Default is false.\n * @returns A Connection object\n * @throws Error if no default connection is found for the given type.\n */\nexport async function getDefault(\n  context: Client,\n  connectionType: ConnectionType,\n  includeCredentials: boolean = false,\n): Promise<Connection> {\n  const listOptions = {\n    connectionType,\n    defaultConnection: true,\n  };\n\n  // Use the list function to find default connections of the specified type\n  const connections = list(context, listOptions);\n\n  // Find the first default connection\n  for await (const connection of connections) {\n    if (includeCredentials) {\n      // If credentials are requested, get the connection with credentials\n      return getWithCredentials(context, connection.name);\n    }\n    return connection;\n  }\n\n  throw new Error(`No default connection found for type: ${connectionType}.`);\n}\n"]}

package/dist/browser/api/datasets/operations.d.ts

@@ -1,4 +1,4 @@
-import { AIProjectContext as Client } from "../index.js";
+import { DatasetUploadInternalOptions, AIProjectContext as Client } from "../index.js";
 import { _PagedDatasetVersion, DatasetVersionUnion, PendingUploadRequest, PendingUploadResponse, AssetCredentialResponse } from "../../models/models.js";
 import { DatasetsGetCredentialsOptionalParams, DatasetsPendingUploadOptionalParams, DatasetsCreateOrUpdateOptionalParams, DatasetsDeleteOptionalParams, DatasetsGetOptionalParams, DatasetsListOptionalParams, DatasetsListVersionsOptionalParams } from "./options.js";
 import { PagedAsyncIterableIterator } from "../../static-helpers/pagingHelpers.js";
@@ -7,16 +7,16 @@ export declare function _getCredentialsSend(context: Client, name: string, versi
 export declare function _getCredentialsDeserialize(result: PathUncheckedResponse): Promise<AssetCredentialResponse>;
 /** Get the SAS credential to access the storage account associated with a Dataset version. */
 export declare function getCredentials(context: Client, name: string, version: string, options?: DatasetsGetCredentialsOptionalParams): Promise<AssetCredentialResponse>;
-export declare function _pendingUploadSend(context: Client, name: string, version: string, body: PendingUploadRequest, options?: DatasetsPendingUploadOptionalParams): StreamableMethod;
+export declare function _pendingUploadSend(context: Client, name: string, version: string, pendingUploadRequest: PendingUploadRequest, options?: DatasetsPendingUploadOptionalParams): StreamableMethod;
 export declare function _pendingUploadDeserialize(result: PathUncheckedResponse): Promise<PendingUploadResponse>;
 /** Start a new or get an existing pending upload of a dataset for a specific version. */
-export declare function pendingUpload(context: Client, name: string, version: string, body: PendingUploadRequest, options?: DatasetsPendingUploadOptionalParams): Promise<PendingUploadResponse>;
-export declare function uploadFile(context: Client, name: string, version: string, filePath: string, connectionName?: string): Promise<DatasetVersionUnion>;
-export declare function uploadFolder(context: Client, name: string, version: string, folderPath: string, connectionName?: string): Promise<DatasetVersionUnion>;
-export declare function _createOrUpdateSend(context: Client, name: string, version: string, body: DatasetVersionUnion, options?: DatasetsCreateOrUpdateOptionalParams): StreamableMethod;
+export declare function pendingUpload(context: Client, name: string, version: string, pendingUploadRequest: PendingUploadRequest, options?: DatasetsPendingUploadOptionalParams): Promise<PendingUploadResponse>;
+export declare function uploadFile(context: Client, name: string, version: string, filePath: string, options?: DatasetUploadInternalOptions): Promise<DatasetVersionUnion>;
+export declare function uploadFolder(context: Client, name: string, version: string, folderPath: string, options?: DatasetUploadInternalOptions): Promise<DatasetVersionUnion>;
+export declare function _createOrUpdateSend(context: Client, name: string, version: string, datasetVersion: DatasetVersionUnion, options?: DatasetsCreateOrUpdateOptionalParams): StreamableMethod;
 export declare function _createOrUpdateDeserialize(result: PathUncheckedResponse): Promise<DatasetVersionUnion>;
 /** Create a new or update an existing DatasetVersion with the given version id */
-export declare function createOrUpdate(context: Client, name: string, version: string, body: DatasetVersionUnion, options?: DatasetsCreateOrUpdateOptionalParams): Promise<DatasetVersionUnion>;
+export declare function createOrUpdate(context: Client, name: string, version: string, datasetVersion: DatasetVersionUnion, options?: DatasetsCreateOrUpdateOptionalParams): Promise<DatasetVersionUnion>;
 export declare function _$deleteSend(context: Client, name: string, version: string, options?: DatasetsDeleteOptionalParams): StreamableMethod;
 export declare function _$deleteDeserialize(result: PathUncheckedResponse): Promise<void>;
 /** Delete the specific version of the DatasetVersion */

package/dist/browser/api/datasets/operations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/datasets/operations.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EACL,oBAAoB,EAIpB,mBAAmB,EACnB,oBAAoB,EAEpB,qBAAqB,EAErB,uBAAuB,EAExB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,oCAAoC,EACpC,mCAAmC,EACnC,oCAAoC,EACpC,4BAA4B,EAC5B,yBAAyB,EACzB,0BAA0B,EAC1B,kCAAkC,EACnC,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AAGjC,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,oCAA6D,GACrE,gBAAgB,CAmBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,uBAAuB,CAAC,CAOlC;AAED,8FAA8F;AAC9F,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,oCAA6D,GACrE,OAAO,CAAC,uBAAuB,CAAC,CAGlC;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,oBAAoB,EAC1B,OAAO,GAAE,mCAA4D,GACpE,gBAAgB,CAqBlB;AAED,wBAAsB,yBAAyB,CAC7C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC,CAOhC;AAED,yFAAyF;AACzF,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,oBAAoB,EAC1B,OAAO,GAAE,mCAA4D,GACpE,OAAO,CAAC,qBAAqB,CAAC,CAGhC;AA+DD,wBAAsB,UAAU,CAC9B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,cAAc,CAAC,EAAE,MAAM,GACtB,OAAO,CAAC,mBAAmB,CAAC,CA+B9B;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,EAClB,cAAc,CAAC,EAAE,MAAM,GACtB,OAAO,CAAC,mBAAmB,CAAC,CAwE9B;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,mBAAmB,EACzB,OAAO,GAAE,oCAA6D,GACrE,gBAAgB,CAqBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,mBAAmB,CAAC,CAO9B;AAED,kFAAkF;AAClF,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,mBAAmB,EACzB,OAAO,GAAE,oCAA6D,GACrE,OAAO,CAAC,mBAAmB,CAAC,CAG9B;AAED,wBAAgB,YAAY,CAC1B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,4BAAqD,GAC7D,gBAAgB,CAmBlB;AAED,wBAAsB,mBAAmB,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAOtF;AAED,wDAAwD;AACxD;;;;GAIG;AACH,wBAAsB,OAAO,CAC3B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,4BAAqD,GAC7D,OAAO,CAAC,IAAI,CAAC,CAGf;AAED,wBAAgB,QAAQ,CACtB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,yBAAkD,GAC1D,gBAAgB,CAmBlB;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAOjG;AAED,qDAAqD;AACrD,wBAAsB,GAAG,CACvB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,yBAAkD,GAC1D,OAAO,CAAC,mBAAmB,CAAC,CAG9B;AAED,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0BAAmD,GAC3D,gBAAgB,CAiBlB;AAED,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,oBAAoB,CAAC,CAO/B;AAED,qDAAqD;AACrD,wBAAgB,IAAI,CAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0BAAmD,GAC3D,0BAA0B,CAAC,mBAAmB,CAAC,CAQjD;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,kCAA2D,GACnE,gBAAgB,CAkBlB;AAED,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,oBAAoB,CAAC,CAO/B;AAED,oDAAoD;AACpD,wBAAgB,YAAY,CAC1B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,kCAA2D,GACnE,0BAA0B,CAAC,mBAAmB,CAAC,CAQjD"}
+{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/datasets/operations.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,4BAA4B,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACvF,OAAO,EACL,oBAAoB,EAIpB,mBAAmB,EACnB,oBAAoB,EAEpB,qBAAqB,EAErB,uBAAuB,EAExB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,oCAAoC,EACpC,mCAAmC,EACnC,oCAAoC,EACpC,4BAA4B,EAC5B,yBAAyB,EACzB,0BAA0B,EAC1B,kCAAkC,EACnC,MAAM,cAAc,CAAC;AAEtB,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AAGjC,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,oCAA6D,GACrE,gBAAgB,CAmBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,uBAAuB,CAAC,CAOlC;AAED,8FAA8F;AAC9F,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,oCAA6D,GACrE,OAAO,CAAC,uBAAuB,CAAC,CAGlC;AAED,wBAAgB,kBAAkB,CAChC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,oBAAoB,EAAE,oBAAoB,EAC1C,OAAO,GAAE,mCAA4D,GACpE,gBAAgB,CAqBlB;AAED,wBAAsB,yBAAyB,CAC7C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,qBAAqB,CAAC,CAOhC;AAED,yFAAyF;AACzF,wBAAsB,aAAa,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,oBAAoB,EAAE,oBAAoB,EAC1C,OAAO,GAAE,mCAA4D,GACpE,OAAO,CAAC,qBAAqB,CAAC,CAGhC;AAoED,wBAAsB,UAAU,CAC9B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,QAAQ,EAAE,MAAM,EAChB,OAAO,CAAC,EAAE,4BAA4B,GACrC,OAAO,CAAC,mBAAmB,CAAC,CA+B9B;AAED,wBAAsB,YAAY,CAChC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,EAClB,OAAO,CAAC,EAAE,4BAA4B,GACrC,OAAO,CAAC,mBAAmB,CAAC,CA6E9B;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,mBAAmB,EACnC,OAAO,GAAE,oCAA6D,GACrE,gBAAgB,CAqBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,mBAAmB,CAAC,CAO9B;AAED,kFAAkF;AAClF,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,mBAAmB,EACnC,OAAO,GAAE,oCAA6D,GACrE,OAAO,CAAC,mBAAmB,CAAC,CAG9B;AAED,wBAAgB,YAAY,CAC1B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,4BAAqD,GAC7D,gBAAgB,CAmBlB;AAED,wBAAsB,mBAAmB,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,CAOtF;AAED,wDAAwD;AACxD;;;;GAIG;AACH,wBAAsB,OAAO,CAC3B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,4BAAqD,GAC7D,OAAO,CAAC,IAAI,CAAC,CAGf;AAED,wBAAgB,QAAQ,CACtB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,yBAAkD,GAC1D,gBAAgB,CAmBlB;AAED,wBAAsB,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO,CAAC,mBAAmB,CAAC,CAOjG;AAED,qDAAqD;AACrD,wBAAsB,GAAG,CACvB,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,yBAAkD,GAC1D,OAAO,CAAC,mBAAmB,CAAC,CAG9B;AAED,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0BAAmD,GAC3D,gBAAgB,CAiBlB;AAED,wBAAsB,gBAAgB,CACpC,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,oBAAoB,CAAC,CAO/B;AAED,qDAAqD;AACrD,wBAAgB,IAAI,CAClB,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0BAAmD,GAC3D,0BAA0B,CAAC,mBAAmB,CAAC,CAQjD;AAED,wBAAgB,iBAAiB,CAC/B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,kCAA2D,GACnE,gBAAgB,CAkBlB;AAED,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,oBAAoB,CAAC,CAO/B;AAED,oDAAoD;AACpD,wBAAgB,YAAY,CAC1B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,kCAA2D,GACnE,0BAA0B,CAAC,mBAAmB,CAAC,CAQjD"}

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

@@ -1,23 +1,28 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 /* eslint-disable tsdoc/syntax */
-import * as fs from "fs";
-import * as nodePath from "path";
+import fs from "fs";
+import nodePath from "path";
 import { _pagedDatasetVersionDeserializer, datasetVersionUnionSerializer, datasetVersionUnionDeserializer, pendingUploadRequestSerializer, pendingUploadResponseDeserializer, 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, 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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    return context.path(path).post(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 _getCredentialsDeserialize(result) {
     const expectedStatuses = ["200"];
@@ -31,16 +36,23 @@ export async function getCredentials(context, name, version, options = { request
     const result = await _getCredentialsSend(context, name, version, options);
     return _getCredentialsDeserialize(result);
 }
-export function _pendingUploadSend(context, name, version, body, options = { requestOptions: {} }) {
-    var _a, _b;
+export function _pendingUploadSend(context, name, version, pendingUploadRequest, options = { requestOptions: {} }) {
     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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).post({
+        ...operationOptionsToRequestParameters(options),
+        contentType: "application/json",
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+        body: pendingUploadRequestSerializer(pendingUploadRequest),
     });
-    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"];
@@ -50,15 +62,13 @@ export async function _pendingUploadDeserialize(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);
+export async function pendingUpload(context, name, version, pendingUploadRequest, options = { requestOptions: {} }) {
+    const result = await _pendingUploadSend(context, name, version, pendingUploadRequest, 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, 
-// Optional connection name for the storage account to be used
-connectionName) {
-    var _a;
+async function createDatasetAndGetItsContainer(context, name, version, options) {
+    const { connectionName, projectOptions = {} } = options || {};
     // Start a pending upload to get the container URL with SAS token
     const pendingUploadResponse = await pendingUpload(context, name, version, {
         pendingUploadType: "BlobReference",
@@ -69,7 +79,7 @@ connectionName) {
     if (!blobReference) {
         throw new Error("Blob reference for consumption is not present");
     }
-    if (!((_a = blobReference.credential) === null || _a === void 0 ? void 0 : _a.type)) {
+    if (!blobReference.credential?.type) {
         throw new Error("Credential type is not present");
     }
     if (blobReference.credential.type !== "SAS") {
@@ -87,12 +97,16 @@ connectionName) {
     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.credential.sasUri);
+    const pipeline = containerClient["storageClientContext"].pipeline;
+    for (const { policy } of projectOptions.additionalPolicies ?? []) {
+        pipeline.addPolicy(policy, { afterPhase: "Sign" });
+    }
     return {
         containerClient,
         version,
     };
 }
-export async function uploadFile(context, name, version, filePath, connectionName) {
+export async function uploadFile(context, name, version, filePath, options) {
     // if file does not exist
     const fileExists = fs.existsSync(filePath);
     if (!fileExists) {
@@ -103,7 +117,7 @@ export async function uploadFile(context, name, version, filePath, connectionNam
     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, connectionName);
+    const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(context, name, version, options);
     // file name as blob name
     const blobName = nodePath.basename(filePath);
     const blockBlobClient = containerClient.getBlockBlobClient(blobName);
@@ -116,7 +130,7 @@ export async function uploadFile(context, name, version, filePath, connectionNam
     });
     return datasetVersion;
 }
-export async function uploadFolder(context, name, version, folderPath, connectionName) {
+export async function uploadFolder(context, name, version, folderPath, options) {
     // Check if the folder exists
     const folderExists = fs.existsSync(folderPath);
     if (!folderExists) {
@@ -127,7 +141,7 @@ export async function uploadFolder(context, name, version, folderPath, connectio
     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, connectionName);
+    const { containerClient, version: outputVersion } = await createDatasetAndGetItsContainer(context, name, version, options);
     // Helper function to recursively get all files in a directory
     async function getAllFiles(dir, fileList = []) {
         const files = await fs.promises.readdir(dir);
@@ -145,11 +159,15 @@ export async function uploadFolder(context, name, version, folderPath, connectio
     }
     // Get all files in the folder
     const allFiles = await getAllFiles(folderPath);
-    if (allFiles.length === 0) {
+    let filteredFiles = allFiles;
+    if (options?.filePattern) {
+        filteredFiles = allFiles.filter((file) => options.filePattern?.test(file));
+    }
+    if (filteredFiles.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) {
+    for (const filePath of filteredFiles) {
         // 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}'`);
@@ -169,16 +187,23 @@ export async function uploadFolder(context, name, version, folderPath, connectio
     });
     return datasetVersion;
 }
-export function _createOrUpdateSend(context, name, version, body, options = { requestOptions: {} }) {
-    var _a, _b;
+export function _createOrUpdateSend(context, name, version, datasetVersion, options = { requestOptions: {} }) {
     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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).patch({
+        ...operationOptionsToRequestParameters(options),
+        contentType: "application/merge-patch+json",
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
+        body: datasetVersionUnionSerializer(datasetVersion),
     });
-    return context.path(path).patch(Object.assign(Object.assign({}, operationOptionsToRequestParameters(options)), { contentType: "application/merge-patch+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"];
@@ -188,20 +213,25 @@ export async function _createOrUpdateDeserialize(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);
+export async function createOrUpdate(context, name, version, datasetVersion, options = { requestOptions: {} }) {
+    const result = await _createOrUpdateSend(context, name, version, datasetVersion, 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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).delete({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];
@@ -221,15 +251,20 @@ export async function $delete(context, name, version, options = { requestOptions
     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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];
@@ -244,13 +279,18 @@ export async function get(context, name, version, options = { requestOptions: {}
     return _getDeserialize(result);
 }
 export function _listSend(context, options = { requestOptions: {} }) {
-    var _a, _b;
     const path = expandUrlTemplate("/datasets{?api-version}", {
         "api-version": context.apiVersion,
     }, {
-        allowReserved: (_a = options === null || options === void 0 ? void 0 : options.requestOptions) === null || _a === void 0 ? void 0 : _a.skipUrlEncoding,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];
@@ -264,14 +304,19 @@ 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}", {
         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,
+        allowReserved: options?.requestOptions?.skipUrlEncoding,
+    });
+    return context.path(path).get({
+        ...operationOptionsToRequestParameters(options),
+        headers: {
+            accept: "application/json",
+            ...options.requestOptions?.headers,
+        },
     });
-    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"];