feat: add generic type for llm func (#1373)

2025-02-24 21:54:36 +08:00
parent fe496a9573
commit b627b335dd
3 changed files with 21 additions and 41 deletions
--- a/lua/avante/types.lua
+++ b/lua/avante/types.lua
@@ -323,7 +323,7 @@ vim.g.avante_login = vim.g.avante_login
 ---@field on_stop AvanteLLMStopCallback
 ---@field on_tool_log? function(tool_name: string, log: string): nil
 ---
---@alias AvanteLLMToolFunc fun(input: any, on_log?: (fun(log: string): nil) | nil): (boolean | string | nil, string | nil)
+---@alias AvanteLLMToolFunc<T> fun(input: T, on_log?: (fun(log: string): nil) | nil): (boolean | string | nil, string | nil)
 ---
 ---@class AvanteLLMTool
 ---@field name string