From 8da09eae1d496c700df951c9142bfd744b544d78 Mon Sep 17 00:00:00 2001 From: Daryl <23520617+PlatyPew@users.noreply.github.com> Date: Fri, 27 Jun 2025 15:27:40 +0800 Subject: [PATCH] fix: forbidden inputs in mistral ai (#2348) --- lua/avante/providers/openai.lua | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/lua/avante/providers/openai.lua b/lua/avante/providers/openai.lua index c894805..224c5bb 100644 --- a/lua/avante/providers/openai.lua +++ b/lua/avante/providers/openai.lua @@ -49,6 +49,8 @@ end function M.is_openrouter(url) return url:match("^https://openrouter%.ai/") end +function M.is_mistral(url) return url:match("^https://api%.mistral%.ai/") end + ---@param opts AvantePromptOptions function M.get_user_message(opts) vim.deprecate("get_user_message", "parse_messages", "0.1.0", "avante.nvim") @@ -557,9 +559,9 @@ function M:parse_curl_args(prompt_opts) model = provider_conf.model, messages = self:parse_messages(prompt_opts), stream = true, - stream_options = { + stream_options = not M.is_mistral(provider_conf.endpoint) and { include_usage = true, - }, + } or nil, tools = tools, }, request_body), }