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

fix: openrouter always returns 200 regardless of the situation (#1586)

Browse Source
This commit is contained in:
yetone
2025-03-14 19:29:16 +08:00
committed by GitHub
parent cd50140cb4
commit f60f150a21
1 changed files with 19 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
lua/avante/llm.lua
View File

@@ -355,6 +355,15 @@ function M.curl(opts)
callback = function(result) callback = function(result)
active_job = nil active_job = nil
cleanup() cleanup()
local headers_map = vim.iter(result.headers):fold({}, function(acc, value)
local pieces = vim.split(value, ":")
local key = pieces[1]
local remain = vim.list_slice(pieces, 2)
if not remain then return acc end
local val = Utils.trim_spaces(table.concat(remain, ":"))
acc[key] = val
return acc
end)
if result.status >= 400 then if result.status >= 400 then
if provider.on_error then if provider.on_error then
provider.on_error(result) provider.on_error(result)
@@ -362,15 +371,6 @@ function M.curl(opts)
Utils.error("API request failed with status " .. result.status, { once = true, title = "Avante" }) Utils.error("API request failed with status " .. result.status, { once = true, title = "Avante" })
end end
if result.status == 429 then if result.status == 429 then
local headers_map = vim.iter(result.headers):fold({}, function(acc, value)
local pieces = vim.split(value, ":")
local key = pieces[1]
local remain = vim.list_slice(pieces, 2)
if not remain then return acc end
local val = Utils.trim_spaces(table.concat(remain, ":"))
acc[key] = val
return acc
end)
local retry_after = 10 local retry_after = 10
if headers_map["retry-after"] then retry_after = tonumber(headers_map["retry-after"]) or 10 end if headers_map["retry-after"] then retry_after = tonumber(headers_map["retry-after"]) or 10 end
handler_opts.on_stop({ reason = "rate_limit", retry_after = retry_after }) handler_opts.on_stop({ reason = "rate_limit", retry_after = retry_after })
@@ -394,6 +394,16 @@ function M.curl(opts)
parse_response_without_stream(result.body) parse_response_without_stream(result.body)
end) end)
end end
if result.status == 200 and spec.url:match("https://openrouter.ai") then
local content_type = headers_map["content-type"]
if content_type and content_type:match("text/html") then
handler_opts.on_stop({
reason = "error",
error = "Your openrouter endpoint setting is incorrect, please set it to https://openrouter.ai/api/v1",
})
end
end
end, end,
}) })