fix bug of Internlm2Protocol.parse (#180)

InternLM · Apr 11, 2024 · e43646f · e43646f
1 parent 581d9fb
commit e43646f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: