From 5cb64bb0be340af6d584f393ecb8db7e34332547 Mon Sep 17 00:00:00 2001 From: Howard <919756752@qq.com> Date: Sat, 29 Jul 2023 11:19:29 +0800 Subject: [PATCH] FIX BUG: only the last variable is replaced --- src/handyllm/prompt_converter.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/handyllm/prompt_converter.py b/src/handyllm/prompt_converter.py index 8dd1b55..3e2aef1 100644 --- a/src/handyllm/prompt_converter.py +++ b/src/handyllm/prompt_converter.py @@ -62,11 +62,10 @@ def chat_replace_variables(self, chat, variable_map: dict, inplace=False): else: new_chat = [] for message in chat: + new_message = {"role": message['role'], "content": message['content']} for var, value in variable_map.items(): if var in message['content']: - new_message = {"role": message['role'], "content": message['content'].replace(var, value)} - else: - new_message = {"role": message['role'], "content": message['content']} + new_message = {"role": new_message['role'], "content": new_message['content'].replace(var, value)} new_chat.append(new_message) return new_chat