From 1906c65e666d49bfe79e8071aa852dc520ea1531 Mon Sep 17 00:00:00 2001 From: liujiangning Date: Thu, 11 Apr 2024 11:59:07 +0800 Subject: [PATCH] fix bug of Internlm2Protocol.parse --- lagent/agents/internlm2_agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/lagent/agents/internlm2_agent.py b/lagent/agents/internlm2_agent.py index 6e159420..6ebfb539 100644 --- a/lagent/agents/internlm2_agent.py +++ b/lagent/agents/internlm2_agent.py @@ -175,7 +175,8 @@ def parse(self, message, plugin_executor: ActionExecutor, code = code.split(self.tool['end'].strip())[0].strip() return 'interpreter', message, dict( name=interpreter_executor.action_names()[0], - parameters=dict(command=code)) + parameters=dict( + command=code)) if interpreter_executor else None return None, message.split(self.tool['start_token'])[0], None def format_response(self, action_return, name) -> dict: