agent_protocol_quickstart_echo_task_handler_server.py

1from agent_protocol import Agent, Step, Task
2
3async def task_handler(task: Task) -> None:
4    print(f"Task created: {task.task_id}")
5    # Create a step
6    step = await Agent.db.create_step(task.task_id, name="Echo step")
7    # Mark step as completed
8    await Agent.db.update_step(task.task_id, step.step_id, status="completed", output=task.input)
9    print(f"Step completed: {step.step_id}")
10
11if __name__ == "__main__":
12    Agent.handle_task(task_handler).start()