add request log

This commit is contained in:
zhangkun9038@dingtalk.com 2025-02-09 22:17:25 +08:00
parent be74f25b82
commit 07abe12f54
3 changed files with 45 additions and 1 deletions

View File

@ -130,6 +130,13 @@ M._defaults = {
temperature = 0,
max_tokens = 4096,
},
baidu = {
endpoint = "https://dashscope.aliyuncs.com/compatible-mode/v1/",
model = "deepseek-v3",
timeout = 30000, -- Timeout in milliseconds
temperature = 0,
max_tokens = 4096,
},
---To add support for custom provider, follow the format below
---See https://github.com/yetone/avante.nvim/wiki#custom-providers for more details
---@type {[string]: AvanteProvider}

View File

@ -39,6 +39,8 @@ M.parse_response = function(ctx, data_stream, _, opts)
end
end
local Log = require("avante.utils.log")
M.parse_curl_args = function(provider, prompt_opts)
local base, body_opts = P.parse_config(provider)
@ -58,7 +60,7 @@ M.parse_curl_args = function(provider, prompt_opts)
headers["Authorization"] = "Bearer " .. api_key
end
return {
local request = {
url = Utils.url_join(base.endpoint, "/chat/completions"),
proxy = base.proxy,
insecure = base.allow_insecure,
@ -70,6 +72,11 @@ M.parse_curl_args = function(provider, prompt_opts)
enable_citation = base.enable_citation,
}, body_opts),
}
-- Log the request details
Log.log_request(request.url, request.headers, request.body)
return request
end
return M

30
lua/avante/utils/log.lua Normal file
View File

@ -0,0 +1,30 @@
local M = {}
local log_file = vim.fn.stdpath("cache") .. "/avante_requests.log"
function M.log_request(url, headers, body)
local timestamp = os.date("%Y-%m-%d %H:%M:%S")
local log_entry = string.format(
[[
[%s] Request Details:
URL: %s
Headers:
%s
Body:
%s
]],
timestamp,
url,
vim.inspect(headers),
vim.inspect(body)
)
-- Append to log file
local file = io.open(log_file, "a")
if file then
file:write(log_entry .. "\n\n")
file:close()
end
end
return M