from agentex.types import TextDelta
from agentex.types.agent_rpc_result import (
StreamTaskMessageDelta,
StreamTaskMessageDone,
StreamTaskMessageFull
)
print("User: Can you give me a simple example of supervised learning?\n")
print("Agent: ", end="", flush=True)
# Stream the response
for chunk in client.agents.send_message_stream(
agent_name=AGENT_NAME,
params={
"content": {
"type": "text",
"author": "user",
"content": "Can you give me a simple example of supervised learning?",
},
},
):
# The result is a TaskMessageUpdate
task_message_update = chunk.result
# Print text deltas as they arrive
if isinstance(task_message_update, StreamTaskMessageDelta):
delta = task_message_update.delta
if isinstance(delta, TextDelta):
print(delta.text_delta, end="", flush=True)
elif isinstance(task_message_update, StreamTaskMessageFull):
content = task_message_update.content
if isinstance(content, TextContent):
print(content.content)
print("\n")