@azure/ai-projects 2.0.0-beta.1 → 2.0.0-beta.3

package/dist/browser/restorePollerHelpers.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"restorePollerHelpers.d.ts","sourceRoot":"","sources":["../../src/restorePollerHelpers.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAGvD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EACL,UAAU,EACV,cAAc,EAGf,MAAM,iBAAiB,CAAC;AAEzB;;GAEG;AACH,MAAM,WAAW,oBAAoB,CACnC,OAAO,EACP,SAAS,SAAS,qBAAqB,GAAG,qBAAqB,CAC/D,SAAQ,gBAAgB;IACxB,sDAAsD;IACtD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B,qDAAqD;IACrD,mBAAmB,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CAC/D;AAED;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,SAAS,SAAS,qBAAqB,EAAE,OAAO,EAC5E,MAAM,EAAE,eAAe,EACvB,eAAe,EAAE,MAAM,EACvB,eAAe,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EACjF,OAAO,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,GACtC,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CA+B9C"}
+{"version":3,"file":"restorePollerHelpers.d.ts","sourceRoot":"","sources":["../../src/restorePollerHelpers.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAGvD,OAAO,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EACL,UAAU,EACV,cAAc,EAGf,MAAM,iBAAiB,CAAC;AAEzB;;GAEG;AACH,MAAM,WAAW,oBAAoB,CACnC,OAAO,EACP,SAAS,SAAS,qBAAqB,GAAG,qBAAqB,CAC/D,SAAQ,gBAAgB;IACxB,sDAAsD;IACtD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B;;OAEG;IACH,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B,qDAAqD;IACrD,mBAAmB,CAAC,EAAE,CAAC,MAAM,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CAC/D;AAED;;;;GAIG;AACH,wBAAgB,aAAa,CAAC,SAAS,SAAS,qBAAqB,EAAE,OAAO,EAC5E,MAAM,EAAE,eAAe,EACvB,eAAe,EAAE,MAAM,EACvB,eAAe,EAAE,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,EACjF,OAAO,CAAC,EAAE,oBAAoB,CAAC,OAAO,CAAC,GACtC,UAAU,CAAC,cAAc,CAAC,OAAO,CAAC,EAAE,OAAO,CAAC,CA2C9C"}

package/dist/browser/restorePollerHelpers.js

@@ -1,6 +1,6 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
-import { _updateMemoriesDeserialize } from "./api/memoryStores/operations.js";
+import { createMemoryStoreUpdateMemoriesPoller } from "./api/memoryStores/memoryStoreUpdateMemoriesPoller.js";
 import { getLongRunningPoller } from "./static-helpers/pollingHelpers.js";
 import { deserializeState, } from "@azure/core-lro";
 /**
@@ -15,8 +15,16 @@ export function restorePoller(client, serializedState, sourceOperation, options)
         throw new Error(`Invalid serialized state: ${serializedState} for sourceOperation ${sourceOperation?.name}`);
     }
     const resourceLocationConfig = metadata?.["resourceLocationConfig"];
-    const { deserializer, expectedStatuses = [] } = getDeserializationHelper(initialRequestUrl, requestMethod) ?? {};
-    const deserializeHelper = options?.processResponseBody ?? deserializer;
+    const helper = getDeserializationHelper(initialRequestUrl, requestMethod);
+    if (helper?.kind === "memoryUpdate") {
+        return createMemoryStoreUpdateMemoriesPoller(client["_client"] ?? client, helper.expectedStatuses ?? ["202", "200"], undefined, {
+            updateIntervalInMs: options?.updateIntervalInMs,
+            abortSignal: options?.abortSignal,
+            restoreFrom: serializedState,
+        });
+    }
+    const deserializeHelper = options?.processResponseBody ?? helper?.deserializer;
+    const expectedStatuses = helper?.expectedStatuses ?? [];
     if (!deserializeHelper) {
         throw new Error(`Please ensure the operation is in this client! We can't find its deserializeHelper for ${sourceOperation?.name}.`);
     }
@@ -30,7 +38,7 @@ export function restorePoller(client, serializedState, sourceOperation, options)
 }
 const deserializeMap = {
     "POST /memory_stores/{name}:update_memories": {
-        deserializer: _updateMemoriesDeserialize,
+        kind: "memoryUpdate",
         expectedStatuses: ["202", "200"],
     },
 };

package/dist/browser/restorePollerHelpers.js.map

@@ -1 +1 @@
-{"version":3,"file":"restorePollerHelpers.js","sourceRoot":"","sources":["../../src/restorePollerHelpers.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,0BAA0B,EAAE,MAAM,kCAAkC,CAAC;AAC9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAG1E,OAAO,EAGL,gBAAgB,GAEjB,MAAM,iBAAiB,CAAC;AAmBzB;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAC3B,MAAuB,EACvB,eAAuB,EACvB,eAAiF,EACjF,OAAuC;IAEvC,MAAM,YAAY,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,iBAAiB,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC;IACpE,IAAI,CAAC,iBAAiB,IAAI,CAAC,aAAa,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CACb,6BAA6B,eAAe,wBAAwB,eAAe,EAAE,IAAI,EAAE,CAC5F,CAAC;IACJ,CAAC;IACD,MAAM,sBAAsB,GAAG,QAAQ,EAAE,CAAC,wBAAwB,CAErD,CAAC;IACd,MAAM,EAAE,YAAY,EAAE,gBAAgB,GAAG,EAAE,EAAE,GAC3C,wBAAwB,CAAC,iBAAiB,EAAE,aAAa,CAAC,IAAI,EAAE,CAAC;IACnE,MAAM,iBAAiB,GAAG,OAAO,EAAE,mBAAmB,IAAI,YAAY,CAAC;IACvE,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,0FAA0F,eAAe,EAAE,IAAI,GAAG,CACnH,CAAC;IACJ,CAAC;IACD,OAAO,oBAAoB,CACxB,MAAc,CAAC,SAAS,CAAC,IAAI,MAAM,EACpC,iBAA4D,EAC5D,gBAAgB,EAChB;QACE,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;QAC/C,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,sBAAsB;QACtB,WAAW,EAAE,eAAe;QAC5B,iBAAiB;KAClB,CACF,CAAC;AACJ,CAAC;AAQD,MAAM,cAAc,GAA0C;IAC5D,4CAA4C,EAAE;QAC5C,YAAY,EAAE,0BAA0B;QACxC,gBAAgB,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;KACjC;CACF,CAAC;AAEF,SAAS,wBAAwB,CAC/B,MAAc,EACd,MAAc;IAEd,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC;IACtC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAElC,+CAA+C;IAC/C,2CAA2C;IAC3C,8CAA8C;IAC9C,IAAI,UAAU,GAAG,CAAC,CAAC,EACjB,YAA+C,CAAC;IAElD,0CAA0C;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC1D,0DAA0D;QAC1D,gBAAgB;QAChB,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,SAAS;QACX,CAAC;QACD,MAAM,aAAa,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAC7C,gCAAgC;QAChC,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAEhD,6DAA6D;QAC7D,IAAI,KAAK,GAAG,IAAI,CAAC;QACjB,KAAK,IAAI,CAAC,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7F,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjF,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAC/C,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;gBAClC,4DAA4D;gBAC5D,qDAAqD;gBACrD,eAAe;gBACf,6BAA6B;gBAC7B,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAC1E,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CACnB,CAAC;gBAEF,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,KAAK,GAAG,KAAK,CAAC;oBACd,MAAM;gBACR,CAAC;gBACD,SAAS;YACX,CAAC;YAED,8CAA8C;YAC9C,wDAAwD;YACxD,2CAA2C;YAC3C,IAAI,cAAc,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvC,KAAK,GAAG,KAAK,CAAC;gBACd,MAAM;YACR,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,sEAAsE;QACtE,IAAI,KAAK,IAAI,aAAa,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YAC/C,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC;YAClC,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc;IACvC,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectClient } from \"./aiProjectClient.js\";\nimport { _updateMemoriesDeserialize } from \"./api/memoryStores/operations.js\";\nimport { getLongRunningPoller } from \"./static-helpers/pollingHelpers.js\";\nimport { OperationOptions, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { AbortSignalLike } from \"@azure/abort-controller\";\nimport {\n  PollerLike,\n  OperationState,\n  deserializeState,\n  ResourceLocationConfig,\n} from \"@azure/core-lro\";\n\n/**\n * Options for restoring a poller.\n */\nexport interface RestorePollerOptions<\n  TResult,\n  TResponse extends PathUncheckedResponse = PathUncheckedResponse,\n> extends OperationOptions {\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  /** Deserialization function for raw response body */\n  processResponseBody?: (result: TResponse) => Promise<TResult>;\n}\n\n/**\n * Creates a poller from the serialized state of another poller. This can be\n * useful when you want to create pollers on a different host or a poller\n * needs to be constructed after the original one is not in scope.\n */\nexport function restorePoller<TResponse extends PathUncheckedResponse, TResult>(\n  client: AIProjectClient,\n  serializedState: string,\n  sourceOperation: (...args: any[]) => PollerLike<OperationState<TResult>, TResult>,\n  options?: RestorePollerOptions<TResult>,\n): PollerLike<OperationState<TResult>, TResult> {\n  const pollerConfig = deserializeState(serializedState).config;\n  const { initialRequestUrl, requestMethod, metadata } = pollerConfig;\n  if (!initialRequestUrl || !requestMethod) {\n    throw new Error(\n      `Invalid serialized state: ${serializedState} for sourceOperation ${sourceOperation?.name}`,\n    );\n  }\n  const resourceLocationConfig = metadata?.[\"resourceLocationConfig\"] as\n    | ResourceLocationConfig\n    | undefined;\n  const { deserializer, expectedStatuses = [] } =\n    getDeserializationHelper(initialRequestUrl, requestMethod) ?? {};\n  const deserializeHelper = options?.processResponseBody ?? deserializer;\n  if (!deserializeHelper) {\n    throw new Error(\n      `Please ensure the operation is in this client! We can't find its deserializeHelper for ${sourceOperation?.name}.`,\n    );\n  }\n  return getLongRunningPoller(\n    (client as any)[\"_client\"] ?? client,\n    deserializeHelper as (result: TResponse) => Promise<TResult>,\n    expectedStatuses,\n    {\n      updateIntervalInMs: options?.updateIntervalInMs,\n      abortSignal: options?.abortSignal,\n      resourceLocationConfig,\n      restoreFrom: serializedState,\n      initialRequestUrl,\n    },\n  );\n}\n\ninterface DeserializationHelper {\n  // eslint-disable-next-line @typescript-eslint/no-unsafe-function-type\n  deserializer: Function;\n  expectedStatuses: string[];\n}\n\nconst deserializeMap: Record<string, DeserializationHelper> = {\n  \"POST /memory_stores/{name}:update_memories\": {\n    deserializer: _updateMemoriesDeserialize,\n    expectedStatuses: [\"202\", \"200\"],\n  },\n};\n\nfunction getDeserializationHelper(\n  urlStr: string,\n  method: string,\n): DeserializationHelper | undefined {\n  const path = new URL(urlStr).pathname;\n  const pathParts = path.split(\"/\");\n\n  // Traverse list to match the longest candidate\n  // matchedLen: the length of candidate path\n  // matchedValue: the matched status code array\n  let matchedLen = -1,\n    matchedValue: DeserializationHelper | undefined;\n\n  // Iterate the responseMap to find a match\n  for (const [key, value] of Object.entries(deserializeMap)) {\n    // Extracting the path from the map key which is in format\n    // GET /path/foo\n    if (!key.startsWith(method)) {\n      continue;\n    }\n    const candidatePath = getPathFromMapKey(key);\n    // Get each part of the url path\n    const candidateParts = candidatePath.split(\"/\");\n\n    // track if we have found a match to return the values found.\n    let found = true;\n    for (let i = candidateParts.length - 1, j = pathParts.length - 1; i >= 1 && j >= 1; i--, j--) {\n      if (candidateParts[i]?.startsWith(\"{\") && candidateParts[i]?.indexOf(\"}\") !== -1) {\n        const start = candidateParts[i]!.indexOf(\"}\") + 1,\n          end = candidateParts[i]?.length;\n        // If the current part of the candidate is a \"template\" part\n        // Try to use the suffix of pattern to match the path\n        // {guid} ==> $\n        // {guid}:export ==> :export$\n        const isMatched = new RegExp(`${candidateParts[i]?.slice(start, end)}`).test(\n          pathParts[j] || \"\",\n        );\n\n        if (!isMatched) {\n          found = false;\n          break;\n        }\n        continue;\n      }\n\n      // If the candidate part is not a template and\n      // the parts don't match mark the candidate as not found\n      // to move on with the next candidate path.\n      if (candidateParts[i] !== pathParts[j]) {\n        found = false;\n        break;\n      }\n    }\n\n    // We finished evaluating the current candidate parts\n    // Update the matched value if and only if we found the longer pattern\n    if (found && candidatePath.length > matchedLen) {\n      matchedLen = candidatePath.length;\n      matchedValue = value;\n    }\n  }\n\n  return matchedValue;\n}\n\nfunction getPathFromMapKey(mapKey: string): string {\n  const pathStart = mapKey.indexOf(\"/\");\n  return mapKey.slice(pathStart);\n}\n"]}
+{"version":3,"file":"restorePollerHelpers.js","sourceRoot":"","sources":["../../src/restorePollerHelpers.ts"],"names":[],"mappings":"AAAA,uCAAuC;AACvC,kCAAkC;AAGlC,OAAO,EAAE,qCAAqC,EAAE,MAAM,uDAAuD,CAAC;AAC9G,OAAO,EAAE,oBAAoB,EAAE,MAAM,oCAAoC,CAAC;AAG1E,OAAO,EAGL,gBAAgB,GAEjB,MAAM,iBAAiB,CAAC;AAmBzB;;;;GAIG;AACH,MAAM,UAAU,aAAa,CAC3B,MAAuB,EACvB,eAAuB,EACvB,eAAiF,EACjF,OAAuC;IAEvC,MAAM,YAAY,GAAG,gBAAgB,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,iBAAiB,EAAE,aAAa,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC;IACpE,IAAI,CAAC,iBAAiB,IAAI,CAAC,aAAa,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CACb,6BAA6B,eAAe,wBAAwB,eAAe,EAAE,IAAI,EAAE,CAC5F,CAAC;IACJ,CAAC;IACD,MAAM,sBAAsB,GAAG,QAAQ,EAAE,CAAC,wBAAwB,CAErD,CAAC;IACd,MAAM,MAAM,GAAG,wBAAwB,CAAC,iBAAiB,EAAE,aAAa,CAAC,CAAC;IAC1E,IAAI,MAAM,EAAE,IAAI,KAAK,cAAc,EAAE,CAAC;QACpC,OAAO,qCAAqC,CACzC,MAAc,CAAC,SAAS,CAAC,IAAI,MAAM,EACpC,MAAM,CAAC,gBAAgB,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,EACzC,SAAS,EACT;YACE,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;YAC/C,WAAW,EAAE,OAAO,EAAE,WAAW;YACjC,WAAW,EAAE,eAAe;SAC7B,CACyD,CAAC;IAC/D,CAAC;IACD,MAAM,iBAAiB,GAAG,OAAO,EAAE,mBAAmB,IAAI,MAAM,EAAE,YAAY,CAAC;IAC/E,MAAM,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,EAAE,CAAC;IACxD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CACb,0FAA0F,eAAe,EAAE,IAAI,GAAG,CACnH,CAAC;IACJ,CAAC;IACD,OAAO,oBAAoB,CACxB,MAAc,CAAC,SAAS,CAAC,IAAI,MAAM,EACpC,iBAA4D,EAC5D,gBAAgB,EAChB;QACE,kBAAkB,EAAE,OAAO,EAAE,kBAAkB;QAC/C,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,sBAAsB;QACtB,WAAW,EAAE,eAAe;QAC5B,iBAAiB;KAClB,CACF,CAAC;AACJ,CAAC;AASD,MAAM,cAAc,GAA0C;IAC5D,4CAA4C,EAAE;QAC5C,IAAI,EAAE,cAAc;QACpB,gBAAgB,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;KACjC;CACF,CAAC;AAEF,SAAS,wBAAwB,CAC/B,MAAc,EACd,MAAc;IAEd,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC;IACtC,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAElC,+CAA+C;IAC/C,2CAA2C;IAC3C,8CAA8C;IAC9C,IAAI,UAAU,GAAG,CAAC,CAAC,EACjB,YAA+C,CAAC;IAElD,0CAA0C;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAC1D,0DAA0D;QAC1D,gBAAgB;QAChB,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC5B,SAAS;QACX,CAAC;QACD,MAAM,aAAa,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC;QAC7C,gCAAgC;QAChC,MAAM,cAAc,GAAG,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAEhD,6DAA6D;QAC7D,IAAI,KAAK,GAAG,IAAI,CAAC;QACjB,KAAK,IAAI,CAAC,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7F,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBACjF,MAAM,KAAK,GAAG,cAAc,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAC/C,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC;gBAClC,4DAA4D;gBAC5D,qDAAqD;gBACrD,eAAe;gBACf,6BAA6B;gBAC7B,MAAM,SAAS,GAAG,IAAI,MAAM,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAC1E,SAAS,CAAC,CAAC,CAAC,IAAI,EAAE,CACnB,CAAC;gBAEF,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,KAAK,GAAG,KAAK,CAAC;oBACd,MAAM;gBACR,CAAC;gBACD,SAAS;YACX,CAAC;YAED,8CAA8C;YAC9C,wDAAwD;YACxD,2CAA2C;YAC3C,IAAI,cAAc,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvC,KAAK,GAAG,KAAK,CAAC;gBACd,MAAM;YACR,CAAC;QACH,CAAC;QAED,qDAAqD;QACrD,sEAAsE;QACtE,IAAI,KAAK,IAAI,aAAa,CAAC,MAAM,GAAG,UAAU,EAAE,CAAC;YAC/C,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC;YAClC,YAAY,GAAG,KAAK,CAAC;QACvB,CAAC;IACH,CAAC;IAED,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc;IACvC,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,OAAO,MAAM,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;AACjC,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectClient } from \"./aiProjectClient.js\";\nimport { createMemoryStoreUpdateMemoriesPoller } from \"./api/memoryStores/memoryStoreUpdateMemoriesPoller.js\";\nimport { getLongRunningPoller } from \"./static-helpers/pollingHelpers.js\";\nimport { OperationOptions, PathUncheckedResponse } from \"@azure-rest/core-client\";\nimport { AbortSignalLike } from \"@azure/abort-controller\";\nimport {\n  PollerLike,\n  OperationState,\n  deserializeState,\n  ResourceLocationConfig,\n} from \"@azure/core-lro\";\n\n/**\n * Options for restoring a poller.\n */\nexport interface RestorePollerOptions<\n  TResult,\n  TResponse extends PathUncheckedResponse = PathUncheckedResponse,\n> extends OperationOptions {\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  /** Deserialization function for raw response body */\n  processResponseBody?: (result: TResponse) => Promise<TResult>;\n}\n\n/**\n * Creates a poller from the serialized state of another poller. This can be\n * useful when you want to create pollers on a different host or a poller\n * needs to be constructed after the original one is not in scope.\n */\nexport function restorePoller<TResponse extends PathUncheckedResponse, TResult>(\n  client: AIProjectClient,\n  serializedState: string,\n  sourceOperation: (...args: any[]) => PollerLike<OperationState<TResult>, TResult>,\n  options?: RestorePollerOptions<TResult>,\n): PollerLike<OperationState<TResult>, TResult> {\n  const pollerConfig = deserializeState(serializedState).config;\n  const { initialRequestUrl, requestMethod, metadata } = pollerConfig;\n  if (!initialRequestUrl || !requestMethod) {\n    throw new Error(\n      `Invalid serialized state: ${serializedState} for sourceOperation ${sourceOperation?.name}`,\n    );\n  }\n  const resourceLocationConfig = metadata?.[\"resourceLocationConfig\"] as\n    | ResourceLocationConfig\n    | undefined;\n  const helper = getDeserializationHelper(initialRequestUrl, requestMethod);\n  if (helper?.kind === \"memoryUpdate\") {\n    return createMemoryStoreUpdateMemoriesPoller(\n      (client as any)[\"_client\"] ?? client,\n      helper.expectedStatuses ?? [\"202\", \"200\"],\n      undefined,\n      {\n        updateIntervalInMs: options?.updateIntervalInMs,\n        abortSignal: options?.abortSignal,\n        restoreFrom: serializedState,\n      },\n    ) as unknown as PollerLike<OperationState<TResult>, TResult>;\n  }\n  const deserializeHelper = options?.processResponseBody ?? helper?.deserializer;\n  const expectedStatuses = helper?.expectedStatuses ?? [];\n  if (!deserializeHelper) {\n    throw new Error(\n      `Please ensure the operation is in this client! We can't find its deserializeHelper for ${sourceOperation?.name}.`,\n    );\n  }\n  return getLongRunningPoller(\n    (client as any)[\"_client\"] ?? client,\n    deserializeHelper as (result: TResponse) => Promise<TResult>,\n    expectedStatuses,\n    {\n      updateIntervalInMs: options?.updateIntervalInMs,\n      abortSignal: options?.abortSignal,\n      resourceLocationConfig,\n      restoreFrom: serializedState,\n      initialRequestUrl,\n    },\n  );\n}\n\ninterface DeserializationHelper {\n  // eslint-disable-next-line @typescript-eslint/no-unsafe-function-type\n  deserializer?: Function;\n  expectedStatuses?: string[];\n  kind?: \"memoryUpdate\";\n}\n\nconst deserializeMap: Record<string, DeserializationHelper> = {\n  \"POST /memory_stores/{name}:update_memories\": {\n    kind: \"memoryUpdate\",\n    expectedStatuses: [\"202\", \"200\"],\n  },\n};\n\nfunction getDeserializationHelper(\n  urlStr: string,\n  method: string,\n): DeserializationHelper | undefined {\n  const path = new URL(urlStr).pathname;\n  const pathParts = path.split(\"/\");\n\n  // Traverse list to match the longest candidate\n  // matchedLen: the length of candidate path\n  // matchedValue: the matched status code array\n  let matchedLen = -1,\n    matchedValue: DeserializationHelper | undefined;\n\n  // Iterate the responseMap to find a match\n  for (const [key, value] of Object.entries(deserializeMap)) {\n    // Extracting the path from the map key which is in format\n    // GET /path/foo\n    if (!key.startsWith(method)) {\n      continue;\n    }\n    const candidatePath = getPathFromMapKey(key);\n    // Get each part of the url path\n    const candidateParts = candidatePath.split(\"/\");\n\n    // track if we have found a match to return the values found.\n    let found = true;\n    for (let i = candidateParts.length - 1, j = pathParts.length - 1; i >= 1 && j >= 1; i--, j--) {\n      if (candidateParts[i]?.startsWith(\"{\") && candidateParts[i]?.indexOf(\"}\") !== -1) {\n        const start = candidateParts[i]!.indexOf(\"}\") + 1,\n          end = candidateParts[i]?.length;\n        // If the current part of the candidate is a \"template\" part\n        // Try to use the suffix of pattern to match the path\n        // {guid} ==> $\n        // {guid}:export ==> :export$\n        const isMatched = new RegExp(`${candidateParts[i]?.slice(start, end)}`).test(\n          pathParts[j] || \"\",\n        );\n\n        if (!isMatched) {\n          found = false;\n          break;\n        }\n        continue;\n      }\n\n      // If the candidate part is not a template and\n      // the parts don't match mark the candidate as not found\n      // to move on with the next candidate path.\n      if (candidateParts[i] !== pathParts[j]) {\n        found = false;\n        break;\n      }\n    }\n\n    // We finished evaluating the current candidate parts\n    // Update the matched value if and only if we found the longer pattern\n    if (found && candidatePath.length > matchedLen) {\n      matchedLen = candidatePath.length;\n      matchedValue = value;\n    }\n  }\n\n  return matchedValue;\n}\n\nfunction getPathFromMapKey(mapKey: string): string {\n  const pathStart = mapKey.indexOf(\"/\");\n  return mapKey.slice(pathStart);\n}\n"]}

package/dist/commonjs/aiProjectClient.d.ts

@@ -10,6 +10,7 @@ import { EvaluatorsOperations } from "./classic/evaluators/index.js";
 import { IndexesOperations } from "./classic/indexes/index.js";
 import { InsightsOperations } from "./classic/insights/index.js";
 import { MemoryStoresOperations } from "./classic/memoryStores/index.js";
+import { TelemetryOperations } from "./classic/telemetry/index.js";
 import { RedTeamsOperations } from "./classic/redTeams/index.js";
 import { SchedulesOperations } from "./classic/schedules/index.js";
 import { TokenCredential } from "@azure/core-auth";
@@ -30,6 +31,7 @@ export { AIProjectClientOptionalParams } from "./api/aiProjectContext.js";
  * @property {ConnectionsOperations} connections - The operation groups for connections
  * @property {MemoryStoresOperations} memoryStores - The operation groups for memoryStores
  * @property {AgentsOperations} agents - The operation groups for agents
+ * @property {TelemetryOperations} telemetry - The operation groups for telemetry
  * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client
  * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client
  */
@@ -64,6 +66,8 @@ export declare class AIProjectClient {
     readonly memoryStores: MemoryStoresOperations;
     /** The operation groups for agents */
     readonly agents: AgentsOperations;
+    /** The operation groups for telemetry */
+    readonly telemetry: TelemetryOperations;
     /**
      * gets the OpenAI client
      * @returns the OpenAI client

package/dist/commonjs/aiProjectClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"aiProjectClient.d.ts","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAIA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAqC,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAClG,OAAO,EAAE,gBAAgB,EAAwB,MAAM,2BAA2B,CAAC;AACnF,OAAO,EAAE,qBAAqB,EAA6B,MAAM,gCAAgC,CAAC;AAClG,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EAAE,qBAAqB,EAA6B,MAAM,gCAAgC,CAAC;AAClG,OAAO,EACL,yBAAyB,EAE1B,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EACL,8BAA8B,EAE/B,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAA4B,MAAM,+BAA+B,CAAC;AAC/F,OAAO,EAAE,iBAAiB,EAAyB,MAAM,4BAA4B,CAAC;AACtF,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EACL,sBAAsB,EAEvB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EAAE,mBAAmB,EAA2B,MAAM,8BAA8B,CAAC;AAC5F,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAGnD,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAE1E;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,qBAAqB,CAAmB;IAChD,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAgC;gBAG9C,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,EAC3B,OAAO,GAAE,6BAAkC;IAoC7C,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,0CAA0C;IAC1C,SAAgB,UAAU,EAAE,oBAAoB,CAAC;IACjD,oDAAoD;IACpD,SAAgB,oBAAoB,EAAE,8BAA8B,CAAC;IACrE,+CAA+C;IAC/C,SAAgB,eAAe,EAAE,yBAAyB,CAAC;IAC3D,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,uCAAuC;IACvC,SAAgB,OAAO,EAAE,iBAAiB,CAAC;IAC3C,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,4CAA4C;IAC5C,SAAgB,YAAY,EAAE,sBAAsB,CAAC;IACrD,sCAAsC;IACtC,SAAgB,MAAM,EAAE,gBAAgB,CAAC;IACzC;;;OAGG;IACU,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC;IAc/C;;;OAGG;IACH,IAAW,QAAQ,IAAI,MAAM,CAE5B;CACF"}
+{"version":3,"file":"aiProjectClient.d.ts","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":"AAIA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,OAAO,EAAqC,6BAA6B,EAAE,MAAM,gBAAgB,CAAC;AAClG,OAAO,EAAE,gBAAgB,EAAwB,MAAM,2BAA2B,CAAC;AACnF,OAAO,EAAE,qBAAqB,EAA6B,MAAM,gCAAgC,CAAC;AAClG,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EAAE,qBAAqB,EAA6B,MAAM,gCAAgC,CAAC;AAClG,OAAO,EACL,yBAAyB,EAE1B,MAAM,oCAAoC,CAAC;AAC5C,OAAO,EACL,8BAA8B,EAE/B,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,oBAAoB,EAA4B,MAAM,+BAA+B,CAAC;AAC/F,OAAO,EAAE,iBAAiB,EAAyB,MAAM,4BAA4B,CAAC;AACtF,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EACL,sBAAsB,EAEvB,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAA2B,MAAM,8BAA8B,CAAC;AAC5F,OAAO,EAAE,kBAAkB,EAA0B,MAAM,6BAA6B,CAAC;AACzF,OAAO,EAAE,mBAAmB,EAA2B,MAAM,8BAA8B,CAAC;AAC5F,OAAO,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAInD,OAAO,EAAE,6BAA6B,EAAE,MAAM,2BAA2B,CAAC;AAE1E;;;;;;;;;;;;;;;;;;;GAmBG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,qBAAqB,CAAmB;IAChD,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,SAAS,CAAS;IAC1B,OAAO,CAAC,WAAW,CAAkB;IACrC,OAAO,CAAC,QAAQ,CAAgC;gBAG9C,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,EAC3B,OAAO,GAAE,6BAAkC;IAqC7C,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,0CAA0C;IAC1C,SAAgB,UAAU,EAAE,oBAAoB,CAAC;IACjD,oDAAoD;IACpD,SAAgB,oBAAoB,EAAE,8BAA8B,CAAC;IACrE,+CAA+C;IAC/C,SAAgB,eAAe,EAAE,yBAAyB,CAAC;IAC3D,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,uCAAuC;IACvC,SAAgB,OAAO,EAAE,iBAAiB,CAAC;IAC3C,wCAAwC;IACxC,SAAgB,QAAQ,EAAE,kBAAkB,CAAC;IAC7C,2CAA2C;IAC3C,SAAgB,WAAW,EAAE,qBAAqB,CAAC;IACnD,4CAA4C;IAC5C,SAAgB,YAAY,EAAE,sBAAsB,CAAC;IACrD,sCAAsC;IACtC,SAAgB,MAAM,EAAE,gBAAgB,CAAC;IACzC,yCAAyC;IACzC,SAAgB,SAAS,EAAE,mBAAmB,CAAC;IAC/C;;;OAGG;IACU,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC;IAsB/C;;;OAGG;IACH,IAAW,QAAQ,IAAI,MAAM,CAE5B;CACF"}

package/dist/commonjs/aiProjectClient.js

@@ -18,9 +18,11 @@ const index_js_8 = require("./classic/evaluators/index.js");
 const index_js_9 = require("./classic/indexes/index.js");
 const index_js_10 = require("./classic/insights/index.js");
 const index_js_11 = require("./classic/memoryStores/index.js");
-const index_js_12 = require("./classic/redTeams/index.js");
-const index_js_13 = require("./classic/schedules/index.js");
+const index_js_12 = require("./classic/telemetry/index.js");
+const index_js_13 = require("./classic/redTeams/index.js");
+const index_js_14 = require("./classic/schedules/index.js");
 const overwriteOpenAIClient_js_1 = require("./overwriteOpenAIClient.js");
+const getCustomFetch_js_1 = require("./getCustomFetch.js");
 /**
  * The main client for the AIProjectClient service. It provides access to the various operations available in the service.
  * @class AIProjectClient
@@ -37,6 +39,7 @@ const overwriteOpenAIClient_js_1 = require("./overwriteOpenAIClient.js");
  * @property {ConnectionsOperations} connections - The operation groups for connections
  * @property {MemoryStoresOperations} memoryStores - The operation groups for memoryStores
  * @property {AgentsOperations} agents - The operation groups for agents
+ * @property {TelemetryOperations} telemetry - The operation groups for telemetry
  * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client
  * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client
  */
@@ -59,25 +62,26 @@ class AIProjectClient {
             userAgentOptions: { userAgentPrefix },
             credentials: {
                 ...options.credentials,
-                scopes: ["https://cognitiveservices.azure.com/.default"],
+                scopes: ["https://ai.azure.com/.default"],
             },
         });
         this._azureScopeClient = (0, index_js_1.createAIProject)(endpoint, credential, {
             ...options,
             userAgentOptions: { userAgentPrefix },
         });
-        this.schedules = (0, index_js_13._getSchedulesOperations)(this._cognitiveScopeClient);
+        this.schedules = (0, index_js_14._getSchedulesOperations)(this._cognitiveScopeClient);
         this.insights = (0, index_js_10._getInsightsOperations)(this._cognitiveScopeClient);
         this.evaluators = (0, index_js_8._getEvaluatorsOperations)(this._cognitiveScopeClient);
         this.evaluationTaxonomies = (0, index_js_7._getEvaluationTaxonomiesOperations)(this._cognitiveScopeClient);
         this.evaluationRules = (0, index_js_6._getEvaluationRulesOperations)(this._cognitiveScopeClient);
-        this.redTeams = (0, index_js_12._getRedTeamsOperations)(this._cognitiveScopeClient);
+        this.redTeams = (0, index_js_13._getRedTeamsOperations)(this._cognitiveScopeClient);
         this.deployments = (0, index_js_5._getDeploymentsOperations)(this._azureScopeClient);
         this.indexes = (0, index_js_9._getIndexesOperations)(this._azureScopeClient);
         this.datasets = (0, index_js_4._getDatasetsOperations)(this._azureScopeClient, this._options);
         this.connections = (0, index_js_3._getConnectionsOperations)(this._azureScopeClient);
         this.memoryStores = (0, index_js_11._getMemoryStoresOperations)(this._cognitiveScopeClient);
         this.agents = (0, index_js_2._getAgentsOperations)(this._azureScopeClient);
+        this.telemetry = (0, index_js_12._getTelemetryOperations)(this.connections);
     }
     /** The operation groups for schedules */
     schedules;
@@ -103,6 +107,8 @@ class AIProjectClient {
     memoryStores;
     /** The operation groups for agents */
     agents;
+    /** The operation groups for telemetry */
+    telemetry;
     /**
      * gets the OpenAI client
      * @returns the OpenAI client
@@ -110,11 +116,16 @@ class AIProjectClient {
     async getOpenAIClient() {
         const scope = "https://ai.azure.com/.default";
         const azureADTokenProvider = await (0, identity_1.getBearerTokenProvider)(this._credential, scope);
+        let customFetch;
+        if (this._options.additionalPolicies?.find((policy) => policy.policy.name === "recording policy")) {
+            customFetch = (0, getCustomFetch_js_1.getCustomFetch)(this._azureScopeClient.pipeline, this._options.httpClient);
+        }
         const openAIOptions = {
             apiKey: azureADTokenProvider,
             baseURL: `${this._endpoint}/openai`,
             defaultQuery: { "api-version": this._options?.apiVersion || "2025-11-15-preview" },
             dangerouslyAllowBrowser: true,
+            fetch: customFetch,
         };
         const openaiClient = new openai_1.default(openAIOptions);
         return (0, overwriteOpenAIClient_js_1.overwriteOpenAIClient)(openaiClient);

package/dist/commonjs/aiProjectClient.js.map

@@ -1 +1 @@
-{"version":3,"file":"aiProjectClient.js","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;;;;AAEjC,4DAA4B;AAC5B,8CAAyD;AACzD,6CAAkG;AAClG,wDAAmF;AACnF,6DAAkG;AAClG,0DAAyF;AACzF,6DAAkG;AAClG,iEAG4C;AAC5C,sEAGiD;AACjD,4DAA+F;AAC/F,yDAAsF;AACtF,2DAAyF;AACzF,+DAGyC;AACzC,2DAAyF;AACzF,4DAA4F;AAE5F,yEAAmE;AAInE;;;;;;;;;;;;;;;;;;GAkBG;AACH,MAAa,eAAe;IAClB,qBAAqB,CAAmB;IACxC,iBAAiB,CAAmB;IACpC,SAAS,CAAS;IAClB,WAAW,CAAkB;IAC7B,QAAQ,CAAgC;IAEhD,YACE,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;QAE3C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;QACrE,MAAM,eAAe,GAAG,iBAAiB;YACvC,CAAC,CAAC,GAAG,iBAAiB,kBAAkB;YACxC,CAAC,CAAC,iBAAiB,CAAC;QACtB,IAAI,CAAC,qBAAqB,GAAG,IAAA,0BAAe,EAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YACvE,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;YACrC,WAAW,EAAE;gBACX,GAAG,OAAO,CAAC,WAAW;gBACtB,MAAM,EAAE,CAAC,8CAA8C,CAAC;aACzD;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,IAAA,0BAAe,EAAC,QAAQ,EAAE,UAAU,EAAE;YAC7D,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,IAAA,mCAAuB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,IAAA,kCAAsB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,IAAI,CAAC,UAAU,GAAG,IAAA,mCAAwB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACvE,IAAI,CAAC,oBAAoB,GAAG,IAAA,6CAAkC,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3F,IAAI,CAAC,eAAe,GAAG,IAAA,wCAA6B,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACjF,IAAI,CAAC,QAAQ,GAAG,IAAA,kCAAsB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,IAAI,CAAC,WAAW,GAAG,IAAA,oCAAyB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,OAAO,GAAG,IAAA,gCAAqB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,GAAG,IAAA,iCAAsB,EAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,IAAI,CAAC,WAAW,GAAG,IAAA,oCAAyB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,GAAG,IAAA,sCAA0B,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3E,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAoB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7D,CAAC;IAED,yCAAyC;IACzB,SAAS,CAAsB;IAC/C,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,0CAA0C;IAC1B,UAAU,CAAuB;IACjD,oDAAoD;IACpC,oBAAoB,CAAiC;IACrE,+CAA+C;IAC/B,eAAe,CAA4B;IAC3D,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,uCAAuC;IACvB,OAAO,CAAoB;IAC3C,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,4CAA4C;IAC5B,YAAY,CAAyB;IACrD,sCAAsC;IACtB,MAAM,CAAmB;IACzC;;;OAGG;IACI,KAAK,CAAC,eAAe;QAC1B,MAAM,KAAK,GAAG,+BAA+B,CAAC;QAC9C,MAAM,oBAAoB,GAAG,MAAM,IAAA,iCAAsB,EAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAEnF,MAAM,aAAa,GAA4C;YAC7D,MAAM,EAAE,oBAAoB;YAC5B,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,SAAS;YACnC,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,IAAI,oBAAoB,EAAE;YAClF,uBAAuB,EAAE,IAAI;SAC9B,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,gBAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,OAAO,IAAA,gDAAqB,EAAC,YAAY,CAAC,CAAC;IAC7C,CAAC;IACD;;;OAGG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF;AA/FD,0CA+FC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport { createAIProject, AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { AgentsOperations, _getAgentsOperations } from \"./classic/agents/index.js\";\nimport { ConnectionsOperations, _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport { DatasetsOperations, _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport { DeploymentsOperations, _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport {\n  EvaluationRulesOperations,\n  _getEvaluationRulesOperations,\n} from \"./classic/evaluationRules/index.js\";\nimport {\n  EvaluationTaxonomiesOperations,\n  _getEvaluationTaxonomiesOperations,\n} from \"./classic/evaluationTaxonomies/index.js\";\nimport { EvaluatorsOperations, _getEvaluatorsOperations } from \"./classic/evaluators/index.js\";\nimport { IndexesOperations, _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport { InsightsOperations, _getInsightsOperations } from \"./classic/insights/index.js\";\nimport {\n  MemoryStoresOperations,\n  _getMemoryStoresOperations,\n} from \"./classic/memoryStores/index.js\";\nimport { RedTeamsOperations, _getRedTeamsOperations } from \"./classic/redTeams/index.js\";\nimport { SchedulesOperations, _getSchedulesOperations } from \"./classic/schedules/index.js\";\nimport { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\n\nexport { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {RedTeamsOperations} redTeams - The operation groups for redTeams\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {MemoryStoresOperations} memoryStores - The operation groups for memoryStores\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client\n */\nexport class AIProjectClient {\n  private _cognitiveScopeClient: AIProjectContext;\n  private _azureScopeClient: AIProjectContext;\n  private _endpoint: string;\n  private _credential: TokenCredential;\n  private _options: AIProjectClientOptionalParams;\n\n  constructor(\n    endpoint: string,\n    credential: TokenCredential,\n    options: AIProjectClientOptionalParams = {},\n  ) {\n    this._endpoint = endpoint;\n    this._credential = credential;\n    this._options = options;\n    const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n    const userAgentPrefix = prefixFromOptions\n      ? `${prefixFromOptions} azsdk-js-client`\n      : `azsdk-js-client`;\n    this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n      credentials: {\n        ...options.credentials,\n        scopes: [\"https://cognitiveservices.azure.com/.default\"],\n      },\n    });\n    this._azureScopeClient = createAIProject(endpoint, credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n    });\n\n    this.schedules = _getSchedulesOperations(this._cognitiveScopeClient);\n    this.insights = _getInsightsOperations(this._cognitiveScopeClient);\n    this.evaluators = _getEvaluatorsOperations(this._cognitiveScopeClient);\n    this.evaluationTaxonomies = _getEvaluationTaxonomiesOperations(this._cognitiveScopeClient);\n    this.evaluationRules = _getEvaluationRulesOperations(this._cognitiveScopeClient);\n    this.redTeams = _getRedTeamsOperations(this._cognitiveScopeClient);\n    this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n    this.indexes = _getIndexesOperations(this._azureScopeClient);\n    this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n    this.connections = _getConnectionsOperations(this._azureScopeClient);\n    this.memoryStores = _getMemoryStoresOperations(this._cognitiveScopeClient);\n    this.agents = _getAgentsOperations(this._azureScopeClient);\n  }\n\n  /** The operation groups for schedules */\n  public readonly schedules: SchedulesOperations;\n  /** The operation groups for insights */\n  public readonly insights: InsightsOperations;\n  /** The operation groups for evaluators */\n  public readonly evaluators: EvaluatorsOperations;\n  /** The operation groups for evaluationTaxonomies */\n  public readonly evaluationTaxonomies: EvaluationTaxonomiesOperations;\n  /** The operation groups for evaluationRules */\n  public readonly evaluationRules: EvaluationRulesOperations;\n  /** The operation groups for redTeams */\n  public readonly redTeams: RedTeamsOperations;\n  /** The operation groups for deployments */\n  public readonly deployments: DeploymentsOperations;\n  /** The operation groups for indexes */\n  public readonly indexes: IndexesOperations;\n  /** The operation groups for datasets */\n  public readonly datasets: DatasetsOperations;\n  /** The operation groups for connections */\n  public readonly connections: ConnectionsOperations;\n  /** The operation groups for memoryStores */\n  public readonly memoryStores: MemoryStoresOperations;\n  /** The operation groups for agents */\n  public readonly agents: AgentsOperations;\n  /**\n   * gets the OpenAI client\n   * @returns the OpenAI client\n   */\n  public async getOpenAIClient(): Promise<OpenAI> {\n    const scope = \"https://ai.azure.com/.default\";\n    const azureADTokenProvider = await getBearerTokenProvider(this._credential, scope);\n\n    const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n      apiKey: azureADTokenProvider,\n      baseURL: `${this._endpoint}/openai`,\n      defaultQuery: { \"api-version\": this._options?.apiVersion || \"2025-11-15-preview\" },\n      dangerouslyAllowBrowser: true,\n    };\n\n    const openaiClient = new OpenAI(openAIOptions);\n    return overwriteOpenAIClient(openaiClient);\n  }\n  /**\n   * gets the endpoint of the client\n   * @returns the endpoint of the client\n   */\n  public get endpoint(): string {\n    return this._endpoint;\n  }\n}\n"]}
+{"version":3,"file":"aiProjectClient.js","sourceRoot":"","sources":["../../src/aiProjectClient.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;AAClC,iCAAiC;;;;AAEjC,4DAA4B;AAC5B,8CAAyD;AACzD,6CAAkG;AAClG,wDAAmF;AACnF,6DAAkG;AAClG,0DAAyF;AACzF,6DAAkG;AAClG,iEAG4C;AAC5C,sEAGiD;AACjD,4DAA+F;AAC/F,yDAAsF;AACtF,2DAAyF;AACzF,+DAGyC;AACzC,4DAA4F;AAC5F,2DAAyF;AACzF,4DAA4F;AAE5F,yEAAmE;AACnE,2DAAqD;AAIrD;;;;;;;;;;;;;;;;;;;GAmBG;AACH,MAAa,eAAe;IAClB,qBAAqB,CAAmB;IACxC,iBAAiB,CAAmB;IACpC,SAAS,CAAS;IAClB,WAAW,CAAkB;IAC7B,QAAQ,CAAgC;IAEhD,YACE,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;QAE3C,IAAI,CAAC,SAAS,GAAG,QAAQ,CAAC;QAC1B,IAAI,CAAC,WAAW,GAAG,UAAU,CAAC;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;QACrE,MAAM,eAAe,GAAG,iBAAiB;YACvC,CAAC,CAAC,GAAG,iBAAiB,kBAAkB;YACxC,CAAC,CAAC,iBAAiB,CAAC;QACtB,IAAI,CAAC,qBAAqB,GAAG,IAAA,0BAAe,EAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,EAAE;YACvE,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;YACrC,WAAW,EAAE;gBACX,GAAG,OAAO,CAAC,WAAW;gBACtB,MAAM,EAAE,CAAC,+BAA+B,CAAC;aAC1C;SACF,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,GAAG,IAAA,0BAAe,EAAC,QAAQ,EAAE,UAAU,EAAE;YAC7D,GAAG,OAAO;YACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,IAAA,mCAAuB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,IAAA,kCAAsB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,IAAI,CAAC,UAAU,GAAG,IAAA,mCAAwB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACvE,IAAI,CAAC,oBAAoB,GAAG,IAAA,6CAAkC,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3F,IAAI,CAAC,eAAe,GAAG,IAAA,wCAA6B,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACjF,IAAI,CAAC,QAAQ,GAAG,IAAA,kCAAsB,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QACnE,IAAI,CAAC,WAAW,GAAG,IAAA,oCAAyB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,OAAO,GAAG,IAAA,gCAAqB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,GAAG,IAAA,iCAAsB,EAAC,IAAI,CAAC,iBAAiB,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC9E,IAAI,CAAC,WAAW,GAAG,IAAA,oCAAyB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACrE,IAAI,CAAC,YAAY,GAAG,IAAA,sCAA0B,EAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC3E,IAAI,CAAC,MAAM,GAAG,IAAA,+BAAoB,EAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC3D,IAAI,CAAC,SAAS,GAAG,IAAA,mCAAuB,EAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC7D,CAAC;IAED,yCAAyC;IACzB,SAAS,CAAsB;IAC/C,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,0CAA0C;IAC1B,UAAU,CAAuB;IACjD,oDAAoD;IACpC,oBAAoB,CAAiC;IACrE,+CAA+C;IAC/B,eAAe,CAA4B;IAC3D,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,uCAAuC;IACvB,OAAO,CAAoB;IAC3C,wCAAwC;IACxB,QAAQ,CAAqB;IAC7C,2CAA2C;IAC3B,WAAW,CAAwB;IACnD,4CAA4C;IAC5B,YAAY,CAAyB;IACrD,sCAAsC;IACtB,MAAM,CAAmB;IACzC,yCAAyC;IACzB,SAAS,CAAsB;IAC/C;;;OAGG;IACI,KAAK,CAAC,eAAe;QAC1B,MAAM,KAAK,GAAG,+BAA+B,CAAC;QAC9C,MAAM,oBAAoB,GAAG,MAAM,IAAA,iCAAsB,EAAC,IAAI,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QACnF,IAAI,WAA0E,CAAC;QAE/E,IACE,IAAI,CAAC,QAAQ,CAAC,kBAAkB,EAAE,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB,CAAC,EAC7F,CAAC;YACD,WAAW,GAAG,IAAA,kCAAc,EAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;QAC1F,CAAC;QAED,MAAM,aAAa,GAA4C;YAC7D,MAAM,EAAE,oBAAoB;YAC5B,OAAO,EAAE,GAAG,IAAI,CAAC,SAAS,SAAS;YACnC,YAAY,EAAE,EAAE,aAAa,EAAE,IAAI,CAAC,QAAQ,EAAE,UAAU,IAAI,oBAAoB,EAAE;YAClF,uBAAuB,EAAE,IAAI;YAC7B,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,MAAM,YAAY,GAAG,IAAI,gBAAM,CAAC,aAAa,CAAC,CAAC;QAC/C,OAAO,IAAA,gDAAqB,EAAC,YAAY,CAAC,CAAC;IAC7C,CAAC;IACD;;;OAGG;IACH,IAAW,QAAQ;QACjB,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;CACF;AA1GD,0CA0GC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n/* eslint-disable tsdoc/syntax */\n\nimport OpenAI from \"openai\";\nimport { getBearerTokenProvider } from \"@azure/identity\";\nimport { createAIProject, AIProjectContext, AIProjectClientOptionalParams } from \"./api/index.js\";\nimport { AgentsOperations, _getAgentsOperations } from \"./classic/agents/index.js\";\nimport { ConnectionsOperations, _getConnectionsOperations } from \"./classic/connections/index.js\";\nimport { DatasetsOperations, _getDatasetsOperations } from \"./classic/datasets/index.js\";\nimport { DeploymentsOperations, _getDeploymentsOperations } from \"./classic/deployments/index.js\";\nimport {\n  EvaluationRulesOperations,\n  _getEvaluationRulesOperations,\n} from \"./classic/evaluationRules/index.js\";\nimport {\n  EvaluationTaxonomiesOperations,\n  _getEvaluationTaxonomiesOperations,\n} from \"./classic/evaluationTaxonomies/index.js\";\nimport { EvaluatorsOperations, _getEvaluatorsOperations } from \"./classic/evaluators/index.js\";\nimport { IndexesOperations, _getIndexesOperations } from \"./classic/indexes/index.js\";\nimport { InsightsOperations, _getInsightsOperations } from \"./classic/insights/index.js\";\nimport {\n  MemoryStoresOperations,\n  _getMemoryStoresOperations,\n} from \"./classic/memoryStores/index.js\";\nimport { TelemetryOperations, _getTelemetryOperations } from \"./classic/telemetry/index.js\";\nimport { RedTeamsOperations, _getRedTeamsOperations } from \"./classic/redTeams/index.js\";\nimport { SchedulesOperations, _getSchedulesOperations } from \"./classic/schedules/index.js\";\nimport { TokenCredential } from \"@azure/core-auth\";\nimport { overwriteOpenAIClient } from \"./overwriteOpenAIClient.js\";\nimport { getCustomFetch } from \"./getCustomFetch.js\";\n\nexport { AIProjectClientOptionalParams } from \"./api/aiProjectContext.js\";\n\n/**\n * The main client for the AIProjectClient service. It provides access to the various operations available in the service.\n * @class AIProjectClient\n * @extends {AIProjectContext}\n * @description The main client for the AIProjectClient\n * @constructor\n * @param {string} endpoint - The endpoint to use\n * @param {TokenCredential} credential - The credential to use\n * @param {AIProjectClientOptionalParams} [options] - Optional parameters for the client.\n * @property {RedTeamsOperations} redTeams - The operation groups for redTeams\n * @property {DeploymentsOperations} deployments - The operation groups for deployments\n * @property {IndexesOperations} indexes - The operation groups for indexes\n * @property {DatasetsOperations} datasets - The operation groups for datasets\n * @property {ConnectionsOperations} connections - The operation groups for connections\n * @property {MemoryStoresOperations} memoryStores - The operation groups for memoryStores\n * @property {AgentsOperations} agents - The operation groups for agents\n * @property {TelemetryOperations} telemetry - The operation groups for telemetry\n * @property {getEndpointUrl} getEndpointUrl - gets the endpoint of the client\n * @property {getOpenAIClient} getOpenAIClient - gets the OpenAI client\n */\nexport class AIProjectClient {\n  private _cognitiveScopeClient: AIProjectContext;\n  private _azureScopeClient: AIProjectContext;\n  private _endpoint: string;\n  private _credential: TokenCredential;\n  private _options: AIProjectClientOptionalParams;\n\n  constructor(\n    endpoint: string,\n    credential: TokenCredential,\n    options: AIProjectClientOptionalParams = {},\n  ) {\n    this._endpoint = endpoint;\n    this._credential = credential;\n    this._options = options;\n    const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n    const userAgentPrefix = prefixFromOptions\n      ? `${prefixFromOptions} azsdk-js-client`\n      : `azsdk-js-client`;\n    this._cognitiveScopeClient = createAIProject(endpoint, this._credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n      credentials: {\n        ...options.credentials,\n        scopes: [\"https://ai.azure.com/.default\"],\n      },\n    });\n    this._azureScopeClient = createAIProject(endpoint, credential, {\n      ...options,\n      userAgentOptions: { userAgentPrefix },\n    });\n\n    this.schedules = _getSchedulesOperations(this._cognitiveScopeClient);\n    this.insights = _getInsightsOperations(this._cognitiveScopeClient);\n    this.evaluators = _getEvaluatorsOperations(this._cognitiveScopeClient);\n    this.evaluationTaxonomies = _getEvaluationTaxonomiesOperations(this._cognitiveScopeClient);\n    this.evaluationRules = _getEvaluationRulesOperations(this._cognitiveScopeClient);\n    this.redTeams = _getRedTeamsOperations(this._cognitiveScopeClient);\n    this.deployments = _getDeploymentsOperations(this._azureScopeClient);\n    this.indexes = _getIndexesOperations(this._azureScopeClient);\n    this.datasets = _getDatasetsOperations(this._azureScopeClient, this._options);\n    this.connections = _getConnectionsOperations(this._azureScopeClient);\n    this.memoryStores = _getMemoryStoresOperations(this._cognitiveScopeClient);\n    this.agents = _getAgentsOperations(this._azureScopeClient);\n    this.telemetry = _getTelemetryOperations(this.connections);\n  }\n\n  /** The operation groups for schedules */\n  public readonly schedules: SchedulesOperations;\n  /** The operation groups for insights */\n  public readonly insights: InsightsOperations;\n  /** The operation groups for evaluators */\n  public readonly evaluators: EvaluatorsOperations;\n  /** The operation groups for evaluationTaxonomies */\n  public readonly evaluationTaxonomies: EvaluationTaxonomiesOperations;\n  /** The operation groups for evaluationRules */\n  public readonly evaluationRules: EvaluationRulesOperations;\n  /** The operation groups for redTeams */\n  public readonly redTeams: RedTeamsOperations;\n  /** The operation groups for deployments */\n  public readonly deployments: DeploymentsOperations;\n  /** The operation groups for indexes */\n  public readonly indexes: IndexesOperations;\n  /** The operation groups for datasets */\n  public readonly datasets: DatasetsOperations;\n  /** The operation groups for connections */\n  public readonly connections: ConnectionsOperations;\n  /** The operation groups for memoryStores */\n  public readonly memoryStores: MemoryStoresOperations;\n  /** The operation groups for agents */\n  public readonly agents: AgentsOperations;\n  /** The operation groups for telemetry */\n  public readonly telemetry: TelemetryOperations;\n  /**\n   * gets the OpenAI client\n   * @returns the OpenAI client\n   */\n  public async getOpenAIClient(): Promise<OpenAI> {\n    const scope = \"https://ai.azure.com/.default\";\n    const azureADTokenProvider = await getBearerTokenProvider(this._credential, scope);\n    let customFetch: NonNullable<ConstructorParameters<typeof OpenAI>[0]>[\"fetch\"];\n\n    if (\n      this._options.additionalPolicies?.find((policy) => policy.policy.name === \"recording policy\")\n    ) {\n      customFetch = getCustomFetch(this._azureScopeClient.pipeline, this._options.httpClient);\n    }\n\n    const openAIOptions: ConstructorParameters<typeof OpenAI>[0] = {\n      apiKey: azureADTokenProvider,\n      baseURL: `${this._endpoint}/openai`,\n      defaultQuery: { \"api-version\": this._options?.apiVersion || \"2025-11-15-preview\" },\n      dangerouslyAllowBrowser: true,\n      fetch: customFetch,\n    };\n\n    const openaiClient = new OpenAI(openAIOptions);\n    return overwriteOpenAIClient(openaiClient);\n  }\n  /**\n   * gets the endpoint of the client\n   * @returns the endpoint of the client\n   */\n  public get endpoint(): string {\n    return this._endpoint;\n  }\n}\n"]}

package/dist/commonjs/api/aiProjectContext.js

@@ -9,7 +9,7 @@ const core_client_1 = require("@azure-rest/core-client");
 function createAIProject(endpoint, credential, options = {}) {
     const endpointUrl = options.endpoint ?? String(endpoint);
     const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;
-    const userAgentInfo = `azsdk-js-ai-projects/2.0.0-beta.1`;
+    const userAgentInfo = `azsdk-js-ai-projects/2.0.0-beta.3`;
     const userAgentPrefix = prefixFromOptions
         ? `${prefixFromOptions} azsdk-js-api ${userAgentInfo}`
         : `azsdk-js-api ${userAgentInfo}`;

package/dist/commonjs/api/aiProjectContext.js.map

@@ -1 +1 @@
-{"version":3,"file":"aiProjectContext.js","sourceRoot":"","sources":["../../../src/api/aiProjectContext.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAoBlC,0CAsCC;AAxDD,4CAAsC;AACtC,mDAAuD;AACvD,yDAA2E;AAgB3E,SAAgB,eAAe,CAC7B,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;IAE3C,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC;IACzD,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;IACrE,MAAM,aAAa,GAAG,mCAAmC,CAAC;IAC1D,MAAM,eAAe,GAAG,iBAAiB;QACvC,CAAC,CAAC,GAAG,iBAAiB,iBAAiB,aAAa,EAAE;QACtD,CAAC,CAAC,gBAAgB,aAAa,EAAE,CAAC;IACpC,MAAM,EAAE,UAAU,EAAE,CAAC,EAAE,GAAG,cAAc,EAAE,GAAG;QAC3C,GAAG,OAAO;QACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;QACrC,cAAc,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,cAAc,EAAE,MAAM,IAAI,kBAAM,CAAC,IAAI,EAAE;QACzE,WAAW,EAAE;YACX,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,+BAA+B,CAAC;SACzE;KACF,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,uBAAS,EAAC,WAAW,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IACzE,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAClE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,4BAAgB,CAAC,mBAAmB,CAAC;IAC9E,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC/B,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACzB,qDAAqD;YACrD,yEAAyE;YACzE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,GAAG,GAClB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GACzD,eAAe,UAAU,EAAE,CAAC;YAC9B,CAAC;YAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;KACF,CAAC,CAAC;IACH,OAAO,EAAE,GAAG,aAAa,EAAE,UAAU,EAAsB,CAAC;AAC9D,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { logger } from \"../logger.js\";\nimport { KnownApiVersions } from \"../models/models.js\";\nimport { Client, ClientOptions, getClient } from \"@azure-rest/core-client\";\nimport { TokenCredential } from \"@azure/core-auth\";\n\nexport interface AIProjectContext extends Client {\n  /** The API version to use for this operation. */\n  /** Known values of {@link KnownApiVersions} that the service accepts. */\n  apiVersion: KnownApiVersions;\n}\n\n/** Optional parameters for the client. */\nexport interface AIProjectClientOptionalParams extends ClientOptions {\n  /** The API version to use for this operation. */\n  /** Known values of {@link KnownApiVersions} that the service accepts. */\n  apiVersion?: KnownApiVersions;\n}\n\nexport function createAIProject(\n  endpoint: string,\n  credential: TokenCredential,\n  options: AIProjectClientOptionalParams = {},\n): AIProjectContext {\n  const endpointUrl = options.endpoint ?? String(endpoint);\n  const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n  const userAgentInfo = `azsdk-js-ai-projects/2.0.0-beta.1`;\n  const userAgentPrefix = prefixFromOptions\n    ? `${prefixFromOptions} azsdk-js-api ${userAgentInfo}`\n    : `azsdk-js-api ${userAgentInfo}`;\n  const { apiVersion: _, ...updatedOptions } = {\n    ...options,\n    userAgentOptions: { userAgentPrefix },\n    loggingOptions: { logger: options.loggingOptions?.logger ?? logger.info },\n    credentials: {\n      scopes: options.credentials?.scopes ?? [\"https://ai.azure.com/.default\"],\n    },\n  };\n  const clientContext = getClient(endpointUrl, credential, updatedOptions);\n  clientContext.pipeline.removePolicy({ name: \"ApiVersionPolicy\" });\n  const apiVersion = options.apiVersion ?? KnownApiVersions.v2025_11_15_preview;\n  clientContext.pipeline.addPolicy({\n    name: \"ClientApiVersionPolicy\",\n    sendRequest: (req, next) => {\n      // Use the apiVersion defined in request url directly\n      // Append one if there is no apiVersion and we have one at client options\n      const url = new URL(req.url);\n      if (!url.searchParams.get(\"api-version\")) {\n        req.url = `${req.url}${\n          Array.from(url.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n        }api-version=${apiVersion}`;\n      }\n\n      return next(req);\n    },\n  });\n  return { ...clientContext, apiVersion } as AIProjectContext;\n}\n"]}
+{"version":3,"file":"aiProjectContext.js","sourceRoot":"","sources":["../../../src/api/aiProjectContext.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AAoBlC,0CAsCC;AAxDD,4CAAsC;AACtC,mDAAuD;AACvD,yDAA2E;AAgB3E,SAAgB,eAAe,CAC7B,QAAgB,EAChB,UAA2B,EAC3B,UAAyC,EAAE;IAE3C,MAAM,WAAW,GAAG,OAAO,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC;IACzD,MAAM,iBAAiB,GAAG,OAAO,EAAE,gBAAgB,EAAE,eAAe,CAAC;IACrE,MAAM,aAAa,GAAG,mCAAmC,CAAC;IAC1D,MAAM,eAAe,GAAG,iBAAiB;QACvC,CAAC,CAAC,GAAG,iBAAiB,iBAAiB,aAAa,EAAE;QACtD,CAAC,CAAC,gBAAgB,aAAa,EAAE,CAAC;IACpC,MAAM,EAAE,UAAU,EAAE,CAAC,EAAE,GAAG,cAAc,EAAE,GAAG;QAC3C,GAAG,OAAO;QACV,gBAAgB,EAAE,EAAE,eAAe,EAAE;QACrC,cAAc,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,cAAc,EAAE,MAAM,IAAI,kBAAM,CAAC,IAAI,EAAE;QACzE,WAAW,EAAE;YACX,MAAM,EAAE,OAAO,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,+BAA+B,CAAC;SACzE;KACF,CAAC;IACF,MAAM,aAAa,GAAG,IAAA,uBAAS,EAAC,WAAW,EAAE,UAAU,EAAE,cAAc,CAAC,CAAC;IACzE,aAAa,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAClE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,4BAAgB,CAAC,mBAAmB,CAAC;IAC9E,aAAa,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC/B,IAAI,EAAE,wBAAwB;QAC9B,WAAW,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACzB,qDAAqD;YACrD,yEAAyE;YACzE,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAC7B,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;gBACzC,GAAG,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,GAAG,GAClB,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GACzD,eAAe,UAAU,EAAE,CAAC;YAC9B,CAAC;YAED,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC;QACnB,CAAC;KACF,CAAC,CAAC;IACH,OAAO,EAAE,GAAG,aAAa,EAAE,UAAU,EAAsB,CAAC;AAC9D,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { logger } from \"../logger.js\";\nimport { KnownApiVersions } from \"../models/models.js\";\nimport { Client, ClientOptions, getClient } from \"@azure-rest/core-client\";\nimport { TokenCredential } from \"@azure/core-auth\";\n\nexport interface AIProjectContext extends Client {\n  /** The API version to use for this operation. */\n  /** Known values of {@link KnownApiVersions} that the service accepts. */\n  apiVersion: KnownApiVersions;\n}\n\n/** Optional parameters for the client. */\nexport interface AIProjectClientOptionalParams extends ClientOptions {\n  /** The API version to use for this operation. */\n  /** Known values of {@link KnownApiVersions} that the service accepts. */\n  apiVersion?: KnownApiVersions;\n}\n\nexport function createAIProject(\n  endpoint: string,\n  credential: TokenCredential,\n  options: AIProjectClientOptionalParams = {},\n): AIProjectContext {\n  const endpointUrl = options.endpoint ?? String(endpoint);\n  const prefixFromOptions = options?.userAgentOptions?.userAgentPrefix;\n  const userAgentInfo = `azsdk-js-ai-projects/2.0.0-beta.3`;\n  const userAgentPrefix = prefixFromOptions\n    ? `${prefixFromOptions} azsdk-js-api ${userAgentInfo}`\n    : `azsdk-js-api ${userAgentInfo}`;\n  const { apiVersion: _, ...updatedOptions } = {\n    ...options,\n    userAgentOptions: { userAgentPrefix },\n    loggingOptions: { logger: options.loggingOptions?.logger ?? logger.info },\n    credentials: {\n      scopes: options.credentials?.scopes ?? [\"https://ai.azure.com/.default\"],\n    },\n  };\n  const clientContext = getClient(endpointUrl, credential, updatedOptions);\n  clientContext.pipeline.removePolicy({ name: \"ApiVersionPolicy\" });\n  const apiVersion = options.apiVersion ?? KnownApiVersions.v2025_11_15_preview;\n  clientContext.pipeline.addPolicy({\n    name: \"ClientApiVersionPolicy\",\n    sendRequest: (req, next) => {\n      // Use the apiVersion defined in request url directly\n      // Append one if there is no apiVersion and we have one at client options\n      const url = new URL(req.url);\n      if (!url.searchParams.get(\"api-version\")) {\n        req.url = `${req.url}${\n          Array.from(url.searchParams.keys()).length > 0 ? \"&\" : \"?\"\n        }api-version=${apiVersion}`;\n      }\n\n      return next(req);\n    },\n  });\n  return { ...clientContext, apiVersion } as AIProjectContext;\n}\n"]}

package/dist/commonjs/api/memoryStores/index.d.ts

@@ -1,3 +1,4 @@
 export { deleteScope, getUpdateResult, updateMemories, searchMemories, deleteMemoryStore, listMemoryStores, getMemoryStore, updateMemoryStore, createMemoryStore, } from "./operations.js";
 export { MemoryStoresDeleteScopeOptionalParams, MemoryStoresGetUpdateResultOptionalParams, MemoryStoresUpdateMemoriesOptionalParams, MemoryStoresSearchMemoriesOptionalParams, MemoryStoresDeleteMemoryStoreOptionalParams, MemoryStoresListMemoryStoresOptionalParams, MemoryStoresGetMemoryStoreOptionalParams, MemoryStoresUpdateMemoryStoreOptionalParams, MemoryStoresCreateMemoryStoreOptionalParams, } from "./options.js";
+export { MemoryStoreUpdateMemoriesPoller, MemoryStoreUpdateOperationState, } from "./memoryStoreUpdateMemoriesPoller.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/commonjs/api/memoryStores/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/memoryStores/index.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,WAAW,EACX,eAAe,EACf,cAAc,EACd,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,cAAc,EACd,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,qCAAqC,EACrC,yCAAyC,EACzC,wCAAwC,EACxC,wCAAwC,EACxC,2CAA2C,EAC3C,0CAA0C,EAC1C,wCAAwC,EACxC,2CAA2C,EAC3C,2CAA2C,GAC5C,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/api/memoryStores/index.ts"],"names":[],"mappings":"AAGA,OAAO,EACL,WAAW,EACX,eAAe,EACf,cAAc,EACd,cAAc,EACd,iBAAiB,EACjB,gBAAgB,EAChB,cAAc,EACd,iBAAiB,EACjB,iBAAiB,GAClB,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,qCAAqC,EACrC,yCAAyC,EACzC,wCAAwC,EACxC,wCAAwC,EACxC,2CAA2C,EAC3C,0CAA0C,EAC1C,wCAAwC,EACxC,2CAA2C,EAC3C,2CAA2C,GAC5C,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,+BAA+B,EAC/B,+BAA+B,GAChC,MAAM,sCAAsC,CAAC"}

package/dist/commonjs/api/memoryStores/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/memoryStores/index.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,iDAUyB;AATvB,4GAAA,WAAW,OAAA;AACX,gHAAA,eAAe,OAAA;AACf,+GAAA,cAAc,OAAA;AACd,+GAAA,cAAc,OAAA;AACd,kHAAA,iBAAiB,OAAA;AACjB,iHAAA,gBAAgB,OAAA;AAChB,+GAAA,cAAc,OAAA;AACd,kHAAA,iBAAiB,OAAA;AACjB,kHAAA,iBAAiB,OAAA","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport {\n  deleteScope,\n  getUpdateResult,\n  updateMemories,\n  searchMemories,\n  deleteMemoryStore,\n  listMemoryStores,\n  getMemoryStore,\n  updateMemoryStore,\n  createMemoryStore,\n} from \"./operations.js\";\nexport {\n  MemoryStoresDeleteScopeOptionalParams,\n  MemoryStoresGetUpdateResultOptionalParams,\n  MemoryStoresUpdateMemoriesOptionalParams,\n  MemoryStoresSearchMemoriesOptionalParams,\n  MemoryStoresDeleteMemoryStoreOptionalParams,\n  MemoryStoresListMemoryStoresOptionalParams,\n  MemoryStoresGetMemoryStoreOptionalParams,\n  MemoryStoresUpdateMemoryStoreOptionalParams,\n  MemoryStoresCreateMemoryStoreOptionalParams,\n} from \"./options.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/api/memoryStores/index.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;;AAElC,iDAUyB;AATvB,4GAAA,WAAW,OAAA;AACX,gHAAA,eAAe,OAAA;AACf,+GAAA,cAAc,OAAA;AACd,+GAAA,cAAc,OAAA;AACd,kHAAA,iBAAiB,OAAA;AACjB,iHAAA,gBAAgB,OAAA;AAChB,+GAAA,cAAc,OAAA;AACd,kHAAA,iBAAiB,OAAA;AACjB,kHAAA,iBAAiB,OAAA","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nexport {\n  deleteScope,\n  getUpdateResult,\n  updateMemories,\n  searchMemories,\n  deleteMemoryStore,\n  listMemoryStores,\n  getMemoryStore,\n  updateMemoryStore,\n  createMemoryStore,\n} from \"./operations.js\";\nexport {\n  MemoryStoresDeleteScopeOptionalParams,\n  MemoryStoresGetUpdateResultOptionalParams,\n  MemoryStoresUpdateMemoriesOptionalParams,\n  MemoryStoresSearchMemoriesOptionalParams,\n  MemoryStoresDeleteMemoryStoreOptionalParams,\n  MemoryStoresListMemoryStoresOptionalParams,\n  MemoryStoresGetMemoryStoreOptionalParams,\n  MemoryStoresUpdateMemoryStoreOptionalParams,\n  MemoryStoresCreateMemoryStoreOptionalParams,\n} from \"./options.js\";\nexport {\n  MemoryStoreUpdateMemoriesPoller,\n  MemoryStoreUpdateOperationState,\n} from \"./memoryStoreUpdateMemoriesPoller.js\";\n"]}

package/dist/commonjs/api/memoryStores/memoryStoreUpdateMemoriesPoller.d.ts

@@ -0,0 +1,43 @@
+import { AIProjectContext as Client } from "../index.js";
+import { MemoryStoreUpdateResult, MemoryStoreUpdateStatus } from "../../models/models.js";
+import { PollerLike, OperationState } from "@azure/core-lro";
+import { AbortSignalLike } from "@azure/abort-controller";
+import { PathUncheckedResponse } from "@azure-rest/core-client";
+/** State of the Memory Store update operation. */
+export type MemoryStoreUpdateOperationState = OperationState<MemoryStoreUpdateResult> & {
+    /** The ID of the memory store update operation. */
+    updateId: string;
+    /** The status of the memory store update operation. */
+    updateStatus: MemoryStoreUpdateStatus;
+    /** The ID of the memory store update operation that superseded this one, if any. */
+    supersededBy?: string;
+};
+/** Custom LROPoller for Memory Store update operations. */
+export type MemoryStoreUpdateMemoriesPoller = PollerLike<MemoryStoreUpdateOperationState, MemoryStoreUpdateResult> & {
+    /** The ID of the memory store update operation. */
+    readonly updateId: string;
+    /** The status of the memory store update operation. */
+    readonly updateStatus: MemoryStoreUpdateStatus;
+    /** The ID of the memory store update operation that superseded this one, if any. */
+    readonly supersededBy?: string;
+};
+/** Options for creating a MemoryStoreUpdateMemoriesPoller. */
+export interface CreateMemoryStoreUpdateMemoriesPollerOptions {
+    /** Interval in milliseconds between update polls */
+    updateIntervalInMs?: number;
+    /** An abort signal to cancel the poller */
+    abortSignal?: AbortSignalLike;
+    /** Used to restore poller from a serialized state */
+    restoreFrom?: string;
+}
+/**
+ * Creates a poller to track the progress of updating memories in a memory store.
+ *
+ * @param client - The AIProjectClient instance.
+ * @param expectedStatuses - The expected HTTP statuses for the polling operation.
+ * @param getInitialResponse - A function that returns a promise of the initial response to start the polling operation.
+ * @param options - Optional parameters for creating the poller.
+ * @returns A MemoryStoreUpdateMemoriesPoller instance.
+ */
+export declare function createMemoryStoreUpdateMemoriesPoller(client: Client, expectedStatuses: string[], getInitialResponse?: () => PromiseLike<PathUncheckedResponse>, options?: CreateMemoryStoreUpdateMemoriesPollerOptions): MemoryStoreUpdateMemoriesPoller;
+//# sourceMappingURL=memoryStoreUpdateMemoriesPoller.d.ts.map

package/dist/commonjs/api/memoryStores/memoryStoreUpdateMemoriesPoller.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memoryStoreUpdateMemoriesPoller.d.ts","sourceRoot":"","sources":["../../../../src/api/memoryStores/memoryStoreUpdateMemoriesPoller.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EACL,uBAAuB,EACvB,uBAAuB,EAIxB,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,UAAU,EACV,cAAc,EAIf,MAAM,iBAAiB,CAAC;AACzB,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAmB,MAAM,yBAAyB,CAAC;AAEjF,kDAAkD;AAClD,MAAM,MAAM,+BAA+B,GAAG,cAAc,CAAC,uBAAuB,CAAC,GAAG;IACtF,mDAAmD;IACnD,QAAQ,EAAE,MAAM,CAAC;IACjB,uDAAuD;IACvD,YAAY,EAAE,uBAAuB,CAAC;IACtC,oFAAoF;IACpF,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,CAAC;AAEF,2DAA2D;AAC3D,MAAM,MAAM,+BAA+B,GAAG,UAAU,CACtD,+BAA+B,EAC/B,uBAAuB,CACxB,GAAG;IACF,mDAAmD;IACnD,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,uDAAuD;IACvD,QAAQ,CAAC,YAAY,EAAE,uBAAuB,CAAC;IAC/C,oFAAoF;IACpF,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC;CAChC,CAAC;AAEF,8DAA8D;AAC9D,MAAM,WAAW,4CAA4C;IAC3D,oDAAoD;IACpD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,2CAA2C;IAC3C,WAAW,CAAC,EAAE,eAAe,CAAC;IAC9B,qDAAqD;IACrD,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAgID;;;;;;;;GAQG;AACH,wBAAgB,qCAAqC,CACnD,MAAM,EAAE,MAAM,EACd,gBAAgB,EAAE,MAAM,EAAE,EAC1B,kBAAkB,CAAC,EAAE,MAAM,WAAW,CAAC,qBAAqB,CAAC,EAC7D,OAAO,GAAE,4CAAiD,GACzD,+BAA+B,CAoDjC"}

package/dist/commonjs/api/memoryStores/memoryStoreUpdateMemoriesPoller.js

@@ -0,0 +1,163 @@
+"use strict";
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createMemoryStoreUpdateMemoriesPoller = createMemoryStoreUpdateMemoriesPoller;
+const models_js_1 = require("../../models/models.js");
+const core_lro_1 = require("@azure/core-lro");
+const core_client_1 = require("@azure-rest/core-client");
+const nonFailureTerminalUpdateStatuses = ["completed", "superseded"];
+function createDefaultUsage() {
+    return {
+        embedding_tokens: 0,
+        input_tokens: 0,
+        input_tokens_details: { cached_tokens: 0 },
+        output_tokens: 0,
+        output_tokens_details: { reasoning_tokens: 0 },
+        total_tokens: 0,
+    };
+}
+function toOperationResponse(response, expectedStatuses) {
+    if (!expectedStatuses.includes(response.status)) {
+        throw (0, core_client_1.createRestError)(response);
+    }
+    return {
+        flatResponse: response,
+        rawResponse: {
+            ...response,
+            statusCode: Number.parseInt(response.status),
+            body: response.body,
+        },
+    };
+}
+function deserializeUpdateResponse(response) {
+    try {
+        return (0, models_js_1.memoryStoreUpdateResponseDeserializer)(response.body);
+    }
+    catch {
+        return undefined;
+    }
+}
+function applyUpdateState(state, response) {
+    const parsed = deserializeUpdateResponse(response);
+    state.updateId = parsed?.update_id ?? "";
+    state.updateStatus = parsed?.status ?? "queued";
+    if (!parsed) {
+        return;
+    }
+    if (parsed.status === "failed") {
+        state.status = "failed";
+        if (!state.error) {
+            state.error = parsed.error
+                ? new Error(parsed.error.message ?? "Memory update failed")
+                : new Error("Memory update failed");
+        }
+        return;
+    }
+    if (nonFailureTerminalUpdateStatuses.includes(parsed.status)) {
+        if (parsed.status === "superseded" && !state.error) {
+            state.supersededBy = parsed.superseded_by;
+        }
+        state.status = "succeeded";
+        state.result = parsed.result ??
+            state.result ?? {
+            memory_operations: [],
+            usage: createDefaultUsage(),
+        };
+        return;
+    }
+    state.status = "running";
+}
+function buildRunningOperation(client, expectedStatuses, getInitialResponse, options) {
+    const pollAbortController = new AbortController();
+    return {
+        sendInitialRequest: async () => {
+            if (!getInitialResponse) {
+                throw new Error("getInitialResponse is required when starting a new poller");
+            }
+            const initialResponse = await getInitialResponse();
+            return toOperationResponse(initialResponse, expectedStatuses);
+        },
+        sendPollRequest: async (path, pollOptions) => {
+            // The poll request will both listen to the user provided abort signal and the poller's own abort signal
+            function abortListener() {
+                pollAbortController.abort();
+            }
+            const abortSignal = pollAbortController.signal;
+            if (options?.abortSignal?.aborted) {
+                pollAbortController.abort();
+            }
+            else if (pollOptions?.abortSignal?.aborted) {
+                pollAbortController.abort();
+            }
+            else if (!abortSignal.aborted) {
+                options?.abortSignal?.addEventListener("abort", abortListener, {
+                    once: true,
+                });
+                pollOptions?.abortSignal?.addEventListener("abort", abortListener, {
+                    once: true,
+                });
+            }
+            let response;
+            try {
+                response = (await client.pathUnchecked(path).get({ abortSignal }));
+            }
+            finally {
+                options?.abortSignal?.removeEventListener("abort", abortListener);
+                pollOptions?.abortSignal?.removeEventListener("abort", abortListener);
+            }
+            return toOperationResponse(response, expectedStatuses);
+        },
+    };
+}
+/**
+ * Creates a poller to track the progress of updating memories in a memory store.
+ *
+ * @param client - The AIProjectClient instance.
+ * @param expectedStatuses - The expected HTTP statuses for the polling operation.
+ * @param getInitialResponse - A function that returns a promise of the initial response to start the polling operation.
+ * @param options - Optional parameters for creating the poller.
+ * @returns A MemoryStoreUpdateMemoriesPoller instance.
+ */
+function createMemoryStoreUpdateMemoriesPoller(client, expectedStatuses, getInitialResponse, options = {}) {
+    if (!getInitialResponse && !options?.restoreFrom) {
+        throw new Error("getInitialResponse is required when starting a new poller");
+    }
+    let initialResponse;
+    const poller = (0, core_lro_1.createHttpPoller)(buildRunningOperation(client, expectedStatuses, getInitialResponse
+        ? async () => {
+            initialResponse = await getInitialResponse();
+            return initialResponse;
+        }
+        : undefined, options), {
+        intervalInMs: options?.updateIntervalInMs,
+        restoreFrom: options?.restoreFrom,
+        updateState: (state, lastResponse) => {
+            const flatResponse = lastResponse.flatResponse;
+            applyUpdateState(state, flatResponse);
+        },
+        processResult: async (operationResponse, state) => {
+            applyUpdateState(state, operationResponse);
+            return state.result;
+        },
+    });
+    if (initialResponse && poller.operationState) {
+        applyUpdateState(poller.operationState, initialResponse);
+    }
+    Object.defineProperties(poller, {
+        updateId: {
+            enumerable: true,
+            get: () => poller.operationState?.updateId,
+        },
+        updateStatus: {
+            enumerable: true,
+            get: () => poller.operationState?.updateStatus,
+        },
+        supersededBy: {
+            enumerable: true,
+            get: () => poller.operationState?.supersededBy,
+        },
+    });
+    return poller;
+}
+//# sourceMappingURL=memoryStoreUpdateMemoriesPoller.js.map

package/dist/commonjs/api/memoryStores/memoryStoreUpdateMemoriesPoller.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memoryStoreUpdateMemoriesPoller.js","sourceRoot":"","sources":["../../../../src/api/memoryStores/memoryStoreUpdateMemoriesPoller.ts"],"names":[],"mappings":";AAAA,uCAAuC;AACvC,kCAAkC;;AA4LlC,sFAyDC;AAlPD,sDAMgC;AAChC,8CAMyB;AAEzB,yDAAiF;AAmCjF,MAAM,gCAAgC,GAA8B,CAAC,WAAW,EAAE,YAAY,CAAC,CAAC;AAEhG,SAAS,kBAAkB;IACzB,OAAO;QACL,gBAAgB,EAAE,CAAC;QACnB,YAAY,EAAE,CAAC;QACf,oBAAoB,EAAE,EAAE,aAAa,EAAE,CAAC,EAAE;QAC1C,aAAa,EAAE,CAAC;QAChB,qBAAqB,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE;QAC9C,YAAY,EAAE,CAAC;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAC1B,QAA+B,EAC/B,gBAA0B;IAE1B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QAChD,MAAM,IAAA,6BAAe,EAAC,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,SAAS,yBAAyB,CAChC,QAA+B;IAE/B,IAAI,CAAC;QACH,OAAO,IAAA,iDAAqC,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IAC9D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CACvB,KAAsC,EACtC,QAA+B;IAE/B,MAAM,MAAM,GAAG,yBAAyB,CAAC,QAAQ,CAAC,CAAC;IACnD,KAAK,CAAC,QAAQ,GAAG,MAAM,EAAE,SAAS,IAAI,EAAE,CAAC;IACzC,KAAK,CAAC,YAAY,GAAG,MAAM,EAAE,MAAM,IAAI,QAAQ,CAAC;IAEhD,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;IACT,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC/B,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;QACxB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACjB,KAAK,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK;gBACxB,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,IAAI,sBAAsB,CAAC;gBAC3D,CAAC,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACxC,CAAC;QACD,OAAO;IACT,CAAC;IAED,IAAI,gCAAgC,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7D,IAAI,MAAM,CAAC,MAAM,KAAK,YAAY,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YACnD,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;QAC5C,CAAC;QACD,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC;QAC3B,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM;YAC1B,KAAK,CAAC,MAAM,IAAI;YACd,iBAAiB,EAAE,EAAE;YACrB,KAAK,EAAE,kBAAkB,EAAE;SAC5B,CAAC;QACJ,OAAO;IACT,CAAC;IAED,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;AAC3B,CAAC;AAED,SAAS,qBAAqB,CAC5B,MAAc,EACd,gBAA0B,EAC1B,kBAA6D,EAC7D,OAAsD;IAEtD,MAAM,mBAAmB,GAAG,IAAI,eAAe,EAAE,CAAC;IAClD,OAAO;QACL,kBAAkB,EAAE,KAAK,IAAI,EAAE;YAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBACxB,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;YAC/E,CAAC;YACD,MAAM,eAAe,GAAG,MAAM,kBAAkB,EAAE,CAAC;YACnD,OAAO,mBAAmB,CAAC,eAAe,EAAE,gBAAgB,CAAC,CAAC;QAChE,CAAC;QACD,eAAe,EAAE,KAAK,EAAE,IAAY,EAAE,WAA+C,EAAE,EAAE;YACvF,wGAAwG;YACxG,SAAS,aAAa;gBACpB,mBAAmB,CAAC,KAAK,EAAE,CAAC;YAC9B,CAAC;YACD,MAAM,WAAW,GAAG,mBAAmB,CAAC,MAAM,CAAC;YAC/C,IAAI,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;gBAClC,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,EAAE,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,aAAa,EAAE;oBAC7D,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;YAED,IAAI,QAA+B,CAAC;YACpC,IAAI,CAAC;gBACH,QAAQ,GAAG,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,WAAW,EAAE,CAAC,CAA0B,CAAC;YAC9F,CAAC;oBAAS,CAAC;gBACT,OAAO,EAAE,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;gBAClE,WAAW,EAAE,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACxE,CAAC;YAED,OAAO,mBAAmB,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QACzD,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;;;;;;;GAQG;AACH,SAAgB,qCAAqC,CACnD,MAAc,EACd,gBAA0B,EAC1B,kBAA6D,EAC7D,UAAwD,EAAE;IAE1D,IAAI,CAAC,kBAAkB,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,CAAC;QACjD,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;IAC/E,CAAC;IACD,IAAI,eAAkD,CAAC;IAEvD,MAAM,MAAM,GAAG,IAAA,2BAAgB,EAC7B,qBAAqB,CACnB,MAAM,EACN,gBAAgB,EAChB,kBAAkB;QAChB,CAAC,CAAC,KAAK,IAAI,EAAE;YACT,eAAe,GAAG,MAAM,kBAAkB,EAAE,CAAC;YAC7C,OAAO,eAAe,CAAC;QACzB,CAAC;QACH,CAAC,CAAC,SAAS,EACb,OAAO,CACR,EACD;QACE,YAAY,EAAE,OAAO,EAAE,kBAAkB;QACzC,WAAW,EAAE,OAAO,EAAE,WAAW;QACjC,WAAW,EAAE,CAAC,KAAK,EAAE,YAAY,EAAE,EAAE;YACnC,MAAM,YAAY,GAAG,YAAY,CAAC,YAAqC,CAAC;YACxE,gBAAgB,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;QACxC,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,EAAE;YAChD,gBAAgB,CAAC,KAAK,EAAE,iBAA0C,CAAC,CAAC;YACpE,OAAO,KAAK,CAAC,MAAiC,CAAC;QACjD,CAAC;KACF,CACiC,CAAC;IAErC,IAAI,eAAe,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;QAC7C,gBAAgB,CAAC,MAAM,CAAC,cAAc,EAAE,eAAe,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,CAAC,gBAAgB,CAAC,MAAM,EAAE;QAC9B,QAAQ,EAAE;YACR,UAAU,EAAE,IAAI;YAChB,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,QAAQ;SAC3C;QACD,YAAY,EAAE;YACZ,UAAU,EAAE,IAAI;YAChB,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,YAAY;SAC/C;QACD,YAAY,EAAE;YACZ,UAAU,EAAE,IAAI;YAChB,GAAG,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,cAAc,EAAE,YAAY;SAC/C;KACF,CAAC,CAAC;IAEH,OAAO,MAAM,CAAC;AAChB,CAAC","sourcesContent":["// Copyright (c) Microsoft Corporation.\n// Licensed under the MIT License.\n\nimport { AIProjectContext as Client } from \"../index.js\";\nimport {\n  MemoryStoreUpdateResult,\n  MemoryStoreUpdateStatus,\n  MemoryStoreUpdateResponse,\n  memoryStoreUpdateResponseDeserializer,\n  MemoryStoreOperationUsage,\n} from \"../../models/models.js\";\nimport {\n  PollerLike,\n  OperationState,\n  createHttpPoller,\n  RunningOperation,\n  OperationResponse,\n} from \"@azure/core-lro\";\nimport { AbortSignalLike } from \"@azure/abort-controller\";\nimport { PathUncheckedResponse, createRestError } from \"@azure-rest/core-client\";\n\n/** State of the Memory Store update operation. */\nexport type MemoryStoreUpdateOperationState = OperationState<MemoryStoreUpdateResult> & {\n  /** The ID of the memory store update operation. */\n  updateId: string;\n  /** The status of the memory store update operation. */\n  updateStatus: MemoryStoreUpdateStatus;\n  /** The ID of the memory store update operation that superseded this one, if any. */\n  supersededBy?: string;\n};\n\n/** Custom LROPoller for Memory Store update operations. */\nexport type MemoryStoreUpdateMemoriesPoller = PollerLike<\n  MemoryStoreUpdateOperationState,\n  MemoryStoreUpdateResult\n> & {\n  /** The ID of the memory store update operation. */\n  readonly updateId: string;\n  /** The status of the memory store update operation. */\n  readonly updateStatus: MemoryStoreUpdateStatus;\n  /** The ID of the memory store update operation that superseded this one, if any. */\n  readonly supersededBy?: string;\n};\n\n/** Options for creating a MemoryStoreUpdateMemoriesPoller. */\nexport interface CreateMemoryStoreUpdateMemoriesPollerOptions {\n  /** Interval in milliseconds between update polls */\n  updateIntervalInMs?: number;\n  /** An abort signal to cancel the poller */\n  abortSignal?: AbortSignalLike;\n  /** Used to restore poller from a serialized state */\n  restoreFrom?: string;\n}\n\nconst nonFailureTerminalUpdateStatuses: MemoryStoreUpdateStatus[] = [\"completed\", \"superseded\"];\n\nfunction createDefaultUsage(): MemoryStoreOperationUsage {\n  return {\n    embedding_tokens: 0,\n    input_tokens: 0,\n    input_tokens_details: { cached_tokens: 0 },\n    output_tokens: 0,\n    output_tokens_details: { reasoning_tokens: 0 },\n    total_tokens: 0,\n  };\n}\n\nfunction toOperationResponse(\n  response: PathUncheckedResponse,\n  expectedStatuses: string[],\n): OperationResponse<PathUncheckedResponse> {\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\nfunction deserializeUpdateResponse(\n  response: PathUncheckedResponse,\n): MemoryStoreUpdateResponse | undefined {\n  try {\n    return memoryStoreUpdateResponseDeserializer(response.body);\n  } catch {\n    return undefined;\n  }\n}\n\nfunction applyUpdateState(\n  state: MemoryStoreUpdateOperationState,\n  response: PathUncheckedResponse,\n): void {\n  const parsed = deserializeUpdateResponse(response);\n  state.updateId = parsed?.update_id ?? \"\";\n  state.updateStatus = parsed?.status ?? \"queued\";\n\n  if (!parsed) {\n    return;\n  }\n\n  if (parsed.status === \"failed\") {\n    state.status = \"failed\";\n    if (!state.error) {\n      state.error = parsed.error\n        ? new Error(parsed.error.message ?? \"Memory update failed\")\n        : new Error(\"Memory update failed\");\n    }\n    return;\n  }\n\n  if (nonFailureTerminalUpdateStatuses.includes(parsed.status)) {\n    if (parsed.status === \"superseded\" && !state.error) {\n      state.supersededBy = parsed.superseded_by;\n    }\n    state.status = \"succeeded\";\n    state.result = parsed.result ??\n      state.result ?? {\n        memory_operations: [],\n        usage: createDefaultUsage(),\n      };\n    return;\n  }\n\n  state.status = \"running\";\n}\n\nfunction buildRunningOperation(\n  client: Client,\n  expectedStatuses: string[],\n  getInitialResponse?: () => PromiseLike<PathUncheckedResponse>,\n  options?: CreateMemoryStoreUpdateMemoriesPollerOptions,\n): RunningOperation<PathUncheckedResponse> {\n  const pollAbortController = new AbortController();\n  return {\n    sendInitialRequest: async () => {\n      if (!getInitialResponse) {\n        throw new Error(\"getInitialResponse is required when starting a new poller\");\n      }\n      const initialResponse = await getInitialResponse();\n      return toOperationResponse(initialResponse, expectedStatuses);\n    },\n    sendPollRequest: async (path: string, pollOptions?: { abortSignal?: AbortSignalLike }) => {\n      // The poll request will 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\n      let response: PathUncheckedResponse;\n      try {\n        response = (await client.pathUnchecked(path).get({ abortSignal })) as PathUncheckedResponse;\n      } finally {\n        options?.abortSignal?.removeEventListener(\"abort\", abortListener);\n        pollOptions?.abortSignal?.removeEventListener(\"abort\", abortListener);\n      }\n\n      return toOperationResponse(response, expectedStatuses);\n    },\n  };\n}\n\n/**\n * Creates a poller to track the progress of updating memories in a memory store.\n *\n * @param client - The AIProjectClient instance.\n * @param expectedStatuses - The expected HTTP statuses for the polling operation.\n * @param getInitialResponse - A function that returns a promise of the initial response to start the polling operation.\n * @param options - Optional parameters for creating the poller.\n * @returns A MemoryStoreUpdateMemoriesPoller instance.\n */\nexport function createMemoryStoreUpdateMemoriesPoller(\n  client: Client,\n  expectedStatuses: string[],\n  getInitialResponse?: () => PromiseLike<PathUncheckedResponse>,\n  options: CreateMemoryStoreUpdateMemoriesPollerOptions = {},\n): MemoryStoreUpdateMemoriesPoller {\n  if (!getInitialResponse && !options?.restoreFrom) {\n    throw new Error(\"getInitialResponse is required when starting a new poller\");\n  }\n  let initialResponse: PathUncheckedResponse | undefined;\n\n  const poller = createHttpPoller<MemoryStoreUpdateResult, MemoryStoreUpdateOperationState>(\n    buildRunningOperation(\n      client,\n      expectedStatuses,\n      getInitialResponse\n        ? async () => {\n            initialResponse = await getInitialResponse();\n            return initialResponse;\n          }\n        : undefined,\n      options,\n    ),\n    {\n      intervalInMs: options?.updateIntervalInMs,\n      restoreFrom: options?.restoreFrom,\n      updateState: (state, lastResponse) => {\n        const flatResponse = lastResponse.flatResponse as PathUncheckedResponse;\n        applyUpdateState(state, flatResponse);\n      },\n      processResult: async (operationResponse, state) => {\n        applyUpdateState(state, operationResponse as PathUncheckedResponse);\n        return state.result as MemoryStoreUpdateResult;\n      },\n    },\n  ) as MemoryStoreUpdateMemoriesPoller;\n\n  if (initialResponse && poller.operationState) {\n    applyUpdateState(poller.operationState, initialResponse);\n  }\n\n  Object.defineProperties(poller, {\n    updateId: {\n      enumerable: true,\n      get: () => poller.operationState?.updateId,\n    },\n    updateStatus: {\n      enumerable: true,\n      get: () => poller.operationState?.updateStatus,\n    },\n    supersededBy: {\n      enumerable: true,\n      get: () => poller.operationState?.supersededBy,\n    },\n  });\n\n  return poller;\n}\n"]}

package/dist/commonjs/api/memoryStores/operations.d.ts

@@ -1,9 +1,9 @@
 import { AIProjectContext as Client } from "../index.js";
-import { MemoryStoreDefinitionUnion, MemoryStore, _AgentsPagedResultMemoryStore, DeleteMemoryStoreResponse, MemoryStoreSearchResponse, MemoryStoreUpdateResponse, MemoryStoreUpdateResult, MemoryStoreDeleteScopeResponse } from "../../models/models.js";
+import { MemoryStoreDefinitionUnion, MemoryStore, _AgentsPagedResultMemoryStore, DeleteMemoryStoreResponse, MemoryStoreSearchResponse, MemoryStoreUpdateResponse, MemoryStoreDeleteScopeResponse } from "../../models/models.js";
 import { PagedAsyncIterableIterator } from "../../static-helpers/pagingHelpers.js";
+import { MemoryStoreUpdateMemoriesPoller } from "./memoryStoreUpdateMemoriesPoller.js";
 import { MemoryStoresDeleteScopeOptionalParams, MemoryStoresGetUpdateResultOptionalParams, MemoryStoresUpdateMemoriesOptionalParams, MemoryStoresSearchMemoriesOptionalParams, MemoryStoresDeleteMemoryStoreOptionalParams, MemoryStoresListMemoryStoresOptionalParams, MemoryStoresGetMemoryStoreOptionalParams, MemoryStoresUpdateMemoryStoreOptionalParams, MemoryStoresCreateMemoryStoreOptionalParams } from "./options.js";
 import { StreamableMethod, PathUncheckedResponse } from "@azure-rest/core-client";
-import { PollerLike, OperationState } from "@azure/core-lro";
 export declare function _deleteScopeSend(context: Client, name: string, scope: string, options?: MemoryStoresDeleteScopeOptionalParams): StreamableMethod;
 export declare function _deleteScopeDeserialize(result: PathUncheckedResponse): Promise<MemoryStoreDeleteScopeResponse>;
 /** Delete all memories associated with a specific scope from a memory store. */
@@ -13,9 +13,8 @@ export declare function _getUpdateResultDeserialize(result: PathUncheckedRespons
 /** Get memory store update result. */
 export declare function getUpdateResult(context: Client, name: string, updateId: string, options?: MemoryStoresGetUpdateResultOptionalParams): Promise<MemoryStoreUpdateResponse>;
 export declare function _updateMemoriesSend(context: Client, name: string, scope: string, options?: MemoryStoresUpdateMemoriesOptionalParams): StreamableMethod;
-export declare function _updateMemoriesDeserialize(result: PathUncheckedResponse): Promise<MemoryStoreUpdateResult>;
 /** Update memory store with conversation memories. */
-export declare function updateMemories(context: Client, name: string, scope: string, options?: MemoryStoresUpdateMemoriesOptionalParams): PollerLike<OperationState<MemoryStoreUpdateResult>, MemoryStoreUpdateResult>;
+export declare function updateMemories(context: Client, name: string, scope: string, options?: MemoryStoresUpdateMemoriesOptionalParams): MemoryStoreUpdateMemoriesPoller;
 export declare function _searchMemoriesSend(context: Client, name: string, scope: string, options?: MemoryStoresSearchMemoriesOptionalParams): StreamableMethod;
 export declare function _searchMemoriesDeserialize(result: PathUncheckedResponse): Promise<MemoryStoreSearchResponse>;
 /** Search for relevant memories from a memory store based on conversation context. */

package/dist/commonjs/api/memoryStores/operations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/memoryStores/operations.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAIL,0BAA0B,EAC1B,WAAW,EAEX,6BAA6B,EAE7B,yBAAyB,EAGzB,yBAAyB,EAEzB,yBAAyB,EAEzB,uBAAuB,EAEvB,8BAA8B,EAE/B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAG/C,OAAO,EACL,qCAAqC,EACrC,yCAAyC,EACzC,wCAAwC,EACxC,wCAAwC,EACxC,2CAA2C,EAC3C,0CAA0C,EAC1C,wCAAwC,EACxC,2CAA2C,EAC3C,2CAA2C,EAC5C,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AAE7D,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,qCAA8D,GACtE,gBAAgB,CAoBlB;AAED,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,8BAA8B,CAAC,CASzC;AAED,gFAAgF;AAChF,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,qCAA8D,GACtE,OAAO,CAAC,8BAA8B,CAAC,CAGzC;AAED,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,yCAAkE,GAC1E,gBAAgB,CAmBlB;AAED,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,sCAAsC;AACtC,wBAAsB,eAAe,CACnC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,yCAAkE,GAC1E,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CA0BlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,uBAAuB,CAAC,CAgBlC;AAED,sDAAsD;AACtD,wBAAgB,cAAc,CAC5B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,UAAU,CAAC,cAAc,CAAC,uBAAuB,CAAC,EAAE,uBAAuB,CAAC,CAM9E;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CA4BlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,sFAAsF;AACtF,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAkBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0CAAmE,GAC3E,gBAAgB,CAqBlB;AAED,wBAAsB,4BAA4B,CAChD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,6BAA6B,CAAC,CASxC;AAED,8BAA8B;AAC9B,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0CAAmE,GAC3E,0BAA0B,CAAC,WAAW,CAAC,CAQzC;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CAkBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,+BAA+B;AAC/B,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,wCAAiE,GACzE,OAAO,CAAC,WAAW,CAAC,CAGtB;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAoBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,WAAW,CAAC,CAGtB;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAwBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,WAAW,CAAC,CAGtB"}
+{"version":3,"file":"operations.d.ts","sourceRoot":"","sources":["../../../../src/api/memoryStores/operations.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,IAAI,MAAM,EAAE,MAAM,aAAa,CAAC;AACzD,OAAO,EAIL,0BAA0B,EAC1B,WAAW,EAEX,6BAA6B,EAE7B,yBAAyB,EAGzB,yBAAyB,EAEzB,yBAAyB,EAEzB,8BAA8B,EAE/B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACL,0BAA0B,EAE3B,MAAM,uCAAuC,CAAC;AAE/C,OAAO,EAEL,+BAA+B,EAChC,MAAM,sCAAsC,CAAC;AAC9C,OAAO,EACL,qCAAqC,EACrC,yCAAyC,EACzC,wCAAwC,EACxC,wCAAwC,EACxC,2CAA2C,EAC3C,0CAA0C,EAC1C,wCAAwC,EACxC,2CAA2C,EAC3C,2CAA2C,EAC5C,MAAM,cAAc,CAAC;AACtB,OAAO,EACL,gBAAgB,EAChB,qBAAqB,EAGtB,MAAM,yBAAyB,CAAC;AAEjC,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,qCAA8D,GACtE,gBAAgB,CAoBlB;AAED,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,8BAA8B,CAAC,CASzC;AAED,gFAAgF;AAChF,wBAAsB,WAAW,CAC/B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,qCAA8D,GACtE,OAAO,CAAC,8BAA8B,CAAC,CAGzC;AAED,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,yCAAkE,GAC1E,gBAAgB,CAmBlB;AAED,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,sCAAsC;AACtC,wBAAsB,eAAe,CACnC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE,yCAAkE,GAC1E,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CA0BlB;AAED,sDAAsD;AACtD,wBAAgB,cAAc,CAC5B,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,+BAA+B,CAUjC;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CA4BlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,sFAAsF;AACtF,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,KAAK,EAAE,MAAM,EACb,OAAO,GAAE,wCAAiE,GACzE,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAkBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,yBAAyB,CAAC,CASpC;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,yBAAyB,CAAC,CAGpC;AAED,wBAAgB,qBAAqB,CACnC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0CAAmE,GAC3E,gBAAgB,CAqBlB;AAED,wBAAsB,4BAA4B,CAChD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,6BAA6B,CAAC,CASxC;AAED,8BAA8B;AAC9B,wBAAgB,gBAAgB,CAC9B,OAAO,EAAE,MAAM,EACf,OAAO,GAAE,0CAAmE,GAC3E,0BAA0B,CAAC,WAAW,CAAC,CAQzC;AAED,wBAAgB,mBAAmB,CACjC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,wCAAiE,GACzE,gBAAgB,CAkBlB;AAED,wBAAsB,0BAA0B,CAC9C,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,+BAA+B;AAC/B,wBAAsB,cAAc,CAClC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,wCAAiE,GACzE,OAAO,CAAC,WAAW,CAAC,CAGtB;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAoBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,WAAW,CAAC,CAGtB;AAED,wBAAgB,sBAAsB,CACpC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,2CAAoE,GAC5E,gBAAgB,CAwBlB;AAED,wBAAsB,6BAA6B,CACjD,MAAM,EAAE,qBAAqB,GAC5B,OAAO,CAAC,WAAW,CAAC,CAStB;AAED,6BAA6B;AAC7B,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,MAAM,EACf,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,0BAA0B,EACtC,OAAO,GAAE,2CAAoE,GAC5E,OAAO,CAAC,WAAW,CAAC,CAGtB"}