Add Agent Memory Tool tests. Run Agent Memory tool samples as tests. (#44445)

dargilco · web-flow · commit 908d293da9b6 · 2025-12-16T13:29:10.000-08:00
diff --git a/sdk/ai/azure-ai-projects/.env.template b/sdk/ai/azure-ai-projects/.env.template
@@ -25,8 +25,8 @@ AZURE_AI_PROJECTS_AZURE_RESOURCE_GROUP=
 AZURE_AI_PROJECTS_AZURE_AOAI_ACCOUNT=
 
 # Used in Memory Store samples
-AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME=
-AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME=
+MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME=
+MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME=
 
 # Used in Agent tools samples
 IMAGE_GENERATION_MODEL_DEPLOYMENT_NAME=
@@ -64,6 +64,10 @@ AZURE_AI_PROJECTS_TESTS_CONTAINER_INGRESS_SUBDOMAIN_SUFFIX=
 # Used in Agent Image Generation tests
 AZURE_AI_PROJECTS_TESTS_IMAGE_GENERATION_MODEL_DEPLOYMENT_NAME=
 
+# Used in Memory Store tests
+AZURE_AI_PROJECTS_TESTS_MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME=
+AZURE_AI_PROJECTS_TESTS_MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME=
+
 # Connection IDs and settings used in agent tool tests
 AZURE_AI_PROJECTS_TESTS_BING_PROJECT_CONNECTION_ID=
 AZURE_AI_PROJECTS_TESTS_AI_SEARCH_PROJECT_CONNECTION_ID=
diff --git a/sdk/ai/azure-ai-projects/README.md b/sdk/ai/azure-ai-projects/README.md
@@ -260,7 +260,9 @@ Generate images based on text prompts with customizable resolution, quality, and
 
 ```python
 tool = ImageGenTool(  # type: ignore[call-overload]
-    model=image_generation_model, quality="low", size="1024x1024"  # Model such as "gpt-image-1-mini"  # type: ignore
+    model=image_generation_model,  # Model such as "gpt-image-1-mini"  # type: ignore
+    quality="low",
+    size="1024x1024",
 )
 ```
 
diff --git a/sdk/ai/azure-ai-projects/assets.json b/sdk/ai/azure-ai-projects/assets.json
@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ai/azure-ai-projects",
-  "Tag": "python/ai/azure-ai-projects_5b7a7fa09e"
+  "Tag": "python/ai/azure-ai-projects_ae8b55e649"
 }
diff --git a/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_image_generation.py b/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_image_generation.py
@@ -62,7 +62,9 @@
 
     # [START tool_declaration]
     tool = ImageGenTool(  # type: ignore[call-overload]
-        model=image_generation_model, quality="low", size="1024x1024"  # Model such as "gpt-image-1-mini"  # type: ignore
+        model=image_generation_model,  # Model such as "gpt-image-1-mini"  # type: ignore
+        quality="low",
+        size="1024x1024",
     )
     # [END tool_declaration]
 
diff --git a/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_memory_search.py b/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_memory_search.py
@@ -28,9 +28,9 @@
        page of your Microsoft Foundry portal.
     2) AZURE_AI_MODEL_DEPLOYMENT_NAME - The deployment name of the Agent's AI model,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model for memory,
+    3) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model for memory,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
-    4) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model for memory,
+    4) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model for memory,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -52,7 +52,7 @@
 endpoint = os.environ["AZURE_AI_PROJECT_ENDPOINT"]
 
 with (
-    DefaultAzureCredential(exclude_interactive_browser_credential=False) as credential,
+    DefaultAzureCredential() as credential,
     AIProjectClient(endpoint=endpoint, credential=credential) as project_client,
     project_client.get_openai_client() as openai_client,
 ):
@@ -67,8 +67,8 @@
 
     # Create a memory store
     definition = MemoryStoreDefaultDefinition(
-        chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-        embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+        chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+        embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
         options=MemoryStoreDefaultOptions(
             user_profile_enabled=True, chat_summary_enabled=True
         ),  # Note: This line will not be needed once the service is fixed to use correct defaults
diff --git a/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_memory_search_async.py b/sdk/ai/azure-ai-projects/samples/agents/tools/sample_agent_memory_search_async.py
@@ -28,9 +28,9 @@
        page of your Microsoft Foundry portal.
     2) AZURE_AI_MODEL_DEPLOYMENT_NAME - The deployment name of the Agent's AI model,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model for memory,
+    3) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model for memory,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
-    4) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model for memory,
+    4) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model for memory,
        as found under the "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -70,8 +70,8 @@ async def main() -> None:
 
         # Create a memory store
         definition = MemoryStoreDefaultDefinition(
-            chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-            embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+            chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+            embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
             options=MemoryStoreDefaultOptions(
                 user_profile_enabled=True, chat_summary_enabled=True
             ),  # Note: This line will not be needed once the service is fixed to use correct defaults
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_advanced.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_advanced.py
@@ -26,9 +26,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -69,8 +69,8 @@
         chat_summary_enabled=True,
     )
     definition = MemoryStoreDefaultDefinition(
-        chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-        embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+        chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+        embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
         options=options,
     )
     memory_store = project_client.memory_stores.create(
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_advanced_async.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_advanced_async.py
@@ -26,9 +26,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -73,8 +73,8 @@ async def main() -> None:
             chat_summary_enabled=True,
         )
         definition = MemoryStoreDefaultDefinition(
-            chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-            embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+            chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+            embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
             options=options,
         )
         memory_store = await project_client.memory_stores.create(
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_basic.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_basic.py
@@ -24,9 +24,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -61,8 +61,8 @@
 
     # Create a memory store
     definition = MemoryStoreDefaultDefinition(
-        chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-        embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+        chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+        embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
         options=MemoryStoreDefaultOptions(
             user_profile_enabled=True, chat_summary_enabled=True
         ),  # Note: This line will not be needed once the service is fixed to use correct defaults
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_basic_async.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_basic_async.py
@@ -25,9 +25,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -66,8 +66,8 @@ async def main() -> None:
 
         # Create a memory store
         definition = MemoryStoreDefaultDefinition(
-            chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-            embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+            chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+            embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
             options=MemoryStoreDefaultOptions(
                 user_profile_enabled=True, chat_summary_enabled=True
             ),  # Note: This line will not be needed once the service is fixed to use correct defaults
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_crud.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_crud.py
@@ -22,9 +22,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -54,8 +54,8 @@
 
     # Create Memory Store
     definition = MemoryStoreDefaultDefinition(
-        chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-        embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+        chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+        embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
     )
     memory_store = project_client.memory_stores.create(
         name=memory_store_name, description="Example memory store for conversations", definition=definition
diff --git a/sdk/ai/azure-ai-projects/samples/memories/sample_memory_crud_async.py b/sdk/ai/azure-ai-projects/samples/memories/sample_memory_crud_async.py
@@ -22,9 +22,9 @@
     Set these environment variables with your own values:
     1) AZURE_AI_PROJECT_ENDPOINT - The Azure AI Project endpoint, as found in the Overview
        page of your Microsoft Foundry portal.
-    2) AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
+    2) MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME - The deployment name of the chat model, as found under the "Name" column in
        the "Models + endpoints" tab in your Microsoft Foundry project.
-    3) AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
+    3) MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME - The deployment name of the embedding model, as found under the
        "Name" column in the "Models + endpoints" tab in your Microsoft Foundry project.
 """
 
@@ -58,8 +58,8 @@ async def main() -> None:
 
         # Create Memory Store
         definition = MemoryStoreDefaultDefinition(
-            chat_model=os.environ["AZURE_AI_CHAT_MODEL_DEPLOYMENT_NAME"],
-            embedding_model=os.environ["AZURE_AI_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
+            chat_model=os.environ["MEMORY_STORE_CHAT_MODEL_DEPLOYMENT_NAME"],
+            embedding_model=os.environ["MEMORY_STORE_EMBEDDING_MODEL_DEPLOYMENT_NAME"],
         )
         memory_store = await project_client.memory_stores.create(
             name=memory_store_name, description="Example memory store for conversations", definition=definition
diff --git a/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_memory_search.py b/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_memory_search.py
diff --git a/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_memory_search_async.py b/sdk/ai/azure-ai-projects/tests/agents/tools/test_agent_memory_search_async.py
diff --git a/sdk/ai/azure-ai-projects/tests/samples/test_samples.py b/sdk/ai/azure-ai-projects/tests/samples/test_samples.py
diff --git a/sdk/ai/azure-ai-projects/tests/samples/test_samples_async.py b/sdk/ai/azure-ai-projects/tests/samples/test_samples_async.py
diff --git a/sdk/ai/azure-ai-projects/tests/test_base.py b/sdk/ai/azure-ai-projects/tests/test_base.py

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,9 @@ Generate images based on text prompts with customizable resolution, quality, and`
`260`	`260`
`261`	`261`	```python
`262`	`262`	`tool = ImageGenTool( # type: ignore[call-overload]`
`263`		`- model=image_generation_model, quality="low", size="1024x1024" # Model such as "gpt-image-1-mini" # type: ignore`
	`263`	`+ model=image_generation_model, # Model such as "gpt-image-1-mini" # type: ignore`
	`264`	`+ quality="low",`
	`265`	`+ size="1024x1024",`
`264`	`266`	`)`
`265`	`267`	```
`266`	`268`
Original file line number	Diff line number	Diff line change
`@@ -2,5 +2,5 @@`
`2`	`2`	`"AssetsRepo": "Azure/azure-sdk-assets",`
`3`	`3`	`"AssetsRepoPrefixPath": "python",`
`4`	`4`	`"TagPrefix": "python/ai/azure-ai-projects",`
`5`		`- "Tag": "python/ai/azure-ai-projects_5b7a7fa09e"`
	`5`	`+ "Tag": "python/ai/azure-ai-projects_ae8b55e649"`
`6`	`6`	`}`
Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,9 @@`
`62`	`62`
`63`	`63`	`# [START tool_declaration]`
`64`	`64`	`tool = ImageGenTool( # type: ignore[call-overload]`
`65`		`- model=image_generation_model, quality="low", size="1024x1024" # Model such as "gpt-image-1-mini" # type: ignore`
	`65`	`+ model=image_generation_model, # Model such as "gpt-image-1-mini" # type: ignore`
	`66`	`+ quality="low",`
	`67`	`+ size="1024x1024",`
`66`	`68`	`)`
`67`	`69`	`# [END tool_declaration]`
`68`	`70`