fix: types (yetone#208)

vdt · Aug 25, 2024 · ee88563 · ee88563
1 parent b494d05
commit ee88563
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 4 deletions.
diff --git a/lua/avante/config.lua b/lua/avante/config.lua
@@ -58,7 +58,7 @@ M.defaults = {
     max_tokens = 4096,
     ["local"] = false,
   },
-  ---@type AvanteGeminiProvider
+  ---@type AvanteSupportedProvider
   cohere = {
     endpoint = "https://api.cohere.com/v1",
     model = "command-r-plus",

diff --git a/lua/avante/providers/init.lua b/lua/avante/providers/init.lua
@@ -69,6 +69,7 @@ local Dressing = require("avante.ui.dressing")
 ---@field setup fun(): nil
 ---@field has fun(): boolean
 ---@field api_key_name string
+---@field model? string
 ---@field parse_api_key fun(): string | nil
 ---@field parse_stream_data? AvanteStreamParser
 ---

diff --git a/lua/avante/sidebar.lua b/lua/avante/sidebar.lua
@@ -31,11 +31,11 @@ local Sidebar = {}
 ---@field augroup integer
 ---@field code avante.CodeState
 ---@field winids table<string, integer> this table stores the winids of the sidebar components (result_container, result, selected_code_container, selected_code, input_container, input), even though they are destroyed.
----@field result_container NuiSplit | nil
+---@field result_container AvanteComp | nil
 ---@field result FloatingWindow | nil
----@field selected_code_container NuiSplit | nil
+---@field selected_code_container AvanteComp | nil
 ---@field selected_code FloatingWindow | nil
----@field input_container NuiSplit | nil
+---@field input_container AvanteComp | nil
 ---@field input FloatingWindow | nil
 
 ---@param id integer the tabpage id retrieved from api.nvim_get_current_tabpage()
@@ -416,7 +416,9 @@ function Sidebar:render_input_container()
 
   ---@type string
   local icon
+  ---@diagnostic disable-next-line: undefined-field
   if _G.MiniIcons ~= nil then
+    ---@diagnostic disable-next-line: undefined-global
     icon, _, _ = MiniIcons.get("filetype", filetype)
   else
     icon = require("nvim-web-devicons").get_icon_by_filetype(filetype, {})

diff --git a/lua/avante/types.lua b/lua/avante/types.lua
@@ -0,0 +1,26 @@
+---@meta
+
+---@class AvanteComp
+---@field winid integer | nil
+---@field bufnr integer | nil
+local AvanteComp = {}
+
+---@return nil
+function AvanteComp:mount() end
+
+---@return nil
+function AvanteComp:unmount() end
+
+---@param event string | string[]
+---@param handler string | function
+---@param options? table<"'once'" | "'nested'", boolean>
+---@return nil
+function AvanteComp:on(event, handler, options) end
+
+-- set keymap for this split
+---@param mode string check `:h :map-modes`
+---@param key string|string[] key for the mapping
+---@param handler string | fun(): nil handler for the mapping
+---@param opts? table<"'expr'"|"'noremap'"|"'nowait'"|"'remap'"|"'script'"|"'silent'"|"'unique'", boolean>
+---@return nil
+function AvanteComp:map(mode, key, handler, opts, ___force___) end