CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: tool logs (#1190)

Browse Source
This commit is contained in:
yetone
2025-02-06 15:15:44 +08:00
committed by GitHub
parent 252a9e7818
commit 4f41154e83
3 changed files with 68 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lua/avante/llm.lua
View File

@@ -146,7 +146,7 @@ M._stream = function(opts)
on_chunk = opts.on_chunk,
on_stop = function(stop_opts)
if stop_opts.reason == "tool_use" and stop_opts.tool_use then
local result, error = LLMTools.process_tool_use(stop_opts.tool_use)
local result, error = LLMTools.process_tool_use(stop_opts.tool_use, opts.on_tool_log)
local tool_result = {
tool_use_id = stop_opts.tool_use.id,
content = error ~= nil and error or result,
@@ -424,10 +424,18 @@ end
---@field on_start AvanteLLMStartCallback
---@field on_chunk AvanteLLMChunkCallback
---@field on_stop AvanteLLMStopCallback
---@field on_tool_log? function(tool_name: string, log: string): nil
---@param opts StreamOptions
M.stream = function(opts)
local is_completed = false
if opts.on_tool_log ~= nil then
local original_on_tool_log = opts.on_tool_log
opts.on_tool_log = vim.schedule_wrap(function(tool_name, log)
if not original_on_tool_log then return end
return original_on_tool_log(tool_name, log)
end)
end
if opts.on_chunk ~= nil then
local original_on_chunk = opts.on_chunk
opts.on_chunk = vim.schedule_wrap(function(chunk)