Merge pull request #113 from brunocroh/main

feat: add support to deepseek as a provider

Author: frankroeder

lua/parrot/config.lua

@@ -172,6 +172,19 @@ local defaults = {
         command = { temperature = 1.1, top_p = 1 },
       },
     },
+    deepseek = {
+      api_key = "",
+      endpoint = "https://api.deepseek.com/chat/completions",
+      topic_prompt = topic_prompt,
+      topic = {
+        model = "deepseek-chat",
+        params = { max_tokens = 64 },
+      },
+      params = {
+        chat = { temperature = 1.1, top_p = 1 },
+        command = { temperature = 1.1, top_p = 1 },
+      },
+    },
     custom = {
       style = "openai",
       api_key = "",

lua/parrot/provider/deepseek.lua

@@ -0,0 +1,83 @@
+local OpenAI = require("parrot.provider.openai")
+local utils = require("parrot.utils")
+local Job = require("plenary.job")
+
+local DeepSeek = setmetatable({}, { __index = OpenAI })
+DeepSeek.__index = DeepSeek
+
+-- Available API parameters for DeepSeek
+-- https://api-docs.deepseek.com/api/deepseek-api
+local AVAILABLE_API_PARAMETERS = {
+  -- required
+  messages = true,
+  model = true,
+  -- optional
+  frequency_penalty = true,
+  max_tokens = true,
+  presence_penalty = true,
+  response_format = true,
+  seed = true,
+  stop = true,
+  stream = true,
+  stream_options = true,
+  temperature = true,
+  tool_choice = true,
+  tools = true,
+  logprobs = true,
+  top_logprobs = true,
+  top_p = true,
+  user = true,
+}
+
+function DeepSeek:new(endpoint, api_key)
+  local instance = OpenAI.new(self, endpoint, api_key)
+  instance.name = "deepseek"
+  return setmetatable(instance, self)
+end
+
+-- Preprocesses the payload before sending to the API
+---@param payload table
+---@return table
+function DeepSeek:preprocess_payload(payload)
+  for _, message in ipairs(payload.messages) do
+    message.content = message.content:gsub("^%s*(.-)%s*$", "%1")
+  end
+  return utils.filter_payload_parameters(AVAILABLE_API_PARAMETERS, payload)
+end
+
+-- Returns the list of available models
+---@param online boolean
+---@return string[]
+function DeepSeek:get_available_models(online)
+  local ids = {
+    "deepseek-chat",
+    "deepseek-reasoner",
+  }
+  if online and self:verify() then
+    local job = Job:new({
+      command = "curl",
+      args = {
+        "https://api.deepseek.com/models",
+        "-H",
+        "Authorization: Bearer " .. self.api_key,
+      },
+      on_exit = function(job)
+        local parsed_response = utils.parse_raw_response(job:result())
+        self:process_onexit(parsed_response)
+        ids = {}
+        local success, decoded = pcall(vim.json.decode, parsed_response)
+        if success and decoded.models then
+          for _, item in ipairs(decoded.models) do
+            table.insert(ids, item.id)
+          end
+        end
+        return ids
+      end,
+    })
+    job:start()
+    job:wait()
+  end
+  return ids
+end
+
+return DeepSeek

lua/parrot/provider/init.lua

@@ -8,6 +8,7 @@ local OpenAI = require("parrot.provider.openai")
 local Perplexity = require("parrot.provider.perplexity")
 local GitHub = require("parrot.provider.github")
 local xAI = require("parrot.provider.xai")
+local DeepSeek = require("parrot.provider.deepseek")
 local logger = require("parrot.logger")
 
 local M = {}
@@ -28,6 +29,7 @@ M.init_provider = function(prov_name, endpoint, api_key, style, models)
     openai = OpenAI,
     pplx = Perplexity,
     xai = xAI,
+    deepseek = DeepSeek,
   }
 
   if providers[prov_name] then