@azure/ai-projects 2.0.0 → 2.0.2

package/dist/react-native/static-helpers/pollingHelpers.js

@@ -3,7 +3,7 @@
 import { createHttpPoller } from "@azure/core-lro";
 import { createRestError } from "@azure-rest/core-client";
 export function getLongRunningPoller(client, processResponseBody, expectedStatuses, options) {
-    const { restoreFrom, getInitialResponse, apiVersion } = options;
+    const { restoreFrom, getInitialResponse, apiVersion, pollHeaders, statusNormalizations } = options;
     if (!restoreFrom && !getInitialResponse) {
         throw new Error("Either restoreFrom or getInitialResponse must be specified");
     }
@@ -15,7 +15,7 @@ export function getLongRunningPoller(client, processResponseBody, expectedStatus
                 throw new Error("getInitialResponse is required when initializing a new poller");
             }
             initialResponse = await getInitialResponse();
-            return getLroResponse(initialResponse, expectedStatuses);
+            return getLroResponse(initialResponse, expectedStatuses, statusNormalizations);
         },
         sendPollRequest: async (path, pollOptions) => {
             // The poll request would both listen to the user provided abort signal and the poller's own abort signal
@@ -40,13 +40,15 @@ export function getLongRunningPoller(client, processResponseBody, expectedStatus
             let response;
             try {
                 const pollingPath = apiVersion ? addApiVersionToUrl(path, apiVersion) : path;
-                response = await client.pathUnchecked(pollingPath).get({ abortSignal });
+                response = await client
+                    .pathUnchecked(pollingPath)
+                    .get({ abortSignal, headers: pollHeaders });
             }
             finally {
                 options.abortSignal?.removeEventListener("abort", abortListener);
                 pollOptions?.abortSignal?.removeEventListener("abort", abortListener);
             }
-            return getLroResponse(response, expectedStatuses);
+            return getLroResponse(response, expectedStatuses, statusNormalizations);
         },
     };
     return createHttpPoller(poller, {
@@ -64,21 +66,26 @@ export function getLongRunningPoller(client, processResponseBody, expectedStatus
  * @param deserializeFn - deserialize function to convert Rest response to modular output
  * @returns - An LRO response that the LRO implementation understands
  */
-function getLroResponse(response, expectedStatuses) {
+function getLroResponse(response, expectedStatuses, statusNormalizations) {
     if (!expectedStatuses.includes(response.status)) {
         throw createRestError(response);
     }
+    const body = response.body;
+    const rawStatus = body?.["status"];
+    const normalizedBody = statusNormalizations && typeof rawStatus === "string" && rawStatus in statusNormalizations
+        ? { ...body, status: statusNormalizations[rawStatus] }
+        : body;
     return {
         flatResponse: response,
         rawResponse: {
             ...response,
             statusCode: Number.parseInt(response.status),
-            body: response.body,
+            body: normalizedBody,
         },
     };
 }
 /**
- * Adds the api-version query parameter on a URL if it's not present.
+ * Sets the api-version query parameter on a URL, replacing any existing value.
  * @param url - the URL to modify
  * @param apiVersion - the API version to set
  * @returns - the URL with the api-version query parameter set
@@ -86,10 +93,15 @@ function getLroResponse(response, expectedStatuses) {
 function addApiVersionToUrl(url, apiVersion) {
     // The base URL is only used for parsing and won't appear in the returned URL
     const urlObj = new URL(url, "https://microsoft.com");
-    if (!urlObj.searchParams.get("api-version")) {
+    const existingVersion = urlObj.searchParams.get("api-version");
+    if (!existingVersion) {
         // Append one if there is no apiVersion
         return `${url}${Array.from(urlObj.searchParams.keys()).length > 0 ? "&" : "?"}api-version=${apiVersion}`;
     }
-    return url;
+    if (existingVersion === apiVersion) {
+        return url;
+    }
+    // Replace the service-returned api-version with the client's configured version
+    return url.replace(`api-version=${existingVersion}`, `api-version=${apiVersion}`);
 }
 //# sourceMappingURL=pollingHelpers.js.map

package/dist/react-native/static-helpers/pollingHelpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"pollingHelpers.js","sourceRoot":"","sources":["../../../src/static-helpers/pollingHelpers.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AASlC,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAiC1D,MAAM,UAAU,oBAAoB,CAClC,MAAc,EACd,mBAA4D,EAC5D,gBAA0B,EAC1B,OAA+C;IAE/C,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC;IAChE,IAAI,CAAC,WAAW,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IACD,IAAI,eAAe,GAA0B,SAAS,CAAC;IACvD,MAAM,mBAAmB,GAAG,IAAI,eAAe,EAAE,CAAC;IAClD,MAAM,MAAM,GAAgC;QAC1C,kBAAkB,EAAE,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;YACnF,CAAC;YACD,eAAe,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC7C,OAAO,cAAc,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QAC3D,CAAC;QACD,eAAe,EAAE,KAAK,EACpB,IAAY,EACZ,WAEC,EACD,EAAE;YACF,yGAAyG;YACzG,SAAS,aAAa;gBACpB,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;YACD,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC/C,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC;gBACjC,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;iBAAM,IAAI,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;gBAC7C,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;iBAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBAChC,OAAO,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE;oBAC5D,IAAI,EAAE,IAAI;iBACX,CAAC,CAAC;gBACH,WAAW,EAAE,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE;oBACjE,IAAI,EAAE,IAAI;iBACX,CAAC,CAAC;YACL,CAAC;YACD,IAAI,QAAQ,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC7E,QAAQ,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC;YAC1E,CAAC;oBAAS,CAAC;gBACT,OAAO,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBACjE,WAAW,EAAE,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACxE,CAAC;YAED,OAAO,cAAc,CAAC,QAAqB,EAAE,gBAAgB,CAAC,CAAC;QACjE,CAAC;KACF,CAAC;IACF,OAAO,gBAAgB,CAAC,MAAM,EAAE;QAC9B,YAAY,EAAE,OAAO,EAAE,kBAAkB;QACzC,sBAAsB,EAAE,OAAO,EAAE,sBAAsB;QACvD,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,aAAa,EAAE,CAAC,MAAe,EAAE,EAAE;YACjC,OAAO,mBAAmB,CAAC,MAAmB,CAAC,CAAC;QAClD,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AACD;;;;;GAKG;AACH,SAAS,cAAc,CACrB,QAAmB,EACnB,gBAA0B;IAE1B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAChD,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;IAED,OAAO;QACL,YAAY,EAAE,QAAQ;QACtB,WAAW,EAAE;YACX,GAAG,QAAQ;YACX,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC5C,IAAI,EAAE,QAAQ,CAAC,IAAI;SACpB;KACF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,GAAW,EAAE,UAAkB;IACzD,6EAA6E;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,EAAE,uBAAuB,CAAC,CAAC;IACrD,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QAC5C,uCAAuC;QACvC,OAAO,GAAG,GAAG,GACX,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAC5D,eAAe,UAAU,EAAE,CAAC;IAC9B,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  PollerLike,\n  OperationState,\n  ResourceLocationConfig,\n  RunningOperation,\n  OperationResponse,\n} from \"@azure/core-lro\";\nimport { createHttpPoller } from \"@azure/core-lro\";\n\nimport type { Client, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { KnownApiVersions } from \"../models/models.js\";\n\nexport interface GetLongRunningPollerOptions<TResponse> {\n  /** Delay to wait until next poll, in milliseconds. */\n  updateIntervalInMs?: number;\n  /**\n   * The signal which can be used to abort requests.\n   */\n  abortSignal?: AbortSignalLike;\n  /**\n   * The potential location of the result of the LRO if specified by the LRO extension in the swagger.\n   */\n  resourceLocationConfig?: ResourceLocationConfig;\n  /**\n   * The original url of the LRO\n   * Should not be null when restoreFrom is set\n   */\n  initialRequestUrl?: string;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  restoreFrom?: string;\n  /**\n   * The api-version of the LRO\n   */\n  apiVersion?: KnownApiVersions;\n  /**\n   * The function to get the initial response\n   */\n  getInitialResponse?: () => PromiseLike<TResponse>;\n}\nexport function getLongRunningPoller<TResponse extends PathUncheckedResponse, TResult = void>(\n  client: Client,\n  processResponseBody: (result: TResponse) => Promise<TResult>,\n  expectedStatuses: string[],\n  options: GetLongRunningPollerOptions<TResponse>,\n): PollerLike<OperationState<TResult>, TResult> {\n  const { restoreFrom, getInitialResponse, apiVersion } = options;\n  if (!restoreFrom && !getInitialResponse) {\n    throw new Error(\"Either restoreFrom or getInitialResponse must be specified\");\n  }\n  let initialResponse: TResponse | undefined = undefined;\n  const pollAbortController = new AbortController();\n  const poller: RunningOperation<TResponse> = {\n    sendInitialRequest: async () => {\n      if (!getInitialResponse) {\n        throw new Error(\"getInitialResponse is required when initializing a new poller\");\n      }\n      initialResponse = await getInitialResponse();\n      return getLroResponse(initialResponse, expectedStatuses);\n    },\n    sendPollRequest: async (\n      path: string,\n      pollOptions?: {\n        abortSignal?: AbortSignalLike;\n      },\n    ) => {\n      // The poll request would both listen to the user provided abort signal and the poller's own abort signal\n      function abortListener(): void {\n        pollAbortController.abort();\n      }\n      const abortSignal = pollAbortController.signal;\n      if (options.abortSignal?.aborted) {\n        pollAbortController.abort();\n      } else if (pollOptions?.abortSignal?.aborted) {\n        pollAbortController.abort();\n      } else if (!abortSignal.aborted) {\n        options.abortSignal?.addEventListener(\"abort\", abortListener, {\n          once: true,\n        });\n        pollOptions?.abortSignal?.addEventListener(\"abort\", abortListener, {\n          once: true,\n        });\n      }\n      let response;\n      try {\n        const pollingPath = apiVersion ? addApiVersionToUrl(path, apiVersion) : path;\n        response = await client.pathUnchecked(pollingPath).get({ abortSignal });\n      } finally {\n        options.abortSignal?.removeEventListener(\"abort\", abortListener);\n        pollOptions?.abortSignal?.removeEventListener(\"abort\", abortListener);\n      }\n\n      return getLroResponse(response as TResponse, expectedStatuses);\n    },\n  };\n  return createHttpPoller(poller, {\n    intervalInMs: options?.updateIntervalInMs,\n    resourceLocationConfig: options?.resourceLocationConfig,\n    restoreFrom: options?.restoreFrom,\n    processResult: (result: unknown) => {\n      return processResponseBody(result as TResponse);\n    },\n  });\n}\n/**\n * Converts a Rest Client response to a response that the LRO implementation understands\n * @param response - a rest client http response\n * @param deserializeFn - deserialize function to convert Rest response to modular output\n * @returns - An LRO response that the LRO implementation understands\n */\nfunction getLroResponse<TResponse extends PathUncheckedResponse>(\n  response: TResponse,\n  expectedStatuses: string[],\n): OperationResponse<TResponse> {\n  if (!expectedStatuses.includes(response.status)) {\n    throw createRestError(response);\n  }\n\n  return {\n    flatResponse: response,\n    rawResponse: {\n      ...response,\n      statusCode: Number.parseInt(response.status),\n      body: response.body,\n    },\n  };\n}\n\n/**\n * Adds the api-version query parameter on a URL if it's not present.\n * @param url - the URL to modify\n * @param apiVersion - the API version to set\n * @returns - the URL with the api-version query parameter set\n */\nfunction addApiVersionToUrl(url: string, apiVersion: string): string {\n  // The base URL is only used for parsing and won't appear in the returned URL\n  const urlObj = new URL(url, \"https://microsoft.com\");\n  if (!urlObj.searchParams.get(\"api-version\")) {\n    // Append one if there is no apiVersion\n    return `${url}${\n      Array.from(urlObj.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n    }api-version=${apiVersion}`;\n  }\n  return url;\n}\n"]}
+{"version":3,"file":"pollingHelpers.js","sourceRoot":"","sources":["../../../src/static-helpers/pollingHelpers.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAUlC,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAGnD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AA2C1D,MAAM,UAAU,oBAAoB,CAClC,MAAc,EACd,mBAA4D,EAC5D,gBAA0B,EAC1B,OAA+C;IAE/C,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,UAAU,EAAE,WAAW,EAAE,oBAAoB,EAAE,GACtF,OAAO,CAAC;IACV,IAAI,CAAC,WAAW,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IACD,IAAI,eAAe,GAA0B,SAAS,CAAC;IACvD,MAAM,mBAAmB,GAAG,IAAI,eAAe,EAAE,CAAC;IAClD,MAAM,MAAM,GAAgC;QAC1C,kBAAkB,EAAE,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,+DAA+D,CAAC,CAAC;YACnF,CAAC;YACD,eAAe,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC7C,OAAO,cAAc,CAAC,eAAe,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,CAAC;QACjF,CAAC;QACD,eAAe,EAAE,KAAK,EACpB,IAAY,EACZ,WAEC,EACD,EAAE;YACF,yGAAyG;YACzG,SAAS,aAAa;gBACpB,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;YACD,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC/C,IAAI,OAAO,CAAC,WAAW,EAAE,OAAO,EAAE,CAAC;gBACjC,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;iBAAM,IAAI,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;gBAC7C,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;iBAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBAChC,OAAO,CAAC,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE;oBAC5D,IAAI,EAAE,IAAI;iBACX,CAAC,CAAC;gBACH,WAAW,EAAE,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE;oBACjE,IAAI,EAAE,IAAI;iBACX,CAAC,CAAC;YACL,CAAC;YACD,IAAI,QAAQ,CAAC;YACb,IAAI,CAAC;gBACH,MAAM,WAAW,GAAG,UAAU,CAAC,CAAC,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC7E,QAAQ,GAAG,MAAM,MAAM;qBACpB,aAAa,CAAC,WAAW,CAAC;qBAC1B,GAAG,CAAC,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC,CAAC;YAChD,CAAC;oBAAS,CAAC;gBACT,OAAO,CAAC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBACjE,WAAW,EAAE,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACxE,CAAC;YAED,OAAO,cAAc,CAAC,QAAqB,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,CAAC;QACvF,CAAC;KACF,CAAC;IACF,OAAO,gBAAgB,CAAC,MAAM,EAAE;QAC9B,YAAY,EAAE,OAAO,EAAE,kBAAkB;QACzC,sBAAsB,EAAE,OAAO,EAAE,sBAAsB;QACvD,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,aAAa,EAAE,CAAC,MAAe,EAAE,EAAE;YACjC,OAAO,mBAAmB,CAAC,MAAmB,CAAC,CAAC;QAClD,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AACD;;;;;GAKG;AACH,SAAS,cAAc,CACrB,QAAmB,EACnB,gBAA0B,EAC1B,oBAAsD;IAEtD,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAChD,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,IAAI,GAAG,QAAQ,CAAC,IAA2C,CAAC;IAClE,MAAM,SAAS,GAAG,IAAI,EAAE,CAAC,QAAQ,CAAC,CAAC;IACnC,MAAM,cAAc,GAClB,oBAAoB,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,IAAI,oBAAoB;QACxF,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,MAAM,EAAE,oBAAoB,CAAC,SAAS,CAAC,EAAE;QACtD,CAAC,CAAC,IAAI,CAAC;IAEX,OAAO;QACL,YAAY,EAAE,QAAQ;QACtB,WAAW,EAAE;YACX,GAAG,QAAQ;YACX,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC5C,IAAI,EAAE,cAAc;SACrB;KACF,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,SAAS,kBAAkB,CAAC,GAAW,EAAE,UAAkB;IACzD,6EAA6E;IAC7E,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,EAAE,uBAAuB,CAAC,CAAC;IACrD,MAAM,eAAe,GAAG,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC/D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,uCAAuC;QACvC,OAAO,GAAG,GAAG,GACX,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAC5D,eAAe,UAAU,EAAE,CAAC;IAC9B,CAAC;IAED,IAAI,eAAe,KAAK,UAAU,EAAE,CAAC;QACnC,OAAO,GAAG,CAAC;IACb,CAAC;IACD,gFAAgF;IAChF,OAAO,GAAG,CAAC,OAAO,CAAC,eAAe,eAAe,EAAE,EAAE,eAAe,UAAU,EAAE,CAAC,CAAC;AACpF,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport type {\n  PollerLike,\n  OperationState,\n  OperationStatus,\n  ResourceLocationConfig,\n  RunningOperation,\n  OperationResponse,\n} from \"@azure/core-lro\";\nimport { createHttpPoller } from \"@azure/core-lro\";\n\nimport type { Client, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { createRestError } from \"@azure-rest/core-client\";\nimport type { AbortSignalLike } from \"@azure/abort-controller\";\nimport type { KnownApiVersions } from \"../models/models.js\";\n\nexport interface GetLongRunningPollerOptions<TResponse> {\n  /** Delay to wait until next poll, in milliseconds. */\n  updateIntervalInMs?: number;\n  /**\n   * The signal which can be used to abort requests.\n   */\n  abortSignal?: AbortSignalLike;\n  /**\n   * The potential location of the result of the LRO if specified by the LRO extension in the swagger.\n   */\n  resourceLocationConfig?: ResourceLocationConfig;\n  /**\n   * The original url of the LRO\n   * Should not be null when restoreFrom is set\n   */\n  initialRequestUrl?: string;\n  /**\n   * A serialized poller which can be used to resume an existing paused Long-Running-Operation.\n   */\n  restoreFrom?: string;\n  /**\n   * The api-version of the LRO\n   */\n  apiVersion?: KnownApiVersions;\n  /**\n   * Additional headers to include on poll requests (e.g. feature-flag headers required by the endpoint).\n   */\n  pollHeaders?: Record<string, string>;\n  /**\n   * Optional mapping from service-specific status values to the core-lro status values\n   * (\"running\", \"succeeded\", \"failed\", \"canceled\"). Use this when the service returns\n   * non-standard terminal status strings (e.g. \"completed\" instead of \"succeeded\").\n   */\n  statusNormalizations?: Record<string, OperationStatus>;\n  /**\n   * The function to get the initial response\n   */\n  getInitialResponse?: () => PromiseLike<TResponse>;\n}\nexport function getLongRunningPoller<TResponse extends PathUncheckedResponse, TResult = void>(\n  client: Client,\n  processResponseBody: (result: TResponse) => Promise<TResult>,\n  expectedStatuses: string[],\n  options: GetLongRunningPollerOptions<TResponse>,\n): PollerLike<OperationState<TResult>, TResult> {\n  const { restoreFrom, getInitialResponse, apiVersion, pollHeaders, statusNormalizations } =\n    options;\n  if (!restoreFrom && !getInitialResponse) {\n    throw new Error(\"Either restoreFrom or getInitialResponse must be specified\");\n  }\n  let initialResponse: TResponse | undefined = undefined;\n  const pollAbortController = new AbortController();\n  const poller: RunningOperation<TResponse> = {\n    sendInitialRequest: async () => {\n      if (!getInitialResponse) {\n        throw new Error(\"getInitialResponse is required when initializing a new poller\");\n      }\n      initialResponse = await getInitialResponse();\n      return getLroResponse(initialResponse, expectedStatuses, statusNormalizations);\n    },\n    sendPollRequest: async (\n      path: string,\n      pollOptions?: {\n        abortSignal?: AbortSignalLike;\n      },\n    ) => {\n      // The poll request would both listen to the user provided abort signal and the poller's own abort signal\n      function abortListener(): void {\n        pollAbortController.abort();\n      }\n      const abortSignal = pollAbortController.signal;\n      if (options.abortSignal?.aborted) {\n        pollAbortController.abort();\n      } else if (pollOptions?.abortSignal?.aborted) {\n        pollAbortController.abort();\n      } else if (!abortSignal.aborted) {\n        options.abortSignal?.addEventListener(\"abort\", abortListener, {\n          once: true,\n        });\n        pollOptions?.abortSignal?.addEventListener(\"abort\", abortListener, {\n          once: true,\n        });\n      }\n      let response;\n      try {\n        const pollingPath = apiVersion ? addApiVersionToUrl(path, apiVersion) : path;\n        response = await client\n          .pathUnchecked(pollingPath)\n          .get({ abortSignal, headers: pollHeaders });\n      } finally {\n        options.abortSignal?.removeEventListener(\"abort\", abortListener);\n        pollOptions?.abortSignal?.removeEventListener(\"abort\", abortListener);\n      }\n\n      return getLroResponse(response as TResponse, expectedStatuses, statusNormalizations);\n    },\n  };\n  return createHttpPoller(poller, {\n    intervalInMs: options?.updateIntervalInMs,\n    resourceLocationConfig: options?.resourceLocationConfig,\n    restoreFrom: options?.restoreFrom,\n    processResult: (result: unknown) => {\n      return processResponseBody(result as TResponse);\n    },\n  });\n}\n/**\n * Converts a Rest Client response to a response that the LRO implementation understands\n * @param response - a rest client http response\n * @param deserializeFn - deserialize function to convert Rest response to modular output\n * @returns - An LRO response that the LRO implementation understands\n */\nfunction getLroResponse<TResponse extends PathUncheckedResponse>(\n  response: TResponse,\n  expectedStatuses: string[],\n  statusNormalizations?: Record<string, OperationStatus>,\n): OperationResponse<TResponse> {\n  if (!expectedStatuses.includes(response.status)) {\n    throw createRestError(response);\n  }\n\n  const body = response.body as Record<string, unknown> | undefined;\n  const rawStatus = body?.[\"status\"];\n  const normalizedBody =\n    statusNormalizations && typeof rawStatus === \"string\" && rawStatus in statusNormalizations\n      ? { ...body, status: statusNormalizations[rawStatus] }\n      : body;\n\n  return {\n    flatResponse: response,\n    rawResponse: {\n      ...response,\n      statusCode: Number.parseInt(response.status),\n      body: normalizedBody,\n    },\n  };\n}\n\n/**\n * Sets the api-version query parameter on a URL, replacing any existing value.\n * @param url - the URL to modify\n * @param apiVersion - the API version to set\n * @returns - the URL with the api-version query parameter set\n */\nfunction addApiVersionToUrl(url: string, apiVersion: string): string {\n  // The base URL is only used for parsing and won't appear in the returned URL\n  const urlObj = new URL(url, \"https://microsoft.com\");\n  const existingVersion = urlObj.searchParams.get(\"api-version\");\n  if (!existingVersion) {\n    // Append one if there is no apiVersion\n    return `${url}${\n      Array.from(urlObj.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n    }api-version=${apiVersion}`;\n  }\n\n  if (existingVersion === apiVersion) {\n    return url;\n  }\n  // Replace the service-returned api-version with the client's configured version\n  return url.replace(`api-version=${existingVersion}`, `api-version=${apiVersion}`);\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@azure/ai-projects",
-  "version": "2.0.0",
+  "version": "2.0.2",
   "description": "Azure AI Projects client library.",
   "engines": {
     "node": ">=20.0.0"
@@ -59,31 +59,31 @@
   },
   "devDependencies": {
     "@azure-rest/ai-inference": "^1.0.0-beta.6",
-    "@azure/arm-cognitiveservices": "7.6.0",
+    "@azure/arm-cognitiveservices": "8.1.0",
     "@azure/monitor-opentelemetry": "^1.16.0",
     "@azure/opentelemetry-instrumentation-azure-sdk": "1.0.0-beta.7",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/instrumentation": "^0.57.0",
     "@opentelemetry/sdk-trace-node": "^2.0.0",
     "@types/node": "^20.19.25",
-    "@vitest/browser-playwright": "^4.0.8",
-    "@vitest/coverage-istanbul": "^4.0.8",
+    "@vitest/browser-playwright": "^4.1.2",
+    "@vitest/coverage-istanbul": "^4.1.2",
     "@modelcontextprotocol/sdk": "^1.25.1",
     "zod": "^4.2.1",
     "cross-env": "^10.1.0",
     "dotenv": "^16.6.1",
     "eslint": "^9.39.1",
     "mkdirp": "^3.0.1",
-    "playwright": "^1.56.1",
+    "playwright": "^1.58.2",
     "prettier": "^3.6.2",
     "rimraf": "^6.1.0",
-    "typescript": "~5.9.3",
-    "vitest": "^4.0.8",
+    "typescript": "~6.0.2",
+    "vitest": "^4.1.2",
     "@azure-tools/test-credential": "^2.1.2",
     "@azure-tools/test-recorder": "^4.1.1",
-    "@azure-tools/test-utils-vitest": "^2.0.1",
     "@azure/dev-tool": "^1.0.0",
-    "@azure/eslint-plugin-azure-sdk": "^3.0.0"
+    "@azure/eslint-plugin-azure-sdk": "^3.0.0",
+    "@azure-tools/test-utils-vitest": "^2.0.1"
   },
   "exports": {
     "./package.json": "./package.json",