From 50a7c2c92ce861a6f7324d3e85cd8797bee59175 Mon Sep 17 00:00:00 2001 From: takatost Date: Mon, 1 Apr 2024 12:51:01 +0800 Subject: [PATCH] fix bug --- api/core/app/apps/base_app_runner.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/api/core/app/apps/base_app_runner.py b/api/core/app/apps/base_app_runner.py index 3ecd3f4375..f1f426b27e 100644 --- a/api/core/app/apps/base_app_runner.py +++ b/api/core/app/apps/base_app_runner.py @@ -169,10 +169,11 @@ class AppRunner: text=advanced_completion_prompt_template.prompt ) - memory_config.role_prefix = MemoryConfig.RolePrefix( - user=advanced_completion_prompt_template.role_prefix.user, - assistant=advanced_completion_prompt_template.role_prefix.assistant - ) + if advanced_completion_prompt_template.role_prefix: + memory_config.role_prefix = MemoryConfig.RolePrefix( + user=advanced_completion_prompt_template.role_prefix.user, + assistant=advanced_completion_prompt_template.role_prefix.assistant + ) else: prompt_template = [] for message in prompt_template_entity.advanced_chat_prompt_template.messages: