Fix custom Memory Stores LRO poller operation (#45662)

Author: dargilco

sdk/ai/azure-ai-projects/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Release History
 
+## 2.0.1 (2026-03-12)
+
+### Bugs Fixed
+
+* Fix custom Memory Stores LRO poller operation to add the missing
+  required `"Foundry-Features": "MemoryStores=V1Preview"` HTTP request header.
+
 ## 2.0.0 (2026-03-06)
 
 First stable release of the client library that uses the Generally Available (GA) version "v1" of the Foundry REST APIs.

sdk/ai/azure-ai-projects/azure/ai/projects/_version.py

@@ -6,4 +6,4 @@
 # Changes may cause incorrect behavior and will be lost if the code is regenerated.
 # --------------------------------------------------------------------------
 
-VERSION = "2.0.0"
+VERSION = "2.0.1"

sdk/ai/azure-ai-projects/azure/ai/projects/aio/operations/_patch_memories_async.py

@@ -22,6 +22,7 @@
     AsyncUpdateMemoriesLROPoller,
     AsyncUpdateMemoriesLROPollingMethod,
 )
+from ...models._enums import _FoundryFeaturesOptInKeys
 from ._operations import JSON, _Unset, ClsType, BetaMemoryStoresOperations as GenerateBetaMemoryStoresOperations
 from ...operations._patch_memories import _serialize_memory_input_items
 from ..._validation import api_version_validation
@@ -349,7 +350,10 @@ def get_long_running_output(pipeline_response):
 
         if polling is True:
             polling_method: AsyncUpdateMemoriesLROPollingMethod = AsyncUpdateMemoriesLROPollingMethod(
-                lro_delay, path_format_arguments=path_format_arguments, **kwargs
+                lro_delay,
+                path_format_arguments=path_format_arguments,
+                headers={"Foundry-Features": _FoundryFeaturesOptInKeys.MEMORY_STORES_V1_PREVIEW.value},
+                **kwargs,
             )
         elif polling is False:
             polling_method = cast(AsyncUpdateMemoriesLROPollingMethod, AsyncNoPolling())

sdk/ai/azure-ai-projects/azure/ai/projects/operations/_patch_memories.py

@@ -22,6 +22,7 @@
     UpdateMemoriesLROPoller,
     UpdateMemoriesLROPollingMethod,
 )
+from ..models._enums import _FoundryFeaturesOptInKeys
 from ._operations import JSON, _Unset, ClsType, BetaMemoryStoresOperations as GenerateBetaMemoryStoresOperations
 from .._validation import api_version_validation
 from .._utils.model_base import _deserialize, _serialize
@@ -384,7 +385,10 @@ def get_long_running_output(pipeline_response):
 
         if polling is True:
             polling_method: UpdateMemoriesLROPollingMethod = UpdateMemoriesLROPollingMethod(
-                lro_delay, path_format_arguments=path_format_arguments, **kwargs
+                lro_delay,
+                path_format_arguments=path_format_arguments,
+                headers={"Foundry-Features": _FoundryFeaturesOptInKeys.MEMORY_STORES_V1_PREVIEW.value},
+                **kwargs,
             )
         elif polling is False:
             polling_method = cast(UpdateMemoriesLROPollingMethod, NoPolling())