bugfix: add toolName in tool_response

akai-shuuichi · Nov 14, 2024 · 5deb75a · 5deb75a
1 parent ba5cdc2
commit 5deb75a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -1022,7 +1022,7 @@ def messages2prompt(self,
             if message['role'] == 'tool':
                 if index == 0 or messages[index - 1]['role'] != 'tool':
                     ret += f'{self.user}'
-                ret += f"\n<tool_response>\n{message['name']}{message['content']}\n</tool_response>"
+                ret += f"\n<tool_response>\n{message['name']}:{message['content']}\n</tool_response>"
                 if index == len(messages) - 1 or messages[index +
                                                           1]['role'] != 'tool':
                     ret += f'{self.eoh}'