Update _debug as well when there are more than two streams being joined.

deepset-ai · Apr 10, 2024 · 8991c12 · 8991c12
1 parent d07e238
commit 8991c12
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/haystack/pipelines/base.py b/haystack/pipelines/base.py
@@ -610,6 +610,11 @@ def run(  # type: ignore
                                     updated_input["meta"] = meta
                             else:
                                 existing_input["inputs"].append(node_output)
+                                if "_debug" in node_output.keys():
+                                    existing_input["_debug"] = {
+                                        **existing_input.get("_debug", {}),
+                                        **node_output.get("_debug", {}),
+                                    }
                                 additional_input = self._combine_node_outputs(existing_input, node_output)
                                 updated_input = {**additional_input, **existing_input}
                             queue[n] = updated_input
@@ -767,6 +772,11 @@ async def _arun(  # noqa: C901,PLR0912 type: ignore
                                     updated_input["meta"] = meta
                             else:
                                 existing_input["inputs"].append(node_output)
+                                if "_debug" in node_output.keys():
+                                    existing_input["_debug"] = {
+                                        **existing_input.get("_debug", {}),
+                                        **node_output.get("_debug", {}),
+                                    }
                                 additional_input = self._combine_node_outputs(existing_input, node_output)
                                 updated_input = {**additional_input, **existing_input}
                             queue[n] = updated_input