Skip to content

Commit f862c5f

Browse files
authored
feat(tools): include tool arguments in events (#2724)
1 parent 339bce8 commit f862c5f

File tree

1 file changed

+17
-3
lines changed

1 file changed

+17
-3
lines changed

lua/codecompanion/interactions/chat/tools/orchestrator.lua

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,11 @@ end
232232
---@return nil
233233
function Orchestrator:_finalize_tools()
234234
self.tools.tool = nil
235-
return utils.fire("ToolsFinished", { id = self.id, bufnr = self.tools.bufnr })
235+
return utils.fire("ToolsFinished", {
236+
bufnr = self.tools.bufnr,
237+
id = self.id,
238+
status = self.tools.status,
239+
})
236240
end
237241

238242
---Setup the tool to be executed
@@ -342,7 +346,12 @@ end
342346
---@param args { cmd: function, input?: any }
343347
---@return nil
344348
function Orchestrator:execute_tool(args)
345-
utils.fire("ToolStarted", { id = self.id, tool = self.tool.name, bufnr = self.tools.bufnr })
349+
utils.fire("ToolStarted", {
350+
bufnr = self.tools.bufnr,
351+
id = self.id,
352+
tool = self.tool.name,
353+
args = self.tool.args,
354+
})
346355
return Runner.new({ index = 1, orchestrator = self, cmd = args.cmd }):setup(args.input)
347356
end
348357

@@ -403,7 +412,12 @@ function Orchestrator:finalize_tool()
403412
pcall(function()
404413
self.handlers.on_exit()
405414
end)
406-
utils.fire("ToolFinished", { id = self.id, name = self.tool.name, bufnr = self.tools.bufnr })
415+
utils.fire("ToolFinished", {
416+
bufnr = self.tools.bufnr,
417+
id = self.id,
418+
name = self.tool.name,
419+
args = self.tool.args,
420+
})
407421
self.tool = nil
408422
end
409423
end

0 commit comments

Comments
 (0)