diff --git a/haystack/nodes/prompt/prompt_node.py b/haystack/nodes/prompt/prompt_node.py index b19888352..2bfa702c6 100644 --- a/haystack/nodes/prompt/prompt_node.py +++ b/haystack/nodes/prompt/prompt_node.py @@ -301,7 +301,12 @@ class PromptNode(BaseComponent): results = self(prompt_collector=prompt_collector, **invocation_context) prompt_template_resolved: PromptTemplate = invocation_context.pop("prompt_template") - output_variable = self.output_variable or prompt_template_resolved.output_variable or "results" + + try: + output_variable = self.output_variable or prompt_template_resolved.output_variable or "results" + except: + output_variable = "results" + invocation_context[output_variable] = results invocation_context["prompts"] = prompt_collector final_result: Dict[str, Any] = {output_variable: results, "invocation_context": invocation_context}