done coding, tests still dont pass

Author: leetwito

.gitignore

@@ -236,3 +236,7 @@ misc/
 # Ignore litellm_uuid.txt
 litellm_uuid.txt
 .aider*
+file.txt
+numbers.txt
+poetry.lock
+poetry.lock

interpreter/core/computer/computer.py

@@ -1,4 +1,5 @@
 import json
+import inspect
 
 from .ai.ai import Ai
 from .browser.browser import Browser
@@ -57,24 +58,14 @@ def __init__(self, interpreter):
             self.interpreter.max_output
         )  # Should mirror interpreter.max_output
 
-        self.system_message = """
+        self.system_message = f"""
 
 # THE COMPUTER API
 
 A python `computer` module is ALREADY IMPORTED, and can be used for many tasks:
 
 ```python
-computer.browser.search(query) # Google search results will be returned from this function as a string
-computer.files.edit(path_to_file, original_text, replacement_text) # Edit a file
-computer.calendar.create_event(title="Meeting", start_date=datetime.datetime.now(), end_date=datetime.datetime.now() + datetime.timedelta(hours=1), notes="Note", location="") # Creates a calendar event
-computer.calendar.get_events(start_date=datetime.date.today(), end_date=None) # Get events between dates. If end_date is None, only gets events for start_date
-computer.calendar.delete_event(event_title="Meeting", start_date=datetime.datetime) # Delete a specific event with a matching title and start date, you may need to get use get_events() to find the specific event object first
-computer.contacts.get_phone_number("John Doe")
-computer.contacts.get_email_address("John Doe")
-computer.mail.send("[email protected]", "Meeting Reminder", "Reminder that our meeting is at 3pm today.", ["path/to/attachment.pdf", "path/to/attachment2.pdf"]) # Send an email with a optional attachments
-computer.mail.get(4, unread=True) # Returns the [number] of unread emails, or all emails if False is passed
-computer.mail.unread_count() # Returns the number of unread emails
-computer.sms.send("555-123-4567", "Hello from the computer!") # Send a text message. MUST be a phone number, so use computer.contacts.get_phone_number frequently here
+{self._get_all_computer_tools_signature_and_description()}
 ```
 
 Do not import the computer module, or any of its sub-modules. They are already imported.
@@ -90,6 +81,57 @@ def languages(self):
     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]
+
+    def _get_all_computer_tools_signature_and_description(self):
+        """
+        This function returns a list of all the computer tools that are available with their signature and description from the function docstrings.
+        for example:
+        computer.browser.search(query) # Searches the web for the specified query and returns the results.
+        computer.calendar.create_event(title: str, start_date: datetime.datetime, end_date: datetime.datetime, location: str = "", notes: str = "", calendar: str = None) -> str # Creates a new calendar event in the default calendar with the given parameters using AppleScript.
+        """
+        tools = self._get_all_computer_tools_list()
+        tools_signature_and_description = []
+        for tool in tools:
+            tool_info = self._extract_tool_info(tool)
+            for method in tool_info["methods"]:
+                # Format as tool_signature # tool_description
+                formatted_info = f"{method['signature']} # {method['description']}"
+                tools_signature_and_description.append(formatted_info)
+        return tools_signature_and_description
+
+    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": []
+        }
+        for name, method in inspect.getmembers(tool, predicate=inspect.ismethod):
+            if not name.startswith("_"):
+                # Get the method signature
+                method_signature = inspect.signature(method)
+                # Construct the signature string without *args and **kwargs
+                param_str = ", ".join(
+                    f"{param.name}"
+                    if param.default == param.empty
+                    else f"{param.name}={param.default!r}"
+                    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})"
+                # Get the method description
+                method_description = method.__doc__ or ""
+                # Append the method details
+                tool_info["methods"].append({
+                    "signature": full_signature,
+                    "description": method_description.strip()
+                })
+        return tool_info
+
+
     def run(self, *args, **kwargs):
         """
         Shortcut for computer.terminal.run

poetry.lock

@@ -83,6 +83,7 @@ isort = "^5.12.0"
 pre-commit = "^3.5.0"
 pytest = "^7.4.0"
 sniffio = "^1.3.0"
+websockets = "^13.1"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

pyproject.toml

@@ -0,0 +1,21 @@
+import unittest
+from unittest import mock
+from interpreter.core.computer.computer import Computer
+
+class TestComputer(unittest.TestCase):
+    def setUp(self):
+        self.computer = Computer(mock.Mock())
+
+    def test_get_all_computer_tools_list(self):
+        # Act
+        tools_list = self.computer._get_all_computer_tools_list()
+
+        # Assert
+        self.assertEqual(len(tools_list), 15)
+
+    def test_get_all_computer_tools_signature_and_description(self):
+        # Act
+        tools_description = self.computer._get_all_computer_tools_signature_and_description()
+
+        # Assert
+        self.assertGreater(len(tools_description), 64)