[Fix] search type (#34)

InternLM · Sep 16, 2023 · f050692 · f050692
1 parent d9ec190
commit f050692
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lagent/actions/google_search.py b/lagent/actions/google_search.py
@@ -76,7 +76,7 @@ def __call__(self, query: str) -> ActionReturn:
             ActionReturn: The action return.
         """
 
-        tool_return = ActionReturn(url=None, args=None)
+        tool_return = ActionReturn(url=None, args=None, type=self.name)
         status_code, response = self._search(
             query, search_type=self.search_type, k=self.k)
         # convert search results to ToolReturn format