Add tool role for langchain usage (#2558)

InternLM · Oct 9, 2024 · c722ff5 · c722ff5
1 parent 7c6b107
commit c722ff5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lmdeploy/model.py b/lmdeploy/model.py
@@ -838,10 +838,12 @@ def messages2prompt(self,
             return self.get_prompt(messages, sequence_start)
         box_map = dict(user=self.user,
                        ipython=self.ipython,
+                       tool=self.ipython,
                        assistant=self.assistant,
                        system=self.system)
         eox_map = dict(user=self.eoh,
                        ipython=self.eoi,
+                       tool=self.eoi,
                        assistant=self.eoa + self.separator,
                        system=self.eosys)
         ret = ''