feat(nvim): show vectorise stats for the vectorise tool

Author: Zhe Yu

lua/vectorcode/integrations/codecompanion/ls_tool.lua

@@ -17,7 +17,7 @@ return function(opts)
     name = tool_name,
     cmds = {
       ---@param agent CodeCompanion.Agent
-      ---@return nil|{ status: string, msg: string }
+      ---@return nil|{ status: string, data: string }
       function(agent, _, _, cb)
         job_runner.run_async({ "ls", "--pipe" }, function(result, error)
           if vim.islist(result) and #result > 0 then

lua/vectorcode/integrations/codecompanion/query_tool.lua

@@ -37,7 +37,7 @@ return check_cli_wrap(function(opts)
     cmds = {
       ---@param agent CodeCompanion.Agent
       ---@param action QueryToolArgs
-      ---@return nil|{ status: string, msg: string }
+      ---@return nil|{ status: string, data: string }
       function(agent, action, _, cb)
         logger.info(
           "CodeCompanion query tool called with the following arguments:\n",

lua/vectorcode/integrations/codecompanion/vectorise_tool.lua

@@ -47,7 +47,7 @@ return function(opts)
     cmds = {
       ---@param agent CodeCompanion.Agent
       ---@param action VectoriseToolArgs
-      ---@return nil|{ status: string, msg: string }
+      ---@return nil|{ status: string, data: string }
       function(agent, action, _, cb)
         local args = { "vectorise", "--pipe" }
         local project_root = action.project_root or ""
@@ -78,7 +78,6 @@ return function(opts)
           args,
           ---@param result VectoriseResult
           function(result, error, code, _)
-            vim.schedule_wrap(vim.notify)(vim.inspect(result))
             if result then
               cb({ status = "success", data = result })
             else
@@ -89,5 +88,30 @@ return function(opts)
         )
       end,
     },
+    output = {
+      ---@param self CodeCompanion.Agent.Tool
+      ---@param agent CodeCompanion.Agent
+      ---@param stdout VectorCode.VectoriseResult[]
+      success = function(self, agent, _, stdout)
+        stdout = stdout[1]
+        agent.chat:add_tool_output(
+          self,
+          string.format(
+            [[**VectorCode Vectorise Tool**:
+  - Added: %d
+  - Updated: %d
+  - Removed: %d
+  - Skipped: %d
+  - Failed: %d
+  ]],
+            stdout.add,
+            stdout.update,
+            stdout.removed,
+            stdout.skipped,
+            stdout.failed
+          )
+        )
+      end,
+    },
   }
 end

lua/vectorcode/types.lua

@@ -9,6 +9,13 @@
 ---@class VectorCode.LsResult
 ---@field project-root string
 
+---@class VectorCode.VectoriseResult
+---@field add integer
+---@field update integer
+---@field removed integer
+---@field skipped integer
+---@field failed integer
+
 ---Type definitions for the cache of a buffer.
 ---@class VectorCode.Cache
 ---@field enabled boolean Whether the async jobs are enabled or not. If the buffer is disabled, no cache will be generated for it.