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

feat: write curl body to a temporary JSON file (#673)

Browse Source 
* Write curl body to a temporary JSON file.

* Remove temp file after use

* feat: use vim.fn.delete to delete files

---------

Co-authored-by: yetone <yetoneful@gmail.com>
This commit is contained in:
Huite 2024-10-07 17:53:18 +02:00 committed by GitHub
parent c3848d086c
commit 2a72dfa711
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lua/avante/llm.lua
View File

@ -1,4 +1,5 @@
local api = vim.api local api = vim.api
local fn = vim.fn
local curl = require("plenary.curl") local curl = require("plenary.curl")
@ -101,8 +102,6 @@ M.stream = function(opts)
---@type AvanteCurlOutput ---@type AvanteCurlOutput
local spec = Provider.parse_curl_args(Provider, code_opts) local spec = Provider.parse_curl_args(Provider, code_opts)
Utils.debug("curl spec:", spec)
---@param line string ---@param line string
local function parse_stream_data(line) local function parse_stream_data(line)
local event = line:match("^event: (.+)$") local event = line:match("^event: (.+)$")
@ -122,11 +121,22 @@ M.stream = function(opts)
local active_job local active_job
local curl_body_file = fn.tempname() .. ".json"
local json_content = vim.json.encode(spec.body)
fn.writefile(vim.split(json_content, "\n"), curl_body_file)
Utils.debug("curl body file:", curl_body_file)
local function cleanup()
if Config.debug then return end
vim.schedule(function() fn.delete(curl_body_file) end)
end
active_job = curl.post(spec.url, { active_job = curl.post(spec.url, {
headers = spec.headers, headers = spec.headers,
proxy = spec.proxy, proxy = spec.proxy,
insecure = spec.insecure, insecure = spec.insecure,
body = vim.json.encode(spec.body), body = curl_body_file,
stream = function(err, data, _) stream = function(err, data, _)
if err then if err then
completed = true completed = true
@ -155,10 +165,12 @@ M.stream = function(opts)
on_error = function() on_error = function()
active_job = nil active_job = nil
completed = true completed = true
cleanup()
opts.on_complete(nil) opts.on_complete(nil)
end, end,
callback = function(result) callback = function(result)
active_job = nil active_job = nil
cleanup()
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)