From 9b5568eb749ee5cf47c24b8f8b9e7bdd704c19bd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?=
 =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= <mirzagitov03@mail.ru>
Date: Mon, 18 May 2026 11:26:10 +0000
Subject: [PATCH] Add test_stream_agent.py

---
 test_stream_agent.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 test_stream_agent.py

diff --git a/test_stream_agent.py b/test_stream_agent.py
new file mode 100644
index 0000000..f3d2cb8
--- /dev/null
+++ b/test_stream_agent.py
@@ -0,0 +1,24 @@
+from langchain_core.messages import AIMessage
+
+from stream_agent import format_message
+
+
+def test_format_message_with_content():
+    message = AIMessage(content="hello")
+
+    assert format_message(message) == "hello"
+
+
+def test_format_message_with_tool_call():
+    message = AIMessage(
+        content="",
+        tool_calls=[
+            {
+                "name": "get_demo_price",
+                "args": {"product": "молоко", "city": "Казань"},
+                "id": "call-1",
+            }
+        ],
+    )
+
+    assert "get_demo_price" in format_message(message)