Update SDK code ai_vi_0

Author: azure-sdk

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

@@ -17,7 +17,6 @@
 from ._configuration import AgentsClientConfiguration
 from ._utils.serialization import Deserializer, Serializer
 from .operations import (
-    AgentsClientOperationsMixin,
     FilesOperations,
     MessagesOperations,
     RunStepsOperations,
@@ -27,12 +26,13 @@
     VectorStoreFilesOperations,
     VectorStoresOperations,
 )
+from .operations._operations import _AgentsClientOperationsMixin
 
 if TYPE_CHECKING:
     from azure.core.credentials import TokenCredential
 
 
-class AgentsClient(AgentsClientOperationsMixin):  # pylint: disable=too-many-instance-attributes
+class AgentsClient(_AgentsClientOperationsMixin):  # pylint: disable=too-many-instance-attributes
     """AgentsClient.
 
     :ivar threads: ThreadsOperations operations

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

@@ -1,4 +1,3 @@
-# pylint: disable=line-too-long,useless-suppression
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.

sdk/ai/azure-ai-agents/azure/ai/agents/_utils/model_base.py

@@ -1,4 +1,4 @@
-# pylint: disable=line-too-long,useless-suppression,too-many-lines
+# pylint: disable=too-many-lines
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.

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

@@ -17,7 +17,6 @@
 from .._utils.serialization import Deserializer, Serializer
 from ._configuration import AgentsClientConfiguration
 from .operations import (
-    AgentsClientOperationsMixin,
     FilesOperations,
     MessagesOperations,
     RunStepsOperations,
@@ -27,12 +26,13 @@
     VectorStoreFilesOperations,
     VectorStoresOperations,
 )
+from .operations._operations import _AgentsClientOperationsMixin
 
 if TYPE_CHECKING:
     from azure.core.credentials_async import AsyncTokenCredential
 
 
-class AgentsClient(AgentsClientOperationsMixin):  # pylint: disable=too-many-instance-attributes
+class AgentsClient(_AgentsClientOperationsMixin):  # pylint: disable=too-many-instance-attributes
     """AgentsClient.
 
     :ivar threads: ThreadsOperations operations

sdk/ai/azure-ai-agents/azure/ai/agents/aio/operations/__init__.py

@@ -20,7 +20,6 @@
 from ._operations import VectorStoresOperations  # type: ignore
 from ._operations import VectorStoreFilesOperations  # type: ignore
 from ._operations import VectorStoreFileBatchesOperations  # type: ignore
-from ._operations import AgentsClientOperationsMixin  # type: ignore
 
 from ._patch import __all__ as _patch_all
 from ._patch import *
@@ -35,7 +34,6 @@
     "VectorStoresOperations",
     "VectorStoreFilesOperations",
     "VectorStoreFileBatchesOperations",
-    "AgentsClientOperationsMixin",
 ]
 __all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 _patch_sdk()

sdk/ai/azure-ai-agents/azure/ai/agents/aio/operations/_operations.py

@@ -3975,7 +3975,7 @@ async def get_next(_continuation_token=None):
         return AsyncItemPaged(get_next, extract_data)
 
 
-class AgentsClientOperationsMixin(
+class _AgentsClientOperationsMixin(
     ClientMixinABC[AsyncPipelineClient[HttpRequest, AsyncHttpResponse], AgentsClientConfiguration]
 ):
 

sdk/ai/azure-ai-agents/azure/ai/agents/operations/__init__.py

@@ -20,7 +20,6 @@
 from ._operations import VectorStoresOperations  # type: ignore
 from ._operations import VectorStoreFilesOperations  # type: ignore
 from ._operations import VectorStoreFileBatchesOperations  # type: ignore
-from ._operations import AgentsClientOperationsMixin  # type: ignore
 
 from ._patch import __all__ as _patch_all
 from ._patch import *
@@ -35,7 +34,6 @@
     "VectorStoresOperations",
     "VectorStoreFilesOperations",
     "VectorStoreFileBatchesOperations",
-    "AgentsClientOperationsMixin",
 ]
 __all__.extend([p for p in _patch_all if p not in __all__])  # pyright: ignore
 _patch_sdk()

sdk/ai/azure-ai-agents/azure/ai/agents/operations/_operations.py

@@ -1,4 +1,4 @@
-# pylint: disable=line-too-long,useless-suppression,too-many-lines
+# pylint: disable=too-many-lines
 # coding=utf-8
 # --------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
@@ -5087,7 +5087,9 @@ def get_next(_continuation_token=None):
         return ItemPaged(get_next, extract_data)
 
 
-class AgentsClientOperationsMixin(ClientMixinABC[PipelineClient[HttpRequest, HttpResponse], AgentsClientConfiguration]):
+class _AgentsClientOperationsMixin(
+    ClientMixinABC[PipelineClient[HttpRequest, HttpResponse], AgentsClientConfiguration]
+):
 
     @overload
     def create_agent(

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

@@ -64,10 +64,7 @@ async def fetch_and_print_new_agent_response(
     return response.id
 
 
-def create_research_summary(
-        message : ThreadMessage,
-        filepath: str = "research_summary.md"
-) -> None:
+def create_research_summary(message: ThreadMessage, filepath: str = "research_summary.md") -> None:
     if not message:
         print("No message content provided, cannot create research summary.")
         return
@@ -167,4 +164,4 @@ async def main() -> None:
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    asyncio.run(main())

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

@@ -60,10 +60,7 @@ def fetch_and_print_new_agent_response(
     return response.id
 
 
-def create_research_summary(
-        message : ThreadMessage,
-        filepath: str = "research_summary.md"
-) -> None:
+def create_research_summary(message: ThreadMessage, filepath: str = "research_summary.md") -> None:
     if not message:
         print("No message content provided, cannot create research summary.")
         return
@@ -154,9 +151,7 @@ def create_research_summary(
             print(f"Run failed: {run.last_error}")
 
         # Fetch the final message from the agent in the thread and create a research summary
-        final_message = agents_client.messages.get_last_message_by_role(
-            thread_id=thread.id, role=MessageRole.AGENT
-        )
+        final_message = agents_client.messages.get_last_message_by_role(thread_id=thread.id, role=MessageRole.AGENT)
         if final_message:
             create_research_summary(final_message)