From 7001aee3fe4249b8945e6cf42e89be28b6acea9c Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Mon, 29 May 2023 16:48:10 +0800 Subject: [PATCH] fix: prompt_template_resolved.output_variable is NoneType issue (#4976) * try except instead or * fix black formatting * bug fix * revert back the formatting --- haystack/nodes/prompt/prompt_node.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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}