From 7916fbf24208f3ee5b9907ebf93e03a30757dbf9 Mon Sep 17 00:00:00 2001
From: Victor Giers <mail@victorgiers.com>
Date: Mon, 15 Jun 2026 21:48:00 +0200
Subject: [PATCH] Add test case to ensure limit node preserves sources from
 merged context

---
 backend/tests/test_workflow_runtime.py | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/backend/tests/test_workflow_runtime.py b/backend/tests/test_workflow_runtime.py
index d254ed2..9522714 100644
--- a/backend/tests/test_workflow_runtime.py
+++ b/backend/tests/test_workflow_runtime.py
@@ -122,6 +122,25 @@ class RuntimeTests(unittest.IsolatedAsyncioTestCase):
         self.assertEqual(statuses["no"], "skipped")
         db.close()
 
+    async def test_limit_preserves_sources_from_merged_context(self):
+        value = graph(
+            [
+                node("input", "input"),
+                node("a", "tool", {"tool": "test.echo", "arguments": {"text": "a" * 2000, "sources": [{"url": "https://a.example", "snippet": "a" * 500}]}}),
+                node("b", "tool", {"tool": "test.echo", "arguments": {"text": "b" * 2000, "sources": [{"url": "https://b.example", "snippet": "b" * 500}]}}),
+                node("merge", "merge", {"values": [{"$ref": "nodes.a.output"}, {"$ref": "nodes.b.output"}], "deduplicate_by": "url"}),
+                node("limit", "limit", {"value": {"$ref": "nodes.merge.output"}, "maximum_chars": 80}),
+                node("output", "output", {"value": {"content": "done", "sources": {"$ref": "nodes.limit.output.sources"}}}),
+            ],
+            [edge("input", "a"), edge("a", "b"), edge("b", "merge"), edge("merge", "limit"), edge("limit", "output")],
+        )
+        run = await self.execute(self.create_run(value))
+        self.assertEqual(run.status, "completed")
+        db = self.Session()
+        assistant = db.query(ChatMessage).filter(ChatMessage.role == "assistant").one()
+        self.assertEqual([item["url"] for item in json.loads(assistant.sources_json)], ["https://a.example", "https://b.example"])
+        db.close()
+
     async def test_failure_cancellation_and_tool_limit(self):
         failing = graph(
             [node("input", "input"), node("bad", "tool", {"tool": "test.echo", "arguments": {"text": "x", "fail": True}}), node("output", "output", {"value": {"$ref": "nodes.bad.output"}})],