fix: add CustomCodeAgent to handle both list and dictionary tool formats

app.py

@@ -232,9 +232,27 @@ logger.debug("Created tools dictionary:")
 for name, tool in tools_dict.items():
     logger.debug(f"  {name}: {tool}")
 
-agent = CodeAgent(
+# Create a custom CodeAgent class that handles both list and dictionary requirements
+class CustomCodeAgent(CodeAgent):
+    def __init__(self, *args, **kwargs):
+        # Store the tools dictionary for template rendering
+        self.tools_dict = kwargs.pop('tools_dict', {})
+        super().__init__(*args, **kwargs)
+
+    def initialize_system_prompt(self):
+        # Override to use tools_dict for template rendering
+        return self.populate_template(
+            self.prompt_templates["system_prompt"],
+            tools=self.tools_dict,
+            task=self.task,
+            managed_agents=self.managed_agents,
+            authorized_imports=self.authorized_imports
+        )
+
+agent = CustomCodeAgent(
     model=model,
-    tools=tools_dict,  # Pass tools as a dictionary for template rendering
+    tools=tools,  # Pass tools as a list for _setup_tools
+    tools_dict=tools_dict,  # Pass tools as a dictionary for template rendering
     max_steps=15,
     verbosity_level=2,
     grammar=None,