Fix action results (#127)

fix action results Co-authored-by: wangzy <[email protected]>
InternLM · Jan 31, 2024 · 3d99921 · 3d99921
1 parent ca1ab4b
commit 3d99921
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lagent/actions/google_search.py b/lagent/actions/google_search.py
@@ -69,7 +69,7 @@ def run(self, query: str, k: int = 10) -> ActionReturn:
             tool_return.state = ActionStatusCode.HTTP_ERROR
         elif status_code == 200:
             parsed_res = self._parse_results(response)
-            tool_return.result = dict(text=str(parsed_res))
+            tool_return.result = [dict(type='text', content=str(parsed_res))]
             tool_return.state = ActionStatusCode.SUCCESS
         else:
             tool_return.errmsg = str(status_code)

diff --git a/lagent/actions/python_interpreter.py b/lagent/actions/python_interpreter.py
@@ -123,7 +123,7 @@ def _call(self, command: str) -> ActionReturn:
             tool_return.state = ActionStatusCode.API_ERROR
             return tool_return
         try:
-            tool_return.result = dict(text=str(res))
+            tool_return.result = [dict(type='text', content=str(res))]
             tool_return.state = ActionStatusCode.SUCCESS
         except Exception as e:
             tool_return.errmsg = repr(e)