Fix ADKAgent test suite for new architecture

Updated all test fixtures and endpoint implementation to match the
new ADKAgent.run(input_data) signature after removing agent_id
parameter. All 271 tests now pass with the simplified architecture
from issue-24-cleanup-agent-registration.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <[email protected]>

Author: mefogle

typescript-sdk/integrations/adk-middleware/CHANGELOG.md

@@ -7,7 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
-## [0.5.0] - 2025-01-08
+## [0.5.0] - 2025-08-05
 
 ### Breaking Changes
 - **BREAKING**: ADKAgent constructor now requires `adk_agent` parameter instead of `agent_id` for direct agent embedding
@@ -25,8 +25,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 - **FIXED**: All 271 tests now pass with new simplified architecture
-- **EXAMPLES**: Updated examples to demonstrate direct agent embedding pattern
+- **TESTS**: Updated all test fixtures to match new ADKAgent.run(input_data) signature without agent_id parameter
+- **TESTS**: Fixed test expectations in test_endpoint.py to work with direct agent embedding architecture
 - **TESTS**: Updated all test fixtures to work with new agent embedding pattern
+- **EXAMPLES**: Updated examples to demonstrate direct agent embedding pattern
 
 ### Added
 - **NEW**: SystemMessage support for ADK agents (issue #22) - SystemMessages as first message are now appended to agent instructions

typescript-sdk/integrations/adk-middleware/src/adk_middleware/endpoint.py

@@ -36,7 +36,7 @@ async def adk_endpoint(input_data: RunAgentInput, request: Request):
         async def event_generator():
             """Generate events from ADK agent."""
             try:
-                async for event in agent.run(input_data, agent_id):
+                async for event in agent.run(input_data):
                     try:
                         encoded = encoder.encode(event)
                         logger.debug(f"HTTP Response: {encoded}")

typescript-sdk/integrations/adk-middleware/tests/test_endpoint.py

@@ -117,8 +117,8 @@ def test_endpoint_agent_id_extraction(self, mock_encoder_class, app, mock_agent,
         client = TestClient(app)
         response = client.post("/agent123", json=sample_input.model_dump())
 
-        # Agent should be called with agent_id extracted from path
-        mock_agent.run.assert_called_once_with(sample_input, "agent123")
+        # Agent should be called with just the input data
+        mock_agent.run.assert_called_once_with(sample_input)
         assert response.status_code == 200
 
     @patch('adk_middleware.endpoint.EventEncoder')
@@ -142,8 +142,8 @@ def test_endpoint_root_path_agent_id(self, mock_encoder_class, app, mock_agent,
         client = TestClient(app)
         response = client.post("/", json=sample_input.model_dump())
 
-        # Agent should be called with empty agent_id for root path
-        mock_agent.run.assert_called_once_with(sample_input, "")
+        # Agent should be called with just the input data
+        mock_agent.run.assert_called_once_with(sample_input)
         assert response.status_code == 200
 
     @patch('adk_middleware.endpoint.EventEncoder')
@@ -167,7 +167,7 @@ def test_endpoint_successful_event_streaming(self, mock_logger, mock_encoder_cla
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event1
             yield mock_event2
 
@@ -204,7 +204,7 @@ def test_endpoint_encoding_error_handling(self, mock_logger, mock_encoder_class,
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event
 
         mock_agent.run = mock_agent_run
@@ -245,7 +245,7 @@ def test_endpoint_encoding_error_double_failure(self, mock_logger, mock_encoder_
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event
 
         mock_agent.run = mock_agent_run
@@ -276,7 +276,7 @@ def test_endpoint_agent_error_handling(self, mock_logger, mock_encoder_class, ap
         mock_encoder_class.return_value = mock_encoder
 
         # Mock agent to raise an error
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             raise RuntimeError("Agent failed")
 
         mock_agent.run = mock_agent_run
@@ -309,7 +309,7 @@ def test_endpoint_agent_error_encoding_failure(self, mock_logger, mock_encoder_c
         mock_encoder_class.return_value = mock_encoder
 
         # Mock agent to raise an error
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             raise RuntimeError("Agent failed")
 
         mock_agent.run = mock_agent_run
@@ -345,7 +345,7 @@ def test_endpoint_returns_streaming_response(self, mock_encoder_class, app, mock
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event
 
         mock_agent.run = mock_agent_run
@@ -385,7 +385,7 @@ def test_endpoint_no_accept_header(self, mock_encoder_class, app, mock_agent, sa
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event
 
         mock_agent.run = mock_agent_run
@@ -459,7 +459,7 @@ def test_create_app_functional_test(self, mock_encoder_class, mock_agent):
             run_id="test_run"
         )
 
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             yield mock_event
 
         mock_agent.run = mock_agent_run
@@ -530,8 +530,8 @@ def test_full_endpoint_flow(self, mock_encoder_class, mock_agent, sample_input):
 
         call_args = []
 
-        async def mock_agent_run(input_data, agent_id):
-            call_args.append((input_data, agent_id))
+        async def mock_agent_run(input_data):
+            call_args.append(input_data)
             for event in events:
                 yield event
 
@@ -552,8 +552,7 @@ async def mock_agent_run(input_data, agent_id):
 
         # Verify agent was called correctly
         assert len(call_args) == 1
-        assert call_args[0][0] == sample_input
-        assert call_args[0][1] == "integration"
+        assert call_args[0] == sample_input
 
         # Verify events were encoded
         assert mock_encoder.encode.call_count == len(events)
@@ -589,7 +588,7 @@ def test_endpoint_with_long_running_stream(self, mock_encoder_class, mock_agent,
         mock_encoder_class.return_value = mock_encoder
 
         # Mock agent to return many events
-        async def mock_agent_run(input_data, agent_id):
+        async def mock_agent_run(input_data):
             for i in range(10):
                 yield RunStartedEvent(
                     type=EventType.RUN_STARTED,