Fix: skip start_token (#145)

InternLM · Feb 2, 2024 · a2c23ef · a2c23ef
1 parent 3be9ec0
commit a2c23ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lagent/agents/internlm2_agent.py b/lagent/agents/internlm2_agent.py
@@ -170,7 +170,7 @@ def parse(self, message, plugin_executor: ActionExecutor,
             return 'interpreter', message, dict(
                 name=interpreter_executor.action_names()[0],
                 parameters=dict(command=code))
-        return None, message, None
+        return None, message.split(self.tool['start_token'])[0], None
 
     def format_response(self, action_return, name) -> dict:
         if action_return.state == ActionStatusCode.SUCCESS: