From 2c23caacd488c89670f6eb5b743d970b0f91257a Mon Sep 17 00:00:00 2001 From: John Wang Date: Fri, 26 May 2023 20:49:38 +0800 Subject: [PATCH] fix: introduction key error (#221) --- api/core/conversation_message_task.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/core/conversation_message_task.py b/api/core/conversation_message_task.py index dcf0c0b957..3b73b19d1f 100644 --- a/api/core/conversation_message_task.py +++ b/api/core/conversation_message_task.py @@ -80,7 +80,10 @@ class ConversationMessageTask: if introduction: prompt_template = OutLinePromptTemplate.from_template(template=PromptBuilder.process_template(introduction)) prompt_inputs = {k: self.inputs[k] for k in prompt_template.input_variables if k in self.inputs} - introduction = prompt_template.format(**prompt_inputs) + try: + introduction = prompt_template.format(**prompt_inputs) + except KeyError: + pass if self.app_model_config.pre_prompt: pre_prompt = PromptBuilder.process_template(self.app_model_config.pre_prompt)