agentic to async

Author: smoreinis

examples/tutorials/10_async/00_base/000_hello_acp/tests/test_agent.py

@@ -1,7 +1,7 @@
 """
-Tests for ab000-hello-acp (agentic agent)
+Tests for ab000-hello-acp (async agent)
 
-This test suite demonstrates testing an agentic agent using the AgentEx testing framework.
+This test suite demonstrates testing an async agent using the AgentEx testing framework.
 
 Test coverage:
 - Event sending and polling for responses
@@ -16,11 +16,15 @@
     pytest tests/test_agent.py -v
 """
 
+import asyncio
 import pytest
 import pytest_asyncio
 
+from agentex.lib.testing.sessions import AsyncAgentTest
+from agentex.lib.testing import stream_agent_response
+
 from agentex.lib.testing import (
-    test_agentic_agent,
+    async_test_agent,
     assert_valid_agent_response,
     assert_agent_response_contains,
 )
@@ -32,30 +36,28 @@ def agent_name():
     """Return the agent name for testing."""
     return AGENT_NAME
 
+
 @pytest_asyncio.fixture
 async def test_agent(agent_name: str):
-    """Fixture to create a test agentic agent."""
-    async with test_agentic_agent(agent_name=agent_name) as test:
+    """Fixture to create a test async agent."""
+    async with async_test_agent(agent_name=agent_name) as test:
         yield test
 
 
 class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, agent_name):
+    async def test_send_event_and_poll(self, test_agent: AsyncAgentTest):
         """Test sending an event and polling for the response."""
-        async with test_agentic_agent(agent_name=agent_name) as test:
-            # First event - should get initial task creation message
-            initial_response = await test.send_event("Start task", timeout_seconds=30.0)
-            assert_valid_agent_response(initial_response)
-            assert_agent_response_contains(initial_response, "Hello! I've received your task")
-
-        # Second event - send user message
-        user_message = "Hello, this is a test message!"
-        response = await test.send_event(user_message, timeout_seconds=30.0)
-
-        # Validate response
+        # Poll for initial task creation message
+        initial_response = await test_agent.poll_for_agent_response(timeout_seconds=15.0)
+        assert_valid_agent_response(initial_response)
+        assert_agent_response_contains(initial_response, "Hello! I've received your task")
+
+        # Send a test message and validate response
+        response = await test_agent.send_event("Hello, this is a test message!", timeout_seconds=30.0)
+        # Validate latest response
         assert_valid_agent_response(response)
         assert_agent_response_contains(response, "Hello! I've received your message")
 
@@ -64,7 +66,7 @@ class TestStreamingEvents:
     """Test streaming event sending."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_stream(self, test_agent):
+    async def test_send_event_and_stream(self, test_agent: AsyncAgentTest):
         """Test sending an event and streaming the response."""
         user_message = "Hello, this is a test message!"
 
@@ -75,7 +77,8 @@ async def test_send_event_and_stream(self, test_agent):
         all_events = []
 
         # Stream events
-        async for event in test_agent.send_event_and_stream(user_message, timeout_seconds=30.0):
+        async for event in stream_agent_response(test_agent.client, test_agent.task_id, timeout=30.0):
+        #async for event in test_agent.send_event_and_stream(user_message, timeout_seconds=30.0):
             all_events.append(event)
             event_type = event.get("type")
 
@@ -98,11 +101,17 @@ async def test_send_event_and_stream(self, test_agent):
                         user_echo_found = True
 
             # Exit early if we've found expected messages
-            if task_creation_found and agent_response_found:
+            if task_creation_found and agent_response_found and user_echo_found:
                 break
 
+        print('all events', all_events)
+        messages = await test_agent.client.messages.list(task_id=test_agent.task_id)
+        print('all messages', messages)
         # Validate we saw expected messages
-        assert task_creation_found or agent_response_found, "Did not receive agent messages"
+        assert task_creation_found, "Did not receive task creation message"
+        assert agent_response_found, "Did not receive agent response to user message"
+        # User echo is optional; no assert 
+        assert user_echo_found, "User echo message not found"
         assert len(all_events) > 0, "Should receive events"
 
 

examples/tutorials/10_async/00_base/010_multiturn/tests/test_agent.py

@@ -1,7 +1,7 @@
 """
-Tests for ab010-multiturn (agentic agent)
+Tests for ab010-multiturn (async agent)
 
-This test suite demonstrates testing a multi-turn agentic agent using the AgentEx testing framework.
+This test suite demonstrates testing a multi-turn async agent using the AgentEx testing framework.
 
 Test coverage:
 - Multi-turn event sending with state management
@@ -21,7 +21,7 @@
 
 from agentex import AsyncAgentex
 from agentex.lib.testing import (
-    test_agentic_agent,
+    async_test_agent,
     assert_valid_agent_response,
 )
 
@@ -39,7 +39,7 @@ async def test_multiturn_with_state_management():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -77,7 +77,7 @@ async def test_multiturn_with_state_management():
 
 @pytest.mark.asyncio
 async def test_streaming_events():
-    """Test streaming events from agentic agent."""
+    """Test streaming events from async agent."""
     # Need client access to check state
     client = AsyncAgentex(api_key="test", base_url="http://localhost:5003")
 
@@ -86,7 +86,7 @@ async def test_streaming_events():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 

examples/tutorials/10_async/00_base/020_streaming/tests/test_agent.py

@@ -1,5 +1,5 @@
 """
-Tests for ab020-streaming (agentic agent)
+Tests for ab020-streaming (async agent)
 
 Test coverage:
 - Event sending and polling
@@ -18,7 +18,7 @@
 from agentex import AsyncAgentex
 from agentex.lib.testing import (
     assert_valid_agent_response,
-    test_agentic_agent,
+    async_test_agent,
 )
 
 AGENT_NAME = "ab020-streaming"
@@ -35,7 +35,7 @@ async def test_send_event_and_poll():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -80,7 +80,7 @@ async def test_streaming_events():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 

examples/tutorials/10_async/00_base/030_tracing/tests/test_agent.py

@@ -1,7 +1,7 @@
 """
-Tests for ab030-tracing (agentic agent)
+Tests for ab030-tracing (async agent)
 
-This test suite demonstrates testing an agentic agent with tracing enabled.
+This test suite demonstrates testing an async agent with tracing enabled.
 
 Test coverage:
 - Basic event sending and polling
@@ -18,7 +18,7 @@
 import pytest
 
 from agentex.lib.testing import (
-    test_agentic_agent,
+    async_test_agent,
 )
 
 AGENT_NAME = "ab030-tracing"
@@ -27,7 +27,7 @@
 @pytest.mark.asyncio
 async def test_basic_event():
     """Test sending an event and receiving a response."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         response = await test.send_event("Hello! Test message", timeout_seconds=30.0)
         # Agent may return empty response depending on configuration
         assert response is not None
@@ -38,7 +38,7 @@ async def test_basic_event():
 @pytest.mark.asyncio
 async def test_streaming_event():
     """Test streaming events from agent."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         events_received = []
 
         async for event in test.send_event_and_stream("Stream this", timeout_seconds=30.0):

examples/tutorials/10_async/00_base/040_other_sdks/tests/test_agent.py

@@ -12,7 +12,7 @@
 import pytest
 
 from agentex import AsyncAgentex
-from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response
+from agentex.lib.testing import async_test_agent, assert_valid_agent_response
 
 AGENT_NAME = "ab040-other-sdks"
 
@@ -28,7 +28,7 @@ async def test_agent_basic():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -65,7 +65,7 @@ async def test_poll_with_tool_use():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -113,7 +113,7 @@ async def test_poll_multiturn():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -153,7 +153,7 @@ async def test_basic_streaming():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 
@@ -218,7 +218,7 @@ async def test_streaming_with_tools():
     agent = next((a for a in agents if a.name == AGENT_NAME), None)
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         # Wait for state initialization
         await asyncio.sleep(1)
 

examples/tutorials/10_async/00_base/080_batch_events/tests/test_agent.py

@@ -14,7 +14,7 @@
 import re
 
 from agentex import AsyncAgentex
-from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response, stream_agent_response
+from agentex.lib.testing import async_test_agent, assert_valid_agent_response, stream_agent_response
 from agentex.types.text_content_param import TextContentParam
 from agentex.types.task_message_content import TextContent
 
@@ -24,7 +24,7 @@
 @pytest.mark.asyncio
 async def test_single_event_and_poll():
     """Test sending a single event and polling for response."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         response = await test.send_event("Process this single event", timeout_seconds=30.0)
         assert_valid_agent_response(response)
         assert "Processed event IDs" in response.content
@@ -41,7 +41,7 @@ async def test_batch_events_and_poll():
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
     num_events = 7
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         for i in range(num_events):
             event_content = TextContentParam(type="text", author="user", content=f"Batch event {i + 1}")
             await client.agents.send_event(agent_id=agent.id, params={"task_id": test.task_id, "content": event_content})
@@ -91,7 +91,7 @@ async def test_batched_streaming():
     assert agent is not None, f"Agent {AGENT_NAME} not found"
 
     num_events = 10
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         for i in range(num_events):
             event_content = TextContentParam(type="text", author="user", content=f"Batch event {i + 1}")
             await client.agents.send_event(agent_id=agent.id, params={"task_id": test.task_id, "content": event_content})

examples/tutorials/10_async/00_base/090_multi_agent_non_temporal/tests/test_agent.py

@@ -10,23 +10,23 @@
 
 import pytest
 
-from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response
+from agentex.lib.testing import async_test_agent, assert_valid_agent_response
 
 AGENT_NAME = "ab090-orchestrator-agent"
 
 
 @pytest.mark.asyncio
 async def test_agent_basic():
     """Test basic agent functionality."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         response = await test.send_event("Test message", timeout_seconds=30.0)
         assert_valid_agent_response(response)
 
 
 @pytest.mark.asyncio
 async def test_agent_streaming():
     """Test streaming responses."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         events = []
         async for event in test.send_event_and_stream("Stream test", timeout_seconds=30.0):
             events.append(event)

examples/tutorials/10_async/10_temporal/000_hello_acp/tests/test_agent.py

@@ -11,23 +11,23 @@
 
 import pytest
 
-from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response
+from agentex.lib.testing import async_test_agent, assert_valid_agent_response
 
 AGENT_NAME = "at000-hello-acp"
 
 
 @pytest.mark.asyncio
 async def test_agent_basic():
     """Test basic agent functionality."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         response = await test.send_event("Test message", timeout_seconds=60.0)
         assert_valid_agent_response(response)
 
 
 @pytest.mark.asyncio
 async def test_agent_streaming():
     """Test streaming responses."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         events = []
         async for event in test.send_event_and_stream("Stream test", timeout_seconds=60.0):
             events.append(event)

examples/tutorials/10_async/10_temporal/010_agent_chat/tests/test_agent.py

@@ -17,23 +17,23 @@
 
 import pytest
 
-from agentex.lib.testing import test_agentic_agent, assert_valid_agent_response
+from agentex.lib.testing import async_test_agent, assert_valid_agent_response
 
 AGENT_NAME = "at010-agent-chat"
 
 
 @pytest.mark.asyncio
 async def test_agent_basic():
     """Test basic agent functionality."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         response = await test.send_event("Test message", timeout_seconds=60.0)
         assert_valid_agent_response(response)
 
 
 @pytest.mark.asyncio
 async def test_agent_streaming():
     """Test streaming responses."""
-    async with test_agentic_agent(agent_name=AGENT_NAME) as test:
+    async with async_test_agent(agent_name=AGENT_NAME) as test:
         events = []
         async for event in test.send_event_and_stream("Stream test", timeout_seconds=60.0):
             events.append(event)