Merge pull request #175 from scaleapi/dm/fix-agentic-test-issues

Fix agentic runs for tests

Author: danielmillerp

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

@@ -65,7 +65,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and polling for the response."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
@@ -104,7 +104,7 @@ class TestStreamingEvents:
     """Test streaming event sending."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_stream(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_stream(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and streaming the response."""
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
         task = task_response.result

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

@@ -65,7 +65,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and polling for the response."""
         # TODO: Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -65,7 +65,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and polling for the response."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -55,7 +55,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and polling for the response."""
         # TODO: Create a task for this conversation
         # task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
@@ -83,7 +83,7 @@ class TestStreamingEvents:
     """Test streaming event sending."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_stream(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_stream(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and streaming the response."""
         # TODO: Create a task for this conversation
         # task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -74,7 +74,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling with MCP tools."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, agent_id: str):
         """Test sending a simple event and polling for the response (no tool use)."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -66,7 +66,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending a single event and polling for the response."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -62,7 +62,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_multi_agent_workflow_complete(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_multi_agent_workflow_complete(self, client: AsyncAgentex, agent_id: str):
         """Test the complete multi-agent workflow with all agents using polling that yields messages."""
         # Create a task for the orchestrator
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
@@ -146,7 +146,7 @@ class TestStreamingEvents:
     """Test streaming event sending."""
 
     @pytest.mark.asyncio
-    async def test_multi_agent_workflow_streaming(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_multi_agent_workflow_streaming(self, client: AsyncAgentex, agent_id: str):
         """Test the multi-agent workflow with streaming responses and early exit."""
         # Create a task for the orchestrator
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

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

@@ -65,7 +65,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and polling for the response."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
@@ -104,7 +104,7 @@ class TestStreamingEvents:
     """Test streaming event sending."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_stream(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_stream(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and streaming the response."""
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
         task = task_response.result

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

@@ -74,7 +74,7 @@ class TestNonStreamingEvents:
     """Test non-streaming event sending and polling with OpenAI Agents SDK."""
 
     @pytest.mark.asyncio
-    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, agent_id: str):
         """Test sending a simple event and polling for the response (no tool use)."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))

examples/tutorials/10_agentic/10_temporal/020_state_machine/tests/test_agent.py

@@ -72,7 +72,7 @@ async def agent_id(client, agent_name):
 class TestNonStreamingEvents:
     """Test non-streaming event sending and polling with state machine workflow."""
     @pytest.mark.asyncio
-    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, agent_id: str):
         """Test sending a simple event and polling for the response (no tool use)."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
@@ -127,7 +127,7 @@ async def test_send_event_and_poll_simple_query(self, client: AsyncAgentex, _age
 class TestStreamingEvents:
     """Test streaming event sending with state machine workflow."""
     @pytest.mark.asyncio
-    async def test_send_event_and_stream(self, client: AsyncAgentex, _agent_name: str, agent_id: str):
+    async def test_send_event_and_stream(self, client: AsyncAgentex, agent_id: str):
         """Test sending an event and streaming the response."""
         # Create a task for this conversation
         task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))