Fix update_agents (Azure#42261)

* Fix update_agents

* Update sdk/ai/azure-ai-agents/tests/test_agents_mock_overloads.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* Update sdk/ai/azure-ai-agents/CHANGELOG.md

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: howieleung

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

@@ -4,6 +4,10 @@
 
 ## 1.1.0b5 (Unreleased)
 
+### Bugs Fixed
+
+- Fixed `update_agent` to execute with body as a keyword parameter.
+
 ### Features Added
 
 - Support `tool_resources` for run async operations.

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

@@ -589,8 +589,8 @@ def update_agent(
 
         if body is not _Unset:
             if isinstance(body, io.IOBase):
-                return super().update_agent(body=body, content_type=content_type, **kwargs)
-            return super().update_agent(body=body, **kwargs)
+                return super().update_agent(agent_id, body, content_type=content_type, **kwargs)
+            return super().update_agent(agent_id, body, **kwargs)
 
         if toolset is not None:
             tools = toolset.definitions

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

@@ -587,8 +587,8 @@ async def update_agent(
 
         if body is not _Unset:
             if isinstance(body, io.IOBase):
-                return await super().update_agent(body=body, content_type=content_type, **kwargs)
-            return await super().update_agent(body=body, **kwargs)
+                return await super().update_agent(agent_id, body, content_type=content_type, **kwargs)
+            return await super().update_agent(agent_id, body, **kwargs)
 
         if toolset is not None:
             tools = toolset.definitions

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

@@ -137,3 +137,22 @@ async def test_create_stream(
         await async_agent.runs.stream(thread_id, body=dict_to_io_bytes(body))
 
         assertion.same_http_requests_from(operation_count=6, api_per_operation_count=1)
+
+    @pytest.mark.asyncio
+    @assert_same_http_requests
+    async def test_update_agents(
+        self, agent: AgentsClient, async_agent: AsyncAgentsClient, assertion: OverloadAssertion
+    ):
+        model = "model"
+        agent_id = "agent_id"
+        body = {"model": model}
+
+        agent.update_agent(agent_id, model=model)
+        agent.update_agent(agent_id, body=body)
+        agent.update_agent(agent_id, body=dict_to_io_bytes(body))
+
+        await async_agent.update_agent(agent_id, model=model)
+        await async_agent.update_agent(agent_id, body=body)
+        await async_agent.update_agent(agent_id, body=dict_to_io_bytes(body))
+
+        assertion.same_http_requests_from(operation_count=6, api_per_operation_count=1)