Skip to content

Commit d7e2fb9

Browse files
fix: correct event type in partial final chunk test
Change TextMessageContentEvent to TextMessageChunkEvent in test to match actual AG-UI protocol event types. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <[email protected]>
1 parent ed8d02c commit d7e2fb9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

typescript-sdk/integrations/adk-middleware/python/tests/test_adk_agent.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -148,8 +148,8 @@ async def test_partial_final_chunk_uses_streaming_translation(self, adk_agent, s
148148
async def fake_translate(self, adk_event, thread_id, run_id):
149149
nonlocal translate_calls
150150
translate_calls += 1
151-
yield TextMessageContentEvent(
152-
type=EventType.TEXT_MESSAGE_CONTENT,
151+
yield TextMessageChunkEvent(
152+
type=EventType.TEXT_MESSAGE_CHUNK,
153153
message_id=adk_event.id,
154154
delta="chunk"
155155
)
@@ -184,7 +184,7 @@ async def run_async(self, *args, **kwargs):
184184
patch.object(adk_agent, "_create_runner", return_value=FakeRunner()):
185185
events = [event async for event in adk_agent.run(sample_input)]
186186

187-
assert any(isinstance(event, TextMessageContentEvent) for event in events)
187+
assert any(isinstance(event, TextMessageChunkEvent) for event in events)
188188
assert translate_calls == 1
189189
assert lro_calls == 0
190190

0 commit comments

Comments
 (0)