34907fc1cd
The utils module has grown too big and contains unrelated functionality. Start moving code related to managing history messages comprising chat history into lua/avante/history module to keep the code more manageable.
102 lines
3.0 KiB
Lua
102 lines
3.0 KiB
Lua
local Utils = require("avante.utils")
|
|
|
|
local M = {}
|
|
|
|
---@param message avante.HistoryMessage
|
|
---@return boolean
|
|
function M.is_tool_use_message(message)
|
|
local content = message.message.content
|
|
if type(content) == "string" then return false end
|
|
if vim.islist(content) then
|
|
for _, item in ipairs(content) do
|
|
if item.type == "tool_use" then return true end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
---@param message avante.HistoryMessage
|
|
---@return boolean
|
|
function M.is_tool_result_message(message)
|
|
local content = message.message.content
|
|
if type(content) == "string" then return false end
|
|
if vim.islist(content) then
|
|
for _, item in ipairs(content) do
|
|
if item.type == "tool_result" then return true end
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
---@param message avante.HistoryMessage
|
|
---@param messages avante.HistoryMessage[]
|
|
---@return avante.HistoryMessage | nil
|
|
function M.get_tool_use_message(message, messages)
|
|
local content = message.message.content
|
|
if type(content) == "string" then return nil end
|
|
if vim.islist(content) then
|
|
local tool_id = nil
|
|
for _, item in ipairs(content) do
|
|
if item.type == "tool_result" then
|
|
tool_id = item.tool_use_id
|
|
break
|
|
end
|
|
end
|
|
if not tool_id then return nil end
|
|
for idx_ = #messages, 1, -1 do
|
|
local message_ = messages[idx_]
|
|
local content_ = message_.message.content
|
|
if type(content_) == "table" then
|
|
for _, item in ipairs(content_) do
|
|
if item.type == "tool_use" and item.id == tool_id then return message_ end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return nil
|
|
end
|
|
|
|
---@param tool_use_message avante.HistoryMessage | nil
|
|
function M.is_edit_func_call_message(tool_use_message)
|
|
local is_replace_func_call = false
|
|
local is_str_replace_editor_func_call = false
|
|
local is_str_replace_based_edit_tool_func_call = false
|
|
local path = nil
|
|
if tool_use_message and M.is_tool_use_message(tool_use_message) then
|
|
local tool_use = tool_use_message.message.content[1]
|
|
---@cast tool_use AvanteLLMToolUse
|
|
return Utils.is_edit_func_call_tool_use(tool_use)
|
|
end
|
|
return is_replace_func_call, is_str_replace_editor_func_call, is_str_replace_based_edit_tool_func_call, path
|
|
end
|
|
|
|
---@param message avante.HistoryMessage
|
|
---@param messages avante.HistoryMessage[]
|
|
---@return avante.HistoryMessage | nil
|
|
function M.get_tool_result_message(message, messages)
|
|
local content = message.message.content
|
|
if type(content) == "string" then return nil end
|
|
if vim.islist(content) then
|
|
local tool_id = nil
|
|
for _, item in ipairs(content) do
|
|
if item.type == "tool_use" then
|
|
tool_id = item.id
|
|
break
|
|
end
|
|
end
|
|
if not tool_id then return nil end
|
|
for idx_ = #messages, 1, -1 do
|
|
local message_ = messages[idx_]
|
|
local content_ = message_.message.content
|
|
if type(content_) == "table" then
|
|
for _, item in ipairs(content_) do
|
|
if item.type == "tool_result" and item.tool_use_id == tool_id then return message_ end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
return nil
|
|
end
|
|
|
|
return M
|