57 lines
1.6 KiB
Lua
57 lines
1.6 KiB
Lua
local Utils = require("avante.utils")
|
|
|
|
---@class avante.utils.history
|
|
local M = {}
|
|
|
|
---@param entries avante.ChatHistoryEntry[]
|
|
---@return avante.ChatHistoryEntry[]
|
|
function M.filter_active_entries(entries)
|
|
local entries_ = {}
|
|
|
|
for i = #entries, 1, -1 do
|
|
local entry = entries[i]
|
|
if entry.reset_memory then break end
|
|
if
|
|
entry.request == nil
|
|
or entry.original_response == nil
|
|
or entry.request == ""
|
|
or entry.original_response == ""
|
|
then
|
|
break
|
|
end
|
|
table.insert(entries_, 1, entry)
|
|
end
|
|
|
|
return entries_
|
|
end
|
|
|
|
---@param entries avante.ChatHistoryEntry[]
|
|
---@return AvanteLLMMessage[]
|
|
function M.entries_to_llm_messages(entries)
|
|
local messages = {}
|
|
for _, entry in ipairs(entries) do
|
|
local user_content = ""
|
|
if entry.selected_filepaths ~= nil then
|
|
user_content = user_content .. "SELECTED FILES:\n\n"
|
|
for _, filepath in ipairs(entry.selected_filepaths) do
|
|
user_content = user_content .. filepath .. "\n"
|
|
end
|
|
end
|
|
if entry.selected_code ~= nil then
|
|
user_content = user_content
|
|
.. "SELECTED CODE:\n\n```"
|
|
.. (entry.selected_code.file_type or "")
|
|
.. (entry.selected_code.path and ":" .. entry.selected_code.path or "")
|
|
.. "\n"
|
|
.. entry.selected_code.content
|
|
.. "\n```\n\n"
|
|
end
|
|
user_content = user_content .. "USER PROMPT:\n\n" .. entry.request
|
|
table.insert(messages, { role = "user", content = user_content })
|
|
table.insert(messages, { role = "assistant", content = Utils.trim_think_content(entry.original_response) })
|
|
end
|
|
return messages
|
|
end
|
|
|
|
return M
|