From 4aacf3256f7b4127af1b724af6fa1fc041f7b463 Mon Sep 17 00:00:00 2001
From: Aksh Parekh <parekh.aksh.c@gmail.com>
Date: Tue, 28 Oct 2025 18:40:36 -0700
Subject: [PATCH] [EXAMPLE] Token Stream in Agent class

---
 tests/structs/test_agent_stream_token.py | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 tests/structs/test_agent_stream_token.py

diff --git a/tests/structs/test_agent_stream_token.py b/tests/structs/test_agent_stream_token.py
new file mode 100644
index 00000000..5cd02207
--- /dev/null
+++ b/tests/structs/test_agent_stream_token.py
@@ -0,0 +1,9 @@
+from swarms.structs.agent import Agent
+
+agent = Agent(
+    model_name="gpt-4.1", 
+    max_loops=1, 
+    stream=True,
+)
+
+agent.run("Tell me a short story about a robot learning to paint.")