Fix the copy

deepset-ai · Apr 10, 2024 · 1177d3f · 1177d3f
1 parent 5f80731
commit 1177d3f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/haystack/nodes/prompt/prompt_node.py b/haystack/nodes/prompt/prompt_node.py
@@ -247,7 +247,10 @@ def _prepare(  # type: ignore
         """
         Prepare prompt invocation.
         """
-        invocation_context = invocation_context.copy() or {}
+        if invocation_context is None:
+            invocation_context = {}
+        else:
+            invocation_context = invocation_context.copy()
 
         if query and "query" not in invocation_context:
             invocation_context["query"] = query