Skip to content

Commit 1777781

Browse files
committed
refactor(nvim): use fully qualified type names for subcommand
1 parent 3eacce5 commit 1777781

2 files changed

Lines changed: 9 additions & 8 deletions

File tree

  • lua
    • codecompanion/_extensions/vectorcode
    • vectorcode/integrations/codecompanion

lua/codecompanion/_extensions/vectorcode/init.lua

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11
---@module "codecompanion"
22

3-
---@alias sub_cmd "ls"|"query"|"vectorise"|"files_ls"|"files_rm"
3+
---@alias VectorCode.CodeCompanion.SubCommand "ls"|"query"|"vectorise"|"files_ls"|"files_rm"
44

55
---@class VectorCode.CodeCompanion.ExtensionOpts
66
---A table where the keys are the subcommand name (`ls`, `query`, `vectorise`, etc.)
77
--- and the values are their config options.
8-
---@field tool_opts? table<sub_cmd|"*", VectorCode.CodeCompanion.ToolOpts>
8+
---@field tool_opts? table<VectorCode.CodeCompanion.SubCommand|"*", VectorCode.CodeCompanion.ToolOpts>
99
---Options related to the `vectorcode_toolbox` tool group
1010
---@field tool_group? VectorCode.CodeCompanion.ToolGroupOpts
1111
---Prompt library that automatically creates VectorCode collections on local files
@@ -22,7 +22,7 @@ local utils = require("vectorcode.utils")
2222

2323
---@type VectorCode.CodeCompanion.ExtensionOpts|{}
2424
local default_extension_opts = {
25-
---@type table<sub_cmd, VectorCode.CodeCompanion.ToolOpts|{}>
25+
---@type table<VectorCode.CodeCompanion.SubCommand, VectorCode.CodeCompanion.ToolOpts|{}>
2626
tool_opts = {
2727
-- NOTE: the other default opts are defined in the source code files of the tools.
2828
-- `include_in_toolbox` is here so that the extension setup works as expected.
@@ -40,11 +40,11 @@ local default_extension_opts = {
4040
prompt_library = require("vectorcode.integrations.codecompanion.prompts.presets"),
4141
}
4242

43-
---@type sub_cmd[]
43+
---@type VectorCode.CodeCompanion.SubCommand[]
4444
local valid_tools = { "ls", "query", "vectorise", "files_ls", "files_rm" }
4545

46-
---@param tool_opts table<sub_cmd|"*", VectorCode.CodeCompanion.ToolOpts>
47-
---@return table<sub_cmd, VectorCode.CodeCompanion.ToolOpts>
46+
---@param tool_opts table<VectorCode.CodeCompanion.SubCommand|"*", VectorCode.CodeCompanion.ToolOpts>
47+
---@return table<VectorCode.CodeCompanion.SubCommand, VectorCode.CodeCompanion.ToolOpts>
4848
local function merge_tool_opts(tool_opts)
4949
local wildcard_opts = tool_opts["*"]
5050
if wildcard_opts then
@@ -55,7 +55,7 @@ local function merge_tool_opts(tool_opts)
5555
end
5656
tool_opts["*"] = nil
5757
end
58-
---@cast tool_opts table<sub_cmd, VectorCode.CodeCompanion.ToolOpts>
58+
---@cast tool_opts table<VectorCode.CodeCompanion.SubCommand, VectorCode.CodeCompanion.ToolOpts>
5959
return tool_opts
6060
end
6161

lua/vectorcode/integrations/codecompanion/init.lua

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
return {
44
chat = {
5-
---@param subcommand sub_cmd
5+
---@param subcommand VectorCode.CodeCompanion.SubCommand
66
---@param opts VectorCode.CodeCompanion.ToolOpts
77
---@return CodeCompanion.Tools.Tool
88
make_tool = function(subcommand, opts)
@@ -15,6 +15,7 @@ return {
1515
error("Unsupported version of codecompanion!")
1616
end
1717
end,
18+
1819
prompts = require("vectorcode.integrations.codecompanion.prompts"),
1920
},
2021
}

0 commit comments

Comments
 (0)