Support disable tools

Author: ericdallo

docs/configuration.md

@@ -131,8 +131,11 @@ interface Config {
     anthropicApiKey?: string;
     rules: [{path: string;}];
     nativeTools: {
-        {[key: string] {enabled: boolean}}
+        filesystem: {enabled: boolean};
+         shell: {enabled: boolean;
+                 excludeCommands: string[]};
     };
+    disabledTools: string[],
     toolCall?: {
       manualApproval?: boolean,
     };
@@ -174,9 +177,10 @@ interface Config {
   "openaiApiKey" : null,
   "anthropicApiKey" : null,
   "rules" : [],
-  "nativeTools": {"filesystem": {"enabled": true}
-                  "shell": {"enabled": true
+  "nativeTools": {"filesystem": {"enabled": true},
+                  "shell": {"enabled": true, 
                             "excludeCommands": []}},
+  "disabledTools": [],
   "toolCall": {
     "manualApproval": false,
   },

docs/protocol.md

@@ -843,6 +843,11 @@ interface ServerTool {
      * The server tool parameters.
      */
     parameters: any; 
+    
+    /**
+     * Whther this tool is disabled.
+     */
+    disabled?: boolean;
 }
 ```
 

src/eca/config.clj

@@ -22,6 +22,7 @@
    :nativeTools {:filesystem {:enabled true}
                  :shell {:enabled true
                          :excludeCommands []}}
+   :disabledTools []
    :mcpTimeoutSeconds 60
    :mcpServers {}
    :ollama {:host "http://localhost"

src/eca/features/tools.clj

@@ -8,15 +8,16 @@
    [eca.features.tools.shell :as f.tools.shell]
    [eca.features.tools.util :as tools.util]
    [eca.logger :as logger]
-   [eca.messenger :as messenger])
+   [eca.messenger :as messenger]
+   [eca.shared :refer [assoc-some]])
   (:import
    [java.util Map]))
 
 (set! *warn-on-reflection* true)
 
 (def ^:private logger-tag "[TOOLS]")
 
-(defn native-definitions [db config]
+(defn ^:private native-definitions [db config]
   (into
    {}
    (map (fn [[name tool]]
@@ -38,9 +39,13 @@
   "Returns all available tools, including both native ECA tools
    (like filesystem and shell tools) and tools provided by MCP servers."
   [db config]
-  (concat
-   (mapv #(assoc % :origin :native) (native-tools db config))
-   (mapv #(assoc % :origin :mcp) (f.mcp/all-tools db))))
+  (let [disabled-tools (set (get-in config [:disabledTools] []))]
+    (filterv
+     (fn [tool]
+       (not (contains? disabled-tools (:name tool))))
+     (concat
+      (mapv #(assoc % :origin :native) (native-tools db config))
+      (mapv #(assoc % :origin :mcp) (f.mcp/all-tools db))))))
 
 (defn call-tool! [^String name ^Map arguments db config]
   (logger/info logger-tag (format "Calling tool '%s' with args '%s'" name arguments))
@@ -49,12 +54,17 @@
     (f.mcp/call-tool! name arguments db)))
 
 (defn init-servers! [db* messenger config]
-  (messenger/tool-server-updated messenger {:type :native
-                                            :name "ECA"
-                                            :status "running"
-                                            :tools (native-tools @db* config)})
-  (f.mcp/initialize-servers-async!
-   {:on-server-updated (fn [server]
-                         (messenger/tool-server-updated messenger (assoc server :type :mcp)))}
-   db*
-   config))
+  (let [disabled-tools (set (get-in config [:disabledTools] []))
+        with-tool-status (fn [tool]
+                           (assoc-some tool :disabled (contains? disabled-tools (:name tool))))]
+    (messenger/tool-server-updated messenger {:type :native
+                                              :name "ECA"
+                                              :status "running"
+                                              :tools (mapv with-tool-status (native-tools @db* config))})
+    (f.mcp/initialize-servers-async!
+     {:on-server-updated (fn [server]
+                           (messenger/tool-server-updated messenger (-> server
+                                                                        (assoc :type :mcp)
+                                                                        (update :tools #(mapv with-tool-status %)))))}
+     db*
+     config)))

src/eca/handlers.clj

@@ -36,23 +36,25 @@
             {}
             ollama-models))))
 
-(defn initialize [{:keys [db* messenger config]} params]
+(defn initialize [{:keys [db* config]} params]
   (logger/logging-task
    :eca/initialize
    (swap! db* assoc
           :client-info (:client-info params)
           :workspace-folders (:workspace-folders params)
           :client-capabilities (:capabilities params)
           :chat-behavior (or (-> params :initialization-options :chat-behavior) (:chat-behavior @db*)))
-   (future
-     (f.tools/init-servers! db* messenger config))
    (initialize-extra-models! db* config)
    {:models (keys (:models @db*))
     :chat-default-model (f.chat/default-model @db* config)
     :chat-behaviors (:chat-behaviors @db*)
     :chat-default-behavior (:chat-default-behavior @db*)
     :chat-welcome-message (:welcomeMessage (:chat config))}))
 
+(defn initialized [{:keys [db* messenger config]}]
+  (future
+    (f.tools/init-servers! db* messenger config)))
+
 (defn shutdown [{:keys [db*]}]
   (logger/logging-task
    :eca/shutdown

src/eca/server.clj

@@ -34,8 +34,8 @@
     (liveness-probe/start! parent-process-id log-wrapper-fn #(exit server)))
   (handlers/initialize (with-config components) params))
 
-(defmethod lsp.server/receive-notification "initialized" [_ _components _params]
-  (logger/info logger-tag "Initialized!"))
+(defmethod lsp.server/receive-notification "initialized" [_ components _params]
+  (handlers/initialized (with-config components)))
 
 (defmethod lsp.server/receive-request "shutdown" [_ components _params]
   (handlers/shutdown (with-config components)))