FIX BUG: only the last variable is replaced

atomiechen · Jul 29, 2023 · 5cb64bb · 5cb64bb
1 parent b1a5b7c
commit 5cb64bb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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