wrapper working now!

src/goose/cli/config.py

@@ -87,10 +87,11 @@ def ensure_config(name: str) -> Profile:
                     + pretty_diff(before.read(), after.read())
                 )
             )
-            should_update = Confirm.ask(
-                "Do you want to update your profile to use the latest?",
-                default=False,
-            )
+            # should_update = Confirm.ask(
+            #     "Do you want to update your profile to use the latest?",
+            #     default=False,
+            # )
+            should_update = False
             if should_update:
                 profiles[name] = profile
                 write_config(profiles)

src/goose/language_server/base.py

@@ -13,7 +13,7 @@
 from goose.language_server.core.lsp_constants import LSPConstants
 from goose.language_server.core import lsp_types
 
-import goose.language_server.types as language_server_types
+from goose.language_server import language_server_types
 from goose.language_server.logger import LanguageServerLogger
 from goose.language_server.core.server import (
     LanguageServerHandler,

src/goose/language_server/client.py

@@ -5,7 +5,7 @@
 
 from goose.language_server.base import LanguageServer
 from goose.language_server.type_helpers import ensure_all_methods_implemented
-import goose.language_server.types as language_server_types
+from goose.language_server import language_server_types
 from goose.utils.language import Language
 from typing import Any, Callable, Iterator, List, Tuple, TypeVar, Union
 

src/goose/toolkit/base.py

@@ -61,5 +61,9 @@ def tools(self) -> Tuple[Tool, ...]:
         This default method looks for functions on the toolkit annotated
         with @tool.
         """
-        candidates = inspect.getmembers(self, predicate=inspect.ismethod)
+
+        def predicate(obj: object) -> bool:
+            return inspect.ismethod(obj) or (hasattr(obj, "_is_method") and obj._is_method)
+
+        candidates = inspect.getmembers(self, predicate=predicate)
         return (Tool.from_function(candidate) for _, candidate in candidates if getattr(candidate, "_is_tool", None))

src/goose/toolkit/language_server.py

@@ -1,4 +1,6 @@
-from typing import List, Optional, Tuple, Type
+import functools
+import types
+from typing import Any, Callable, List, Optional, Tuple, Type, TypeVar
 from goose.language_server.client import LanguageServerClient
 from goose.language_server.config import LanguageServerConfig
 from goose.language_server.logger import LanguageServerLogger
@@ -7,6 +9,18 @@
 from goose.utils import load_plugins
 
 
+def method_changes_file(func: Callable) -> Callable:
+    @functools.wraps(func)
+    def wrap_method(*args: list, **kwargs: dict) -> Callable:
+        print(f"Before calling {func.__name__}")
+        result = func(*args, **kwargs)
+        print(f"After calling {func.__name__}")
+        return result
+
+    wrap_method._is_method = True
+    return wrap_method
+
+
 class LanguageServerCoordinator(Toolkit):
     _instance: Optional["LanguageServerCoordinator"] = None
 
@@ -33,6 +47,15 @@ def __init__(self, notifier: Notifier, requires: Optional[Requirements] = None)
             ls = language_server_cls.from_env(config=language_server_config, logger=language_server_logger)
             self.language_server_client.register_language_server(ls)
 
+        developer_toolkit_instance = requires.get("developer")
+        for method in ["write_file", "patch_file"]:
+            decorated_method = method_changes_file(getattr(developer_toolkit_instance, method))
+            setattr(
+                developer_toolkit_instance,
+                method,
+                decorated_method,
+            )
+
     @tool
     def request_definition(self, file_path: str, line: int, column: int) -> List[Tuple[str, int]]:
         """