Support chat/queryFiles

Author: ericdallo

src/eca/features/chat.clj

@@ -926,9 +926,16 @@
    db*
    config]
   {:chat-id chat-id
-   :contexts (set/difference (set (f.context/all-contexts query db* config))
+   :contexts (set/difference (set (f.context/all-contexts query false db* config))
                              (set contexts))})
 
+(defn query-files
+  [{:keys [query chat-id]}
+   db*
+   config]
+  {:chat-id chat-id
+   :files (set (f.context/all-contexts query true db* config))})
+
 (defn query-commands
   [{:keys [query chat-id]}
    db*

src/eca/features/context.clj

@@ -101,7 +101,7 @@
       {:type "file"
        :path path})))
 
-(defn all-contexts [query db* config]
+(defn all-contexts [query files-only? db* config]
   (let [query (or (some-> query string/trim) "")
         first-project-path (shared/uri->filename (:uri (first (:workspace-folders @db*))))
         relative-path (and query
@@ -133,9 +133,12 @@
                                              :path (shared/uri->filename uri)})
                         (:workspace-folders @db*))
         mcp-resources (mapv #(assoc % :type "mcpResource") (f.mcp/all-resources @db*))]
-    (concat [{:type "repoMap"}
-             {:type "cursor"}]
-            root-dirs
-            relative-files
-            workspace-files
-            mcp-resources)))
+    (if files-only?
+      (concat relative-files
+              workspace-files)
+      (concat [{:type "repoMap"}
+               {:type "cursor"}]
+              root-dirs
+              relative-files
+              workspace-files
+              mcp-resources))))

src/eca/handlers.clj

@@ -101,9 +101,13 @@
   (metrics/task metrics :eca/chat-query-context
     (f.chat/query-context params db* config)))
 
+(defn chat-query-files [{:keys [db* config metrics]} params]
+  (metrics/task metrics :eca/chat-query-files
+    (f.chat/query-files params db* config)))
+
 (defn chat-query-commands [{:keys [db* config metrics]} params]
   (metrics/task metrics :eca/chat-query-commands
-    (f.chat/query-commands params db* config)))
+                (f.chat/query-commands params db* config)))
 
 (defn chat-tool-call-approve [{:keys [messenger db* metrics]} params]
   (metrics/task metrics :eca/chat-tool-call-approve

src/eca/server.clj

@@ -50,6 +50,9 @@
 (defmethod jsonrpc.server/receive-request "chat/queryContext" [_ components params]
   (handlers/chat-query-context (with-config components) params))
 
+(defmethod jsonrpc.server/receive-request "chat/queryFiles" [_ components params]
+  (handlers/chat-query-files (with-config components) params))
+
 (defmethod jsonrpc.server/receive-request "chat/queryCommands" [_ components params]
   (handlers/chat-query-commands (with-config components) params))