Feature: add Tabby support (#24)

* feat: add tabby backend

* docs: add Tabby docs

Author: chmanie

README.md

@@ -169,6 +169,36 @@ cmp_ai:setup({
 })
 ```
 
+To use Tabby:
+
+```lua
+local cmp_ai = require('cmp_ai.config')
+
+cmp_ai:setup({
+  max_lines = 1000,
+  provider = 'Tabby',
+  notify = true,
+  provider_options = {
+    -- These are optional
+    -- user = 'yourusername',
+    -- temperature = 0.2,
+    -- seed = 'randomstring',
+  },
+  notify_callback = function(msg)
+    vim.notify(msg)
+  end,
+  run_on_every_keystroke = true,
+  ignored_file_types = {
+    -- default is not to ignore
+    -- uncomment to ignore in lua:
+    -- lua = true
+  },
+})
+```
+
+You will also need to make sure you have the Tabby api key in your environment, `TABBY_API_KEY`.
+
+
 ### `notify`
 
 As some completion sources can be quit slow, setting this to `true` will trigger

doc/cmp-ai.txt

@@ -189,6 +189,37 @@ To use Ollama <https://ollama.ai>:
     })
 <
 
+To use Tabby:
+
+>lua
+    local cmp_ai = require('cmp_ai.config')
+    
+    cmp_ai:setup({
+      max_lines = 1000,
+      provider = 'Tabby',
+      provider_options = {
+	-- These are optional
+	-- user = 'yourusername',
+	-- temperature = 0.2,
+	-- seed = 'randomstring',
+      },
+      notify = true,
+      notify_callback = function(msg)
+        vim.notify(msg)
+      end,
+      run_on_every_keystroke = true,
+      ignored_file_types = {
+        -- default is not to ignore
+        -- uncomment to ignore in lua:
+        -- lua = true
+      },
+    })
+<
+
+
+You will also need to make sure you have the Tabby api key in
+your environment, `TABBY_API_KEY`.
+
 
 NOTIFY ~
 

lua/cmp_ai/backends/tabby.lua

@@ -0,0 +1,62 @@
+local requests = require('cmp_ai.requests')
+
+Tabby = requests:new(nil)
+
+function Tabby:new(o)
+  o = o or {}
+  setmetatable(o, self)
+  self.__index = self
+  self.params = vim.tbl_deep_extend('keep', o or {}, {
+    base_url = 'http://127.0.0.1:8080/v1/completions',
+    options = {},
+  })
+
+  self.api_key = os.getenv('TABBY_API_KEY')
+
+  return o
+end
+
+function Tabby:complete(lines_before, lines_after, cb)
+  if not self.api_key then
+    vim.schedule(function()
+      vim.notify('TABBY_API_KEY environment variable not set', vim.log.levels.ERROR)
+    end)
+    return
+  end
+
+  local headers = {
+    'Authorization: Bearer ' .. self.api_key,
+  }
+
+  local data = {
+    language = vim.bo.filetype,
+    segments = {
+      prefix = lines_before,
+      suffix = lines_after,
+      filepath = vim.api.nvim_buf_get_name(0),
+    },
+    user = self.params.options.user,
+    temperature = self.params.options.temperature,
+    seed = self.params.options.seed,
+  }
+
+  self:Get(self.params.base_url, headers, data, function(answer)
+    local new_data = {}
+    if answer.error ~= nil then
+      vim.notify('Tabby error: ' .. answer.error)
+      return
+    end
+    if answer.choices ~= nil and answer.choices[1] ~= nil then
+      table.insert(new_data, answer.choices[1].text)
+    end
+    cb(new_data)
+  end)
+end
+
+function Tabby:test()
+  self:complete('def factorial(n)\n    if', '    return ans\n', function(data)
+    dump(data)
+  end)
+end
+
+return Tabby