Skip to content

Commit c525791

Browse files
committed
Add temporal tests
1 parent b3ab3cf commit c525791

File tree

113 files changed

+1597
-15972
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

113 files changed

+1597
-15972
lines changed

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

Lines changed: 3 additions & 64 deletions
Original file line numberDiff line numberDiff line change
@@ -23,14 +23,12 @@
2323
import pytest_asyncio
2424
from test_utils.agentic import (
2525
poll_messages,
26-
stream_agent_response,
2726
send_event_and_poll_yielding,
2827
)
2928

3029
from agentex import AsyncAgentex
3130
from agentex.types import TaskMessage
3231
from agentex.types.agent_rpc_params import ParamsCreateTaskRequest
33-
from agentex.types.text_content_param import TextContentParam
3432

3533
# Configuration from environment variables
3634
AGENTEX_API_BASE_URL = os.environ.get("AGENTEX_API_BASE_URL", "http://localhost:5003")
@@ -83,7 +81,7 @@ async def test_send_event_and_poll(self, client: AsyncAgentex, agent_id: str):
8381
if message.content and message.content.type == "text" and message.content.author == "agent":
8482
assert "Hello! I've received your task" in message.content.content
8583
break
86-
84+
8785
await asyncio.sleep(1.5)
8886
# Send an event and poll for response
8987
user_message = "Hello, this is a test message!"
@@ -105,67 +103,8 @@ class TestStreamingEvents:
105103

106104
@pytest.mark.asyncio
107105
async def test_send_event_and_stream(self, client: AsyncAgentex, agent_id: str):
108-
"""Test sending an event and streaming the response."""
109-
task_response = await client.agents.create_task(agent_id, params=ParamsCreateTaskRequest(name=uuid.uuid1().hex))
110-
task = task_response.result
111-
assert task is not None
112-
113-
user_message = "Hello, this is a test message!"
106+
pass
114107

115-
# Collect events from stream
116-
all_events = []
117-
118-
# Flags to track what we've received
119-
task_creation_found = False
120-
user_echo_found = False
121-
agent_response_found = False
122-
123-
async def collect_stream_events(): #noqa: ANN101
124-
nonlocal task_creation_found, user_echo_found, agent_response_found
125-
126-
async for event in stream_agent_response(
127-
client=client,
128-
task_id=task.id,
129-
timeout=30,
130-
):
131-
# Check events as they arrive
132-
event_type = event.get("type")
133-
if event_type == "full":
134-
content = event.get("content", {})
135-
if content.get("content") is None:
136-
continue # Skip empty content
137-
if content.get("type") == "text" and content.get("author") == "agent":
138-
# Check for initial task creation message
139-
if "Hello! I've received your task" in content.get("content", ""):
140-
task_creation_found = True
141-
# Check for agent response to user message
142-
elif "Hello! I've received your message" in content.get("content", ""):
143-
# Agent response should come after user echo
144-
assert user_echo_found, "Agent response arrived before user message echo (incorrect order)"
145-
agent_response_found = True
146-
elif content.get("type") == "text" and content.get("author") == "user":
147-
# Check for user message echo
148-
if content.get("content") == user_message:
149-
user_echo_found = True
150-
151-
# Exit early if we've found all expected messages
152-
if task_creation_found and user_echo_found and agent_response_found:
153-
break
154-
155-
assert task_creation_found, "Task creation message not found in stream"
156-
assert user_echo_found, "User message echo not found in stream"
157-
assert agent_response_found, "Agent response not found in stream"
158-
159-
160-
# Start streaming task
161-
stream_task = asyncio.create_task(collect_stream_events())
162-
163-
# Send the event
164-
event_content = TextContentParam(type="text", author="user", content=user_message)
165-
await client.agents.send_event(agent_id=agent_id, params={"task_id": task.id, "content": event_content})
166-
167-
# Wait for streaming to complete
168-
await stream_task
169108

170109
if __name__ == "__main__":
171-
pytest.main([__file__, "-v"])
110+
pytest.main([__file__, "-v"])

examples/tutorials/10_agentic/10_temporal/010_agent_chat/.dockerignore

Lines changed: 0 additions & 43 deletions
This file was deleted.

examples/tutorials/10_agentic/10_temporal/010_agent_chat/Dockerfile

Lines changed: 0 additions & 47 deletions
This file was deleted.

examples/tutorials/10_agentic/10_temporal/010_agent_chat/README.md

Lines changed: 0 additions & 47 deletions
This file was deleted.

0 commit comments

Comments
 (0)