Howie/update agents 2 (#42262)

* Fix update_agents

* update change log

* Update test_agents_mock_overloads.py

* Update CHANGELOG.md

Author: howieleung

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

@@ -6,6 +6,7 @@
 
 ### Bugs Fixed
 - `AgentsResponseFormatOption`, `MessageInputContent`, `MessageAttachmentToolDefinition`, `AgentsToolChoiceOption` are now public.
+- Fixed `update_agents` to execute with body as a keyword parameter.
 
 ### Sample updates
 

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)