run black

Author: howieleung

sdk/ai/azure-ai-agents/azure/ai/agents/models/_patch.py

@@ -40,8 +40,10 @@
 )
 
 from azure.ai.agents.models import RunStatus
+
 # NOTE: Avoid importing RunsOperations here to prevent circular import with operations package.
 from typing import TYPE_CHECKING
+
 if TYPE_CHECKING:  # pragma: no cover - type checking only
     from ..operations import RunsOperations  # noqa: F401
 
@@ -1704,7 +1706,9 @@ def execute_tool_calls(self, tool_calls: List[Any]) -> Any:
         """
         return self._execute_tool_calls(tool_calls)
 
-    def _execute_tool_calls(self, tool_calls: List[Any], run: Optional[ThreadRun] = None, run_handler: Optional['RunHandler'] = None) -> Any:
+    def _execute_tool_calls(
+        self, tool_calls: List[Any], run: Optional[ThreadRun] = None, run_handler: Optional["RunHandler"] = None
+    ) -> Any:
         tool_outputs = []
 
         for tool_call in tool_calls:
@@ -1726,7 +1730,7 @@ def _execute_tool_calls(self, tool_calls: List[Any], run: Optional[ThreadRun] =
                     tool_output = {"tool_call_id": tool_call.id, "output": str(e)}
                     tool_outputs.append(tool_output)
 
-        return tool_outputs    
+        return tool_outputs
 
 
 class AsyncToolSet(BaseToolSet):
@@ -1777,6 +1781,7 @@ async def execute_tool_calls(self, tool_calls: List[Any]) -> Any:
 BaseAsyncAgentEventHandlerT = TypeVar("BaseAsyncAgentEventHandlerT", bound="BaseAsyncAgentEventHandler")
 # BaseAgentEventHandlerT is defined after BaseAgentEventHandler class to avoid forward reference during parsing.
 
+
 async def async_chain(*iterators: AsyncIterator[T]) -> AsyncIterator[T]:
     for iterator in iterators:
         async for item in iterator:
@@ -1878,9 +1883,7 @@ def _start(self, runs_operations: "RunsOperations", run: ThreadRun, polling_inte
             time.sleep(polling_interval)
             run = runs_operations.get(thread_id=run.thread_id, run_id=run.id)
 
-            if run.status == RunStatus.REQUIRES_ACTION and isinstance(
-                run.required_action, SubmitToolOutputsAction
-            ):
+            if run.status == RunStatus.REQUIRES_ACTION and isinstance(run.required_action, SubmitToolOutputsAction):
                 tool_calls = run.required_action.submit_tool_outputs.tool_calls
                 if not tool_calls:
                     logger.warning("No tool calls provided - cancelling run")
@@ -1896,7 +1899,8 @@ def _start(self, runs_operations: "RunsOperations", run: ThreadRun, polling_inte
                     if _has_errors_in_toolcalls_output(tool_outputs):
                         if current_retry >= runs_operations._function_tool_max_retry:  # pylint:disable=no-else-return
                             logger.warning(
-                                "Tool outputs contain errors - reaching max retry %s", runs_operations._function_tool_max_retry
+                                "Tool outputs contain errors - reaching max retry %s",
+                                runs_operations._function_tool_max_retry,
                             )
                             run = runs_operations.cancel(thread_id=run.thread_id, run_id=run.id)
                         else:
@@ -1966,22 +1970,21 @@ def submit_mcp_tool_approval(
         tool_call: RequiredMcpToolCall,
         **kwargs,
     ) -> Optional[ToolApproval]:
-            # NOTE: Implementation intentionally returns None; override in subclasses for real approval logic.
-            """Return a ``ToolApproval`` for an MCP tool call or ``None`` to indicate rejection/cancellation.
-
-            Override this to implement approval policies (interactive prompt, RBAC, heuristic checks, etc.).
-            Returning ``None`` triggers cancellation logic in ``_start``.
+        # NOTE: Implementation intentionally returns None; override in subclasses for real approval logic.
+        """Return a ``ToolApproval`` for an MCP tool call or ``None`` to indicate rejection/cancellation.
 
-            :param run: Current run containing the MCP approval request.
-            :type run: ThreadRun
-            :param tool_call: The MCP tool call requiring approval.
-            :type tool_call: RequiredMcpToolCall
-            :keyword kwargs: Additional keyword arguments for extensibility.
-            :return: A populated ``ToolApproval`` instance on approval, or ``None`` to decline.
-            :rtype: Optional[ToolApproval]
-            """
-            return None
+        Override this to implement approval policies (interactive prompt, RBAC, heuristic checks, etc.).
+        Returning ``None`` triggers cancellation logic in ``_start``.
 
+        :param run: Current run containing the MCP approval request.
+        :type run: ThreadRun
+        :param tool_call: The MCP tool call requiring approval.
+        :type tool_call: RequiredMcpToolCall
+        :keyword kwargs: Additional keyword arguments for extensibility.
+        :return: A populated ``ToolApproval`` instance on approval, or ``None`` to decline.
+        :rtype: Optional[ToolApproval]
+        """
+        return None
 
 
 class BaseAgentEventHandler(Iterator[T]):
@@ -2044,6 +2047,7 @@ def until_done(self) -> None:
         except StopIteration:
             pass
 
+
 # Now that BaseAgentEventHandler is defined, we can bind the TypeVar.
 BaseAgentEventHandlerT = TypeVar("BaseAgentEventHandlerT", bound="BaseAgentEventHandler")
 

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

@@ -555,7 +555,7 @@ def create_and_process(
         # Monitor and process the run status
         if not run_handler:
             run_handler = _models.RunHandler()
-            
+
         return run_handler._start(self, run, polling_interval)
 
     @overload

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

@@ -49,11 +49,15 @@
 # Initialize function tool with user functions
 functions = FunctionTool(functions=user_functions)
 
+
 class MyRunHandler(RunHandler):
-    def submit_function_call_output(self, run: ThreadRun, tool_call: RequiredFunctionToolCall, tool_call_details: RequiredFunctionToolCallDetails) -> Optional[Any]:
+    def submit_function_call_output(
+        self, run: ThreadRun, tool_call: RequiredFunctionToolCall, tool_call_details: RequiredFunctionToolCallDetails
+    ) -> Optional[Any]:
         print(f"Call function: {tool_call_details.name}")
         return functions.execute(tool_call)
 
+
 with project_client:
     agents_client = project_client.agents
 

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

@@ -65,14 +65,18 @@
 mcp_tool.allow_tool(search_api_code)
 print(f"Allowed tools: {mcp_tool.allowed_tools}")
 
+
 class MyRunHandler(RunHandler):
-    def submit_mcp_tool_approval(self, run: ThreadRun, tool_call: RequiredMcpToolCall, **kwargs) -> Optional[ToolApproval]:
+    def submit_mcp_tool_approval(
+        self, run: ThreadRun, tool_call: RequiredMcpToolCall, **kwargs
+    ) -> Optional[ToolApproval]:
         return ToolApproval(
             tool_call_id=tool_call.id,
             approve=True,
             headers=mcp_tool.headers,
         )
 
+
 # Create agent with MCP tool and process agent run
 with project_client:
     agents_client = project_client.agents
@@ -83,7 +87,7 @@ def submit_mcp_tool_approval(self, run: ThreadRun, tool_call: RequiredMcpToolCal
         model=os.environ["MODEL_DEPLOYMENT_NAME"],
         name="my-mcp-agent",
         instructions="You are a helpful agent that can use MCP tools to assist users. Use the available MCP tools to answer questions and perform tasks.",
-        toolset=toolset
+        toolset=toolset,
     )
     # [END create_agent_with_mcp_tool]
 
@@ -105,7 +109,7 @@ def submit_mcp_tool_approval(self, run: ThreadRun, tool_call: RequiredMcpToolCal
     # [START handle_tool_approvals]
     # Create and process agent run in thread with MCP tools
     mcp_tool.update_headers("SuperSecret", "123456")
-    
+
     run_handler = MyRunHandler()
     # mcp_tool.set_approval_mode("never")  # Uncomment to disable approval requirement
     run = agents_client.runs.create_and_process(thread_id=thread.id, agent_id=agent.id, run_handler=run_handler)