yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-05 11:38:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: prompt_template_resolved.output_variable is NoneType issue (#4976)

Browse Source 
* try except instead or

* fix black formatting

* bug fix

* revert back the formatting
This commit is contained in:
Fanli Lin 2023-05-29 16:48:10 +08:00 committed by GitHub
parent 4aaf4fcc31
commit 7001aee3fe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
haystack/nodes/prompt/prompt_node.py
View File

@ -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}