Fix toolset in SWE

ComposioHQ · Jun 24, 2024 · 99e0c21 · 99e0c21
1 parent 702abbf
commit 99e0c21
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/composio_swe/composio_swe/agent/swe.py b/python/composio_swe/composio_swe/agent/swe.py
@@ -1,6 +1,7 @@
 import datetime
 import json
 import typing as t
+from itertools import chain
 from pathlib import Path
 from typing import Any, Dict, List
 
@@ -60,8 +61,10 @@ def __init__(self, args: CoderAgentArgs):
         )
         cmd_manager_tool_set = ComposioToolSet().get_tools(apps=[App.CMDMANAGERTOOL])
         history_keeper_tool_set = ComposioToolSet().get_tools(apps=[App.HISTORYKEEPER])
-        self.composio_toolset = (
-            local_workspace_tool_set + cmd_manager_tool_set + history_keeper_tool_set
+        self.composio_toolset = list(
+            chain(
+                local_workspace_tool_set, cmd_manager_tool_set, history_keeper_tool_set
+            )
         )
         self.composio_client = Composio()