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

fix: gemini generation config (#2133)

Browse Source
This commit is contained in:
yetone
2025-06-04 01:57:25 +08:00
committed by GitHub
parent 55acc44003
commit 47fa08ee5b
2 changed files with 7 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lua/avante/config.lua
View File

@@ -281,8 +281,9 @@ M._defaults = {
timeout = 30000, -- Timeout in milliseconds timeout = 30000, -- Timeout in milliseconds
use_ReAct_prompt = true, use_ReAct_prompt = true,
extra_request_body = { extra_request_body = {
temperature = 0.75, generationConfig = {
max_tokens = 65536, temperature = 0.75,
},
}, },
}, },
---@type AvanteSupportedProvider ---@type AvanteSupportedProvider

14
lua/avante/providers/gemini.lua
View File

@@ -181,17 +181,11 @@ end
---@param provider_instance AvanteProviderFunctor The provider instance (self). ---@param provider_instance AvanteProviderFunctor The provider instance (self).
---@param prompt_opts AvantePromptOptions Prompt options including messages, tools, system_prompt. ---@param prompt_opts AvantePromptOptions Prompt options including messages, tools, system_prompt.
---@param provider_conf table Provider configuration from config.lua (e.g., model, top-level temperature/max_tokens). ---@param provider_conf table Provider configuration from config.lua (e.g., model, top-level temperature/max_tokens).
---@param request_body table Request-specific overrides, typically from provider_conf.request_config_overrides. ---@param request_body_ table Request-specific overrides, typically from provider_conf.request_config_overrides.
---@return table The fully constructed request body. ---@return table The fully constructed request body.
function M.prepare_request_body(provider_instance, prompt_opts, provider_conf, request_body) function M.prepare_request_body(provider_instance, prompt_opts, provider_conf, request_body_)
request_body = vim.tbl_deep_extend("force", request_body, { local request_body = {}
generationConfig = { request_body.generationConfig = request_body_.generationConfig or {}
temperature = request_body.temperature,
maxOutputTokens = request_body.max_tokens,
},
})
request_body.temperature = nil
request_body.max_tokens = nil
local use_ReAct_prompt = provider_conf.use_ReAct_prompt == true local use_ReAct_prompt = provider_conf.use_ReAct_prompt == true