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

package/dist/react-native/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/react-native/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/react-native/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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@azure/ai-projects",
-  "version": "2.0.0-beta.1",
+  "version": "2.0.0-beta.3",
   "description": "Azure AI Projects client library.",
   "engines": {
     "node": ">=20.0.0"
@@ -61,6 +61,7 @@
     "@azure-rest/core-client": "^2.1.0",
     "@azure/abort-controller": "^2.1.2",
     "@azure/ai-agents": "1.1.0",
+    "@azure/arm-cognitiveservices": "7.6.0",
     "@azure/core-auth": "^1.6.0",
     "@azure/core-lro": "^3.0.0",
     "@azure/core-paging": "^1.5.0",
@@ -75,28 +76,30 @@
   },
   "devDependencies": {
     "@azure-rest/ai-inference": "^1.0.0-beta.6",
-    "@azure/identity": "4.11.1",
-    "@azure/monitor-opentelemetry": "^1.14.1",
+    "@azure/identity": "4.13.0",
+    "@azure/monitor-opentelemetry": "^1.14.2",
     "@azure/opentelemetry-instrumentation-azure-sdk": "1.0.0-beta.7",
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/instrumentation": "^0.57.0",
     "@opentelemetry/sdk-trace-node": "^2.0.0",
-    "@types/node": "^20.19.24",
+    "@types/node": "^20.19.25",
     "@vitest/browser-playwright": "^4.0.8",
     "@vitest/coverage-istanbul": "^4.0.8",
-    "cross-env": "^7.0.3",
+    "@modelcontextprotocol/sdk": "^1.25.1",
+    "zod": "^4.2.1",
+    "cross-env": "^10.1.0",
     "dotenv": "^16.6.1",
     "eslint": "^9.39.1",
     "mkdirp": "^3.0.1",
-    "playwright": "^1.50.1",
+    "playwright": "^1.56.1",
     "prettier": "^3.6.2",
     "rimraf": "^6.1.0",
     "tshy": "^3.0.3",
     "typescript": "~5.9.3",
     "vitest": "^4.0.8",
     "@azure-tools/test-credential": "^2.1.2",
-    "@azure-tools/test-utils-vitest": "^2.0.1",
     "@azure-tools/test-recorder": "^4.1.1",
+    "@azure-tools/test-utils-vitest": "^2.0.1",
     "@azure/dev-tool": "^1.0.0",
     "@azure/eslint-plugin-azure-sdk": "^3.0.0"
   },
@@ -127,9 +130,13 @@
       "azure"
     ],
     "extraFiles": {
-      "./data": [
-        "javascript/data",
-        "typescript/data"
+      "./samples-dev/files/data": [
+        "javascript/files/data",
+        "typescript/files/data"
+      ],
+      "./samples-dev/agents/assets": [
+        "javascript/agents/assets",
+        "typescript/agents/assets"
       ]
     },
     "apiRefLink": "https://learn.microsoft.com/javascript/api/@azure/ai-projects"