renamed all (#3207)

python/packages/azurefunctions/tests/test_app.py

@@ -10,7 +10,7 @@ from unittest.mock import ANY, AsyncMock, Mock, patch
 import azure.durable_functions as df
 import azure.functions as func
 import pytest
-from agent_framework import AgentRunResponse, ChatMessage, ErrorContent
+from agent_framework import AgentResponse, ChatMessage, ErrorContent
 
 from agent_framework_azurefunctions import AgentFunctionApp
 from agent_framework_azurefunctions._app import WAIT_FOR_RESPONSE_FIELD, WAIT_FOR_RESPONSE_HEADER
@@ -332,7 +332,7 @@ class TestAgentEntityOperations:
         """Test that entity can run agent operation."""
         mock_agent = Mock()
         mock_agent.run = AsyncMock(
-            return_value=AgentRunResponse(messages=[ChatMessage(role="assistant", text="Test response")])
+            return_value=AgentResponse(messages=[ChatMessage(role="assistant", text="Test response")])
         )
 
         entity = AgentEntity(mock_agent)
@@ -343,7 +343,7 @@ class TestAgentEntityOperations:
             {"message": "Test message", "thread_id": "test-conv-123", "correlationId": "corr-app-entity-1"},
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Test response"
         assert entity.state.message_count == 2
 
@@ -351,7 +351,7 @@ class TestAgentEntityOperations:
         """Test that the entity stores conversation history."""
         mock_agent = Mock()
         mock_agent.run = AsyncMock(
-            return_value=AgentRunResponse(messages=[ChatMessage(role="assistant", text="Response 1")])
+            return_value=AgentResponse(messages=[ChatMessage(role="assistant", text="Response 1")])
         )
 
         entity = AgentEntity(mock_agent)
@@ -390,7 +390,7 @@ class TestAgentEntityOperations:
         """Test that the entity increments the message count."""
         mock_agent = Mock()
         mock_agent.run = AsyncMock(
-            return_value=AgentRunResponse(messages=[ChatMessage(role="assistant", text="Response")])
+            return_value=AgentResponse(messages=[ChatMessage(role="assistant", text="Response")])
         )
 
         entity = AgentEntity(mock_agent)
@@ -437,7 +437,7 @@ class TestAgentEntityFactory:
         """Test that the entity function handles the run operation."""
         mock_agent = Mock()
         mock_agent.run = AsyncMock(
-            return_value=AgentRunResponse(messages=[ChatMessage(role="assistant", text="Response")])
+            return_value=AgentResponse(messages=[ChatMessage(role="assistant", text="Response")])
         )
 
         entity_function = create_agent_entity(mock_agent)
@@ -465,7 +465,7 @@ class TestAgentEntityFactory:
         """Test that the entity function handles the deprecated run_agent operation for backward compatibility."""
         mock_agent = Mock()
         mock_agent.run = AsyncMock(
-            return_value=AgentRunResponse(messages=[ChatMessage(role="assistant", text="Response")])
+            return_value=AgentResponse(messages=[ChatMessage(role="assistant", text="Response")])
         )
 
         entity_function = create_agent_entity(mock_agent)
@@ -619,7 +619,7 @@ class TestErrorHandling:
             mock_context, {"message": "Test message", "thread_id": "conv-1", "correlationId": "corr-app-error-1"}
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert len(result.messages) == 1
         content = result.messages[0].contents[0]
         assert isinstance(content, ErrorContent)

python/packages/azurefunctions/tests/test_entities.py

@@ -12,7 +12,7 @@ from typing import Any, TypeVar
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
-from agent_framework import AgentRunResponse, AgentRunResponseUpdate, ChatMessage, ErrorContent, Role
+from agent_framework import AgentResponse, AgentResponseUpdate, ChatMessage, ErrorContent, Role
 from pydantic import BaseModel
 
 from agent_framework_azurefunctions._durable_agent_state import (
@@ -37,12 +37,12 @@ def _role_value(chat_message: DurableAgentStateMessage) -> str:
     return str(role_value)
 
 
-def _agent_response(text: str | None) -> AgentRunResponse:
-    """Create an AgentRunResponse with a single assistant message."""
+def _agent_response(text: str | None) -> AgentResponse:
+    """Create an AgentResponse with a single assistant message."""
     message = (
         ChatMessage(role="assistant", text=text) if text is not None else ChatMessage(role="assistant", contents=[])
     )
-    return AgentRunResponse(messages=[message])
+    return AgentResponse(messages=[message])
 
 
 class RecordingCallback:
@@ -54,12 +54,12 @@ class RecordingCallback:
 
     async def on_streaming_response_update(
         self,
-        update: AgentRunResponseUpdate,
+        update: AgentResponseUpdate,
         context: Any,
     ) -> None:
         await self.stream_mock(update, context)
 
-    async def on_agent_response(self, response: AgentRunResponse, context: Any) -> None:
+    async def on_agent_response(self, response: AgentResponse, context: Any) -> None:
         await self.response_mock(response, context)
 
 
@@ -132,7 +132,7 @@ class TestAgentEntityRunAgent:
         assert getattr(sent_message.role, "value", sent_message.role) == "user"
 
         # Verify result
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Test response"
 
     async def test_run_agent_executes_agent(self) -> None:
@@ -159,18 +159,18 @@ class TestAgentEntityRunAgent:
         assert getattr(sent_message.role, "value", sent_message.role) == "user"
 
         # Verify result
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Test response"
 
     async def test_run_agent_streaming_callbacks_invoked(self) -> None:
         """Ensure streaming updates trigger callbacks and run() is not used."""
 
         updates = [
-            AgentRunResponseUpdate(text="Hello"),
-            AgentRunResponseUpdate(text=" world"),
+            AgentResponseUpdate(text="Hello"),
+            AgentResponseUpdate(text=" world"),
         ]
 
-        async def update_generator() -> AsyncIterator[AgentRunResponseUpdate]:
+        async def update_generator() -> AsyncIterator[AgentResponseUpdate]:
             for update in updates:
                 yield update
 
@@ -192,7 +192,7 @@ class TestAgentEntityRunAgent:
             },
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert "Hello" in result.text
         assert callback.stream_mock.await_count == len(updates)
         assert callback.response_mock.await_count == 1
@@ -239,7 +239,7 @@ class TestAgentEntityRunAgent:
             },
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Final response"
         assert callback.stream_mock.await_count == 0
         assert callback.response_mock.await_count == 1
@@ -605,7 +605,7 @@ class TestErrorHandling:
             mock_context, {"message": "Message", "thread_id": "conv-1", "correlationId": "corr-entity-error-1"}
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert len(result.messages) == 1
         content = result.messages[0].contents[0]
         assert isinstance(content, ErrorContent)
@@ -624,7 +624,7 @@ class TestErrorHandling:
             mock_context, {"message": "Message", "thread_id": "conv-1", "correlationId": "corr-entity-error-2"}
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert len(result.messages) == 1
         content = result.messages[0].contents[0]
         assert isinstance(content, ErrorContent)
@@ -643,7 +643,7 @@ class TestErrorHandling:
             mock_context, {"message": "Message", "thread_id": "conv-1", "correlationId": "corr-entity-error-3"}
         )
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert len(result.messages) == 1
         content = result.messages[0].contents[0]
         assert isinstance(content, ErrorContent)
@@ -682,7 +682,7 @@ class TestErrorHandling:
         )
 
         # Even on error, message info should be preserved
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert len(result.messages) == 1
         content = result.messages[0].contents[0]
         assert isinstance(content, ErrorContent)
@@ -793,7 +793,7 @@ class TestRunRequestSupport:
 
         result = await entity.run(mock_context, request)
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Response"
 
     async def test_run_agent_with_dict_request(self) -> None:
@@ -814,7 +814,7 @@ class TestRunRequestSupport:
 
         result = await entity.run(mock_context, request_dict)
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == "Response"
 
     async def test_run_agent_with_string_raises_without_correlation(self) -> None:
@@ -869,7 +869,7 @@ class TestRunRequestSupport:
 
         result = await entity.run(mock_context, request)
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         assert result.text == '{"answer": 42}'
         assert result.value is None
 
@@ -887,7 +887,7 @@ class TestRunRequestSupport:
 
         result = await entity.run(mock_context, request)
 
-        assert isinstance(result, AgentRunResponse)
+        assert isinstance(result, AgentResponse)
         # Agent should have been called (tool disabling is framework-dependent)
         mock_agent.run.assert_called_once()
 

python/packages/azurefunctions/tests/test_orchestration.py

@@ -6,7 +6,7 @@ from typing import Any
 from unittest.mock import Mock
 
 import pytest
-from agent_framework import AgentRunResponse, AgentThread, ChatMessage
+from agent_framework import AgentResponse, AgentThread, ChatMessage
 from azure.durable_functions.models.Task import TaskBase, TaskState
 
 from agent_framework_azurefunctions import AgentFunctionApp, DurableAIAgent
@@ -39,7 +39,7 @@ def _create_entity_task(task_id: int = 1) -> TaskBase:
 
 
 class TestAgentResponseHelpers:
-    """Tests for helper utilities that prepare AgentRunResponse values."""
+    """Tests for helper utilities that prepare AgentResponse values."""
 
     @staticmethod
     def _create_agent_task() -> AgentTask:
@@ -48,7 +48,7 @@ class TestAgentResponseHelpers:
 
     def test_load_agent_response_from_instance(self) -> None:
         task = self._create_agent_task()
-        response = AgentRunResponse(messages=[ChatMessage(role="assistant", text='{"foo": "bar"}')])
+        response = AgentResponse(messages=[ChatMessage(role="assistant", text='{"foo": "bar"}')])
 
         loaded = task._load_agent_response(response)
 
@@ -57,7 +57,7 @@ class TestAgentResponseHelpers:
 
     def test_load_agent_response_from_serialized(self) -> None:
         task = self._create_agent_task()
-        serialized = AgentRunResponse(messages=[ChatMessage(role="assistant", text="structured")]).to_dict()
+        serialized = AgentResponse(messages=[ChatMessage(role="assistant", text="structured")]).to_dict()
         serialized["value"] = {"answer": 42}
 
         loaded = task._load_agent_response(serialized)
@@ -65,7 +65,7 @@ class TestAgentResponseHelpers:
         assert loaded is not None
         assert loaded.value == {"answer": 42}
         loaded_dict = loaded.to_dict()
-        assert loaded_dict["type"] == "agent_run_response"
+        assert loaded_dict["type"] == "agent_response"
 
     def test_load_agent_response_rejects_none(self) -> None:
         task = self._create_agent_task()
@@ -86,7 +86,7 @@ class TestAgentResponseHelpers:
 
         # Simulate successful entity task completion
         entity_task.state = TaskState.SUCCEEDED
-        entity_task.result = AgentRunResponse(messages=[ChatMessage(role="assistant", text="Test response")]).to_dict()
+        entity_task.result = AgentResponse(messages=[ChatMessage(role="assistant", text="Test response")]).to_dict()
 
         # Clear pending_tasks to simulate that parent has processed the child
         task.pending_tasks.clear()
@@ -94,9 +94,9 @@ class TestAgentResponseHelpers:
         # Call try_set_value
         task.try_set_value(entity_task)
 
-        # Verify task completed successfully with AgentRunResponse
+        # Verify task completed successfully with AgentResponse
         assert task.state == TaskState.SUCCEEDED
-        assert isinstance(task.result, AgentRunResponse)
+        assert isinstance(task.result, AgentResponse)
         assert task.result.text == "Test response"
 
     def test_try_set_value_failure(self) -> None:
@@ -128,9 +128,7 @@ class TestAgentResponseHelpers:
 
         # Simulate successful entity task with JSON response
         entity_task.state = TaskState.SUCCEEDED
-        entity_task.result = AgentRunResponse(
-            messages=[ChatMessage(role="assistant", text='{"answer": "42"}')]
-        ).to_dict()
+        entity_task.result = AgentResponse(messages=[ChatMessage(role="assistant", text='{"answer": "42"}')]).to_dict()
 
         # Clear pending_tasks to simulate that parent has processed the child
         task.pending_tasks.clear()
@@ -140,7 +138,7 @@ class TestAgentResponseHelpers:
 
         # Verify task completed and value was parsed
         assert task.state == TaskState.SUCCEEDED
-        assert isinstance(task.result, AgentRunResponse)
+        assert isinstance(task.result, AgentResponse)
         assert isinstance(task.result.value, TestSchema)
         assert task.result.value.answer == "42"
 
@@ -152,7 +150,7 @@ class TestAgentResponseHelpers:
             name: str
 
         task = self._create_agent_task()
-        response = AgentRunResponse(messages=[ChatMessage(role="assistant", text='{"name": "test"}')])
+        response = AgentResponse(messages=[ChatMessage(role="assistant", text='{"name": "test"}')])
 
         # Value should be None initially
         assert response.value is None
@@ -173,7 +171,7 @@ class TestAgentResponseHelpers:
 
         task = self._create_agent_task()
         existing_value = SampleSchema(name="existing")
-        response = AgentRunResponse(
+        response = AgentResponse(
             messages=[ChatMessage(role="assistant", text='{"name": "new"}')],
             value=existing_value,
         )