Tests fix

Author: KillianLucas

interpreter/core/computer/computer.py

@@ -1,5 +1,5 @@
-import json
 import inspect
+import json
 
 from .ai.ai import Ai
 from .browser.browser import Browser
@@ -58,14 +58,18 @@ def __init__(self, interpreter):
             self.interpreter.max_output
         )  # Should mirror interpreter.max_output
 
+        computer_tools = "\n".join(
+            self._get_all_computer_tools_signature_and_description()
+        )
+
         self.system_message = f"""
 
 # THE COMPUTER API
 
 A python `computer` module is ALREADY IMPORTED, and can be used for many tasks:
 
 ```python
-{"\n".join(self._get_all_computer_tools_signature_and_description())}
+{computer_tools}
 ```
 
 Do not import the computer module, or any of its sub-modules. They are already imported.
@@ -82,7 +86,23 @@ def languages(self, value):
         self.terminal.languages = value
 
     def _get_all_computer_tools_list(self):
-        return [self.mouse, self.keyboard, self.display, self.clipboard, self.mail, self.sms, self.calendar, self.contacts, self.browser, self.os, self.vision, self.skills, self.docs, self.ai, self.files]
+        return [
+            self.mouse,
+            self.keyboard,
+            self.display,
+            self.clipboard,
+            self.mail,
+            self.sms,
+            self.calendar,
+            self.contacts,
+            self.browser,
+            self.os,
+            self.vision,
+            self.skills,
+            self.docs,
+            self.ai,
+            self.files,
+        ]
 
     def _get_all_computer_tools_signature_and_description(self):
         """
@@ -105,34 +125,41 @@ def _extract_tool_info(self, tool):
         """
         Helper function to extract the signature and description of a tool's methods.
         """
-        tool_info = {
-            "signature": tool.__class__.__name__,
-            "methods": []
-        }
+        tool_info = {"signature": tool.__class__.__name__, "methods": []}
         if tool.__class__.__name__ == "Browser":
             methods = []
             for name in dir(tool):
-                if 'driver' in name:
+                if "driver" in name:
                     continue  # Skip methods containing 'driver' in their name
                 attr = getattr(tool, name)
-                if callable(attr) and not name.startswith('_') and not hasattr(attr, '__wrapped__') and not isinstance(attr, property):
+                if (
+                    callable(attr)
+                    and not name.startswith("_")
+                    and not hasattr(attr, "__wrapped__")
+                    and not isinstance(attr, property)
+                ):
                     # Construct the method signature manually
                     param_str = ", ".join(
-                        param for param in attr.__code__.co_varnames[:attr.__code__.co_argcount]
+                        param
+                        for param in attr.__code__.co_varnames[
+                            : attr.__code__.co_argcount
+                        ]
                     )
                     full_signature = f"computer.{tool.__class__.__name__.lower()}.{name}({param_str})"
                     # Get the method description
                     method_description = attr.__doc__ or ""
                     # Append the method details
-                    tool_info["methods"].append({
-                        "signature": full_signature,
-                        "description": method_description.strip()
-                    })
-            return tool_info            
+                    tool_info["methods"].append(
+                        {
+                            "signature": full_signature,
+                            "description": method_description.strip(),
+                        }
+                    )
+            return tool_info
 
         for name, method in inspect.getmembers(tool, predicate=inspect.ismethod):
             # Check if the method should be ignored based on its decorator
-            if not name.startswith("_") and not hasattr(method, '__wrapped__'):
+            if not name.startswith("_") and not hasattr(method, "__wrapped__"):
                 # Get the method signature
                 method_signature = inspect.signature(method)
                 # Construct the signature string without *args and **kwargs
@@ -143,17 +170,20 @@ def _extract_tool_info(self, tool):
                     for param in method_signature.parameters.values()
                     if param.kind not in (param.VAR_POSITIONAL, param.VAR_KEYWORD)
                 )
-                full_signature = f"computer.{tool.__class__.__name__.lower()}.{name}({param_str})"
+                full_signature = (
+                    f"computer.{tool.__class__.__name__.lower()}.{name}({param_str})"
+                )
                 # Get the method description
                 method_description = method.__doc__ or ""
                 # Append the method details
-                tool_info["methods"].append({
-                    "signature": full_signature,
-                    "description": method_description.strip()
-                })
+                tool_info["methods"].append(
+                    {
+                        "signature": full_signature,
+                        "description": method_description.strip(),
+                    }
+                )
         return tool_info
 
-
     def run(self, *args, **kwargs):
         """
         Shortcut for computer.terminal.run