Jhakulin/list op update (#40871)

* update list operations

* update tests

* update README

* update

* Fix the instrumentation tests (#40877)

* Fix the instrumentation tests

* Better instrumentation handling

* pylint

---------

Co-authored-by: Nikolay Rovinskiy <[email protected]>

Author: Jarno Hakulinen

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

@@ -293,8 +293,6 @@ Here is an example:
 ```python
 conn_id = os.environ["AZURE_BING_CONNECTION_ID"]
 
-print(conn_id)
-
 # Initialize agent bing tool and add the connection id
 bing = BingGroundingTool(connection_id=conn_id)
 
@@ -351,7 +349,7 @@ get sensible result, the index needs to have "embedding", "token", "category" an
 ```python
 # Fetch and log all messages
 messages = agents_client.messages.list(thread_id=thread.id, order=ListSortOrder.ASCENDING)
-for message in messages.data:
+for message in messages:
     if message.role == MessageRole.AGENT and message.url_citation_annotations:
         placeholder_annotations = {
             annotation.text: f" [see {annotation.url_citation.title}] ({annotation.url_citation.url})"
@@ -1044,13 +1042,10 @@ To retrieve messages from agents, use the following example:
 
 ```python
 messages = agents_client.messages.list(thread_id=thread.id, order=ListSortOrder.ASCENDING)
-
-# The messages are following in the reverse order,
-# we will iterate them and output only text contents.
-for data_point in messages.data:
-    last_message_content = data_point.content[-1]
-    if isinstance(last_message_content, MessageTextContent):
-        print(f"{data_point.role}: {last_message_content.text.value}")
+for msg in messages:
+    if msg.text_messages:
+        last_text = msg.text_messages[-1]
+        print(f"{msg.role}: {last_text.text.value}")
 ```
 
 <!-- END SNIPPET -->
@@ -1069,20 +1064,22 @@ Here is an example retrieving file ids from messages and save to the local drive
 messages = agents_client.messages.list(thread_id=thread.id)
 print(f"Messages: {messages}")
 
-for image_content in messages.image_contents:
-    file_id = image_content.image_file.file_id
-    print(f"Image File ID: {file_id}")
-    file_name = f"{file_id}_image_file.png"
-    agents_client.files.save(file_id=file_id, file_name=file_name)
-    print(f"Saved image file to: {Path.cwd() / file_name}")
-
-for file_path_annotation in messages.file_path_annotations:
-    print(f"File Paths:")
-    print(f"Type: {file_path_annotation.type}")
-    print(f"Text: {file_path_annotation.text}")
-    print(f"File ID: {file_path_annotation.file_path.file_id}")
-    print(f"Start Index: {file_path_annotation.start_index}")
-    print(f"End Index: {file_path_annotation.end_index}")
+for msg in messages:
+    # Save every image file in the message
+    for img in msg.image_contents:
+        file_id = img.image_file.file_id
+        file_name = f"{file_id}_image_file.png"
+        agents_client.files.save(file_id=file_id, file_name=file_name)
+        print(f"Saved image file to: {Path.cwd() / file_name}")
+
+    # Print details of every file-path annotation
+    for ann in msg.file_path_annotations:
+        print("File Paths:")
+        print(f"  Type: {ann.type}")
+        print(f"  Text: {ann.text}")
+        print(f"  File ID: {ann.file_path.file_id}")
+        print(f"  Start Index: {ann.start_index}")
+        print(f"  End Index: {ann.end_index}")
 ```
 
 <!-- END SNIPPET -->

sdk/ai/azure-ai-agents/apiview-properties.json

@@ -72,13 +72,6 @@
         "azure.ai.agents.models.MessageTextUrlCitationAnnotation": "Azure.AI.Agents.MessageTextUrlCitationAnnotation",
         "azure.ai.agents.models.MessageTextUrlCitationDetails": "Azure.AI.Agents.MessageTextUrlCitationDetails",
         "azure.ai.agents.models.MicrosoftFabricToolDefinition": "Azure.AI.Agents.MicrosoftFabricToolDefinition",
-        "azure.ai.agents.models.OpenAIPageableListOfAgent": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfAgentThread": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfRunStep": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfThreadMessage": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfThreadRun": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfVectorStore": "Azure.AI.Agents.OpenAIPageableListOf",
-        "azure.ai.agents.models.OpenAIPageableListOfVectorStoreFile": "Azure.AI.Agents.OpenAIPageableListOf",
         "azure.ai.agents.models.OpenApiAuthDetails": "Azure.AI.Agents.OpenApiAuthDetails",
         "azure.ai.agents.models.OpenApiAnonymousAuthDetails": "Azure.AI.Agents.OpenApiAnonymousAuthDetails",
         "azure.ai.agents.models.OpenApiConnectionAuthDetails": "Azure.AI.Agents.OpenApiConnectionAuthDetails",

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_2d7743fd82"
+  "Tag": "python/ai/azure-ai-agents_dab4ae41b1"
 }

sdk/ai/azure-ai-agents/azure/ai/agents/_configuration.py

@@ -46,7 +46,7 @@ def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCr
         self.endpoint = endpoint
         self.credential = credential
         self.api_version = api_version
-        self.credential_scopes = kwargs.pop("credential_scopes", ["https://cognitiveservices.azure.com/.default"])
+        self.credential_scopes = kwargs.pop("credential_scopes", ["https://ai.azure.com/.default"])
         kwargs.setdefault("sdk_moniker", "ai-agents/{}".format(VERSION))
         self.polling_interval = kwargs.get("polling_interval", 30)
         self._configure(**kwargs)

sdk/ai/azure-ai-agents/azure/ai/agents/aio/_configuration.py

@@ -48,7 +48,7 @@ def __init__(
         self.endpoint = endpoint
         self.credential = credential
         self.api_version = api_version
-        self.credential_scopes = kwargs.pop("credential_scopes", ["https://cognitiveservices.azure.com/.default"])
+        self.credential_scopes = kwargs.pop("credential_scopes", ["https://ai.azure.com/.default"])
         kwargs.setdefault("sdk_moniker", "ai-agents/{}".format(VERSION))
         self.polling_interval = kwargs.get("polling_interval", 30)
         self._configure(**kwargs)