1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
local _99 = require("99")
local M = {}
local function is_selectable_provider(provider)
return type(provider) == "table"
and type(provider._get_provider_name) == "function"
and type(provider._build_command) == "function"
end
--- @param provider _99.Providers.BaseProvider?
--- @param callback fun(models: string[], current: string): nil
function M.get_models(provider, callback)
provider = provider or _99.get_provider()
provider.fetch_models(function(models, err)
if err then
vim.notify("99: " .. err, vim.log.levels.ERROR)
return
end
if not models or #models == 0 then
vim.notify("99: No models available", vim.log.levels.WARN)
return
end
callback(models, _99.get_model())
end)
end
--- @return { names: string[], lookup: table<string, _99.Providers.BaseProvider>, current: string }
function M.get_providers()
local names = {}
local lookup = {}
for name, provider in pairs(_99.Providers) do
if is_selectable_provider(provider) then
table.insert(names, name)
lookup[name] = provider
end
end
table.sort(names)
local current = ""
local current_provider = _99.get_provider()
if is_selectable_provider(current_provider) then
current = current_provider._get_provider_name()
elseif #names > 0 then
current = names[1]
end
return {
names = names,
lookup = lookup,
current = current,
}
end
--- @param model string
function M.on_model_selected(model)
_99.set_model(model)
vim.notify("99: Model set to " .. model)
end
--- @param name string
--- @param lookup table<string, _99.Providers.BaseProvider>
function M.on_provider_selected(name, lookup)
local provider = lookup[name]
if not provider then
vim.notify(
"99: Invalid provider selection: " .. tostring(name),
vim.log.levels.ERROR
)
return
end
_99.set_provider(provider)
vim.notify(
"99: Provider set to " .. name .. " (model: " .. _99.get_model() .. ")"
)
end
return M
|
