phyer/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: support deepseek (#69)

Browse Source 
The DeepSeek API reduces costs through disk caching, and DeepSeek-Coder-V2-0724 achieves GPT-4-Turbo-0409 level code capabilities with excellent math and reasoning skills.
This commit is contained in:
Chao Li 2024-08-18 21:33:45 +08:00 committed by GitHub
parent b979b3d8ce
commit 6bef72e287
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -81,8 +81,8 @@ _See [config.lua#L9](./lua/avante/config.lua) for the full config_
```lua ```lua
{ {
---@alias Provider "openai" | "claude" | "azure" ---@alias Provider "openai" | "claude" | "azure" | "deepseek"
provider = "claude", -- "claude" or "openai" or "azure" provider = "claude", -- "claude" or "openai" or "azure" or "deepseek"
openai = { openai = {
endpoint = "https://api.openai.com", endpoint = "https://api.openai.com",
model = "gpt-4o", model = "gpt-4o",
@ -167,6 +167,12 @@ Given its early stage, `avante.nvim` currently supports the following basic func
> ```sh > ```sh
> export AZURE_OPENAI_API_KEY=your-api-key > export AZURE_OPENAI_API_KEY=your-api-key
> ``` > ```
>
> For DeepSeek
>
> ```sh
> export DEEPSEEK_API_KEY=you-api-key
> ```
1. Open a code file in Neovim. 1. Open a code file in Neovim.
2. Use the `:AvanteAsk` command to query the AI about the code. 2. Use the `:AvanteAsk` command to query the AI about the code.

20
lua/avante/ai_bot.lua
View File

@ -17,6 +17,7 @@ local E = {
openai = "OPENAI_API_KEY", openai = "OPENAI_API_KEY",
claude = "ANTHROPIC_API_KEY", claude = "ANTHROPIC_API_KEY",
azure = "AZURE_OPENAI_API_KEY", azure = "AZURE_OPENAI_API_KEY",
deepseek = "DEEPSEEK_API_KEY",
}, },
_once = false, _once = false,
} }
@ -298,6 +299,23 @@ local function call_openai_api_stream(question, code_lang, code_content, selecte
max_tokens = Config.azure.max_tokens, max_tokens = Config.azure.max_tokens,
stream = true, stream = true,
} }
elseif Config.provider == "deepseek" then
api_key = os.getenv(E.key("deepseek"))
url = Utils.trim_suffix(Config.deepseek.endpoint, "/") .. "/chat/completions"
headers = {
["Content-Type"] = "application/json",
["Authorization"] = "Bearer " .. api_key,
}
body = {
model = Config.deepseek.model,
messages = {
{ role = "system", content = system_prompt },
{ role = "user", content = user_prompt },
},
temperature = Config.deepseek.temperature,
max_tokens = Config.deepseek.max_tokens,
stream = true,
}
else else
url = Utils.trim_suffix(Config.openai.endpoint, "/") .. "/v1/chat/completions" url = Utils.trim_suffix(Config.openai.endpoint, "/") .. "/v1/chat/completions"
headers = { headers = {
@ -364,7 +382,7 @@ end
---@param on_chunk fun(chunk: string): any ---@param on_chunk fun(chunk: string): any
---@param on_complete fun(err: string|nil): any ---@param on_complete fun(err: string|nil): any
function M.call_ai_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete) function M.call_ai_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete)
if Config.provider == "openai" or Config.provider == "azure" then if Config.provider == "openai" or Config.provider == "azure" or Config.provider == "deepseek" then
call_openai_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete) call_openai_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete)
elseif Config.provider == "claude" then elseif Config.provider == "claude" then
call_claude_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete) call_claude_api_stream(question, code_lang, code_content, selected_content_content, on_chunk, on_complete)

10
lua/avante/config.lua
View File

@ -6,8 +6,8 @@ local M = {}
---@class avante.Config ---@class avante.Config
M.defaults = { M.defaults = {
---@alias Provider "openai" | "claude" | "azure" ---@alias Provider "openai" | "claude" | "azure" | "deepseek"
provider = "claude", -- "claude" or "openai" or "azure" provider = "claude", -- "claude" or "openai" or "azure" or "deepseek"
openai = { openai = {
endpoint = "https://api.openai.com", endpoint = "https://api.openai.com",
model = "gpt-4o", model = "gpt-4o",
@ -27,6 +27,12 @@ M.defaults = {
temperature = 0, temperature = 0,
max_tokens = 4096, max_tokens = 4096,
}, },
deepseek = {
endpoint = "https://api.deepseek.com",
model = "deepseek-coder",
temperature = 0,
max_tokens = 4096,
},
behaviour = { behaviour = {
auto_apply_diff_after_generation = false, -- Whether to automatically apply diff after LLM response. auto_apply_diff_after_generation = false, -- Whether to automatically apply diff after LLM response.
}, },