Add tests for tools and improve samples for interprise file search. (#42446)

* Create new tests for tools.

* Fix

Author: nick863

sdk/ai/azure-ai-agents/README.md

@@ -294,6 +294,10 @@ else:
 ds = VectorStoreDataSource(asset_identifier=asset_uri, asset_type=VectorStoreDataSourceAssetType.URI_ASSET)
 vector_store = agents_client.vector_stores.create_and_poll(data_sources=[ds], name="sample_vector_store")
 print(f"Created vector store, vector store ID: {vector_store.id}")
+vector_store_files = {}
+for fle in agents_client.vector_store_files.list(vector_store.id):
+    uploaded_file = agents_client.files.get(fle.id)
+    vector_store_files[fle.id] = uploaded_file.filename
 
 # Create a file search tool
 file_search_tool = FileSearchTool(vector_store_ids=[vector_store.id])

sdk/ai/azure-ai-agents/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "python",
   "TagPrefix": "python/ai/azure-ai-agents",
-  "Tag": "python/ai/azure-ai-agents_e418bdd70b"
+  "Tag": "python/ai/azure-ai-agents_ad0998d8d9"
 }

sdk/ai/azure-ai-agents/dev_requirements.txt

@@ -3,6 +3,7 @@
 ../../identity/azure-identity
 ../../core/azure-core-tracing-opentelemetry
 aiohttp
+jsonref
 opentelemetry-sdk
 opentelemetry-exporter-otlp-proto-grpc
 azure-ai-ml

sdk/ai/azure-ai-agents/samples/agents_async/sample_agents_vector_store_enterprise_file_search_async.py

@@ -28,7 +28,6 @@
     VectorStoreDataSource,
     VectorStoreDataSourceAssetType,
     ListSortOrder,
-    MessageTextContent,
 )
 from azure.identity.aio import DefaultAzureCredential
 
@@ -47,6 +46,10 @@ async def main():
         ds = VectorStoreDataSource(asset_identifier=asset_uri, asset_type=VectorStoreDataSourceAssetType.URI_ASSET)
         vector_store = await agents_client.vector_stores.create_and_poll(data_sources=[ds], name="sample_vector_store")
         print(f"Created vector store, vector store ID: {vector_store.id}")
+        vector_store_files = {}
+        async for fle in agents_client.vector_store_files.list(vector_store.id):
+            uploaded_file = await agents_client.files.get(fle.id)
+            vector_store_files[fle.id] = uploaded_file.filename
 
         # Create a file search tool
         file_search_tool = FileSearchTool(vector_store_ids=[vector_store.id])
@@ -80,9 +83,16 @@ async def main():
 
         messages = agents_client.messages.list(thread_id=thread.id, order=ListSortOrder.ASCENDING)
         async for msg in messages:
-            last_part = msg.content[-1]
-            if isinstance(last_part, MessageTextContent):
-                print(f"{msg.role}: {last_part.text.value}")
+            if msg.text_messages:
+                last_text = msg.text_messages[-1].text.value
+                for annotation in msg.text_messages[-1].text.annotations:
+                    
+                    citation = (
+                        vector_store_files.get(
+                            annotation.file_citation.file_id, annotation.file_citation.file_id)
+                    )
+                    last_text = last_text.replace(annotation.text, f" [{citation}]")
+                print(f"{msg.role}: {last_text}")
 
 
 if __name__ == "__main__":

sdk/ai/azure-ai-agents/samples/agents_tools/sample_agents_enterprise_file_search.py

@@ -57,6 +57,10 @@
     ds = VectorStoreDataSource(asset_identifier=asset_uri, asset_type=VectorStoreDataSourceAssetType.URI_ASSET)
     vector_store = agents_client.vector_stores.create_and_poll(data_sources=[ds], name="sample_vector_store")
     print(f"Created vector store, vector store ID: {vector_store.id}")
+    vector_store_files = {}
+    for fle in agents_client.vector_store_files.list(vector_store.id):
+        uploaded_file = agents_client.files.get(fle.id)
+        vector_store_files[fle.id] = uploaded_file.filename
 
     # Create a file search tool
     file_search_tool = FileSearchTool(vector_store_ids=[vector_store.id])
@@ -95,5 +99,12 @@
     messages = agents_client.messages.list(thread_id=thread.id, order=ListSortOrder.ASCENDING)
     for msg in messages:
         if msg.text_messages:
-            last_text = msg.text_messages[-1]
-            print(f"{msg.role}: {last_text.text.value}")
+            last_text = msg.text_messages[-1].text.value
+            for annotation in msg.text_messages[-1].text.annotations:
+                
+                citation = (
+                    vector_store_files.get(
+                        annotation.file_citation.file_id, annotation.file_citation.file_id)
+                )
+                last_text = last_text.replace(annotation.text, f" [{citation}]")
+            print(f"{msg.role}: {last_text}")

sdk/ai/azure-ai-agents/tests/assets/weather_openapi.json

@@ -0,0 +1,62 @@
+{
+  "openapi": "3.1.0",
+  "info": {
+    "title": "get weather data",
+    "description": "Retrieves current weather data for a location based on wttr.in.",
+    "version": "v1.0.0"
+  },
+  "servers": [
+    {
+      "url": "https://wttr.in"
+    }
+  ],
+  "auth": [],
+  "paths": {
+    "/{location}": {
+      "get": {
+        "description": "Get weather information for a specific location",
+        "operationId": "GetCurrentWeather",
+        "parameters": [
+          {
+            "name": "location",
+            "in": "path",
+            "description": "City or location to retrieve the weather for",
+            "required": true,
+            "schema": {
+              "type": "string"
+            }
+          },
+          {
+           "name": "format",
+           "in": "query",
+           "description": "Always use j1 value for this parameter",
+           "required": true,
+           "schema": {
+             "type": "string",
+             "default": "j1"
+          }
+         }
+        ],
+        "responses": {
+          "200": {
+            "description": "Successful response",
+            "content": {
+              "text/plain": {
+                "schema": {
+                  "type": "string"
+                }
+              }
+            }
+          },
+          "404": {
+            "description": "Location not found"
+          }
+        },
+        "deprecated": false
+      }
+    }
+  },
+  "components": {
+    "schemes": {}
+  }
+}

sdk/ai/azure-ai-agents/tests/conftest.py

@@ -147,6 +147,12 @@ def azure_workspace_triad_sanitizer():
         value="/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/00000/providers/Microsoft.MachineLearningServices/workspaces/00000/connections/someindex",
     )
 
+    # Sanitize the plain bing grounding.
+    add_body_key_sanitizer(
+        json_path="tools[*].bing_grounding.search_configurations[*].connection_id",
+        value="/subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/00000/providers/Microsoft.CognitiveServices/accounts/00000/projects/00000/connections/00000",
+    )
+
     # Sanitize deep research tool bing connection ID
     add_body_key_sanitizer(
         json_path="tools[*].deep_research.bing_grounding_connections[*].connection_id",