d0f0580d64
Ensure that chat history loaded from a file has resemblance of correct data. Namely title and timestamp are present and are strings, and entires, messages, and todos are lists. In case of inconsistencies replace with empty/default data. This should help with #2584. More changes are needed to sanitize individual entries.
67 lines
1.5 KiB
Lua
67 lines
1.5 KiB
Lua
local Base = require("avante.llm_tools.base")
|
|
|
|
---@class AvanteLLMTool
|
|
local M = setmetatable({}, Base)
|
|
|
|
M.name = "update_todo_status"
|
|
|
|
M.description = "Update the status of TODO"
|
|
|
|
---@type AvanteLLMToolParam
|
|
M.param = {
|
|
type = "table",
|
|
fields = {
|
|
{
|
|
name = "id",
|
|
description = "The ID of the TODO to update",
|
|
type = "string",
|
|
},
|
|
{
|
|
name = "status",
|
|
description = "The status of the TODO to update",
|
|
type = "string",
|
|
choices = { "todo", "doing", "done", "cancelled" },
|
|
},
|
|
},
|
|
}
|
|
|
|
---@type AvanteLLMToolReturn[]
|
|
M.returns = {
|
|
{
|
|
name = "success",
|
|
description = "Whether the TODO was updated successfully",
|
|
type = "boolean",
|
|
},
|
|
{
|
|
name = "error",
|
|
description = "Error message if the TODOs could not be updated",
|
|
type = "string",
|
|
optional = true,
|
|
},
|
|
}
|
|
|
|
M.on_render = function() return {} end
|
|
|
|
---@type AvanteLLMToolFunc<{ id: string, status: string }>
|
|
function M.func(input, opts)
|
|
local on_complete = opts.on_complete
|
|
local sidebar = require("avante").get()
|
|
if not sidebar then return false, "Avante sidebar not found" end
|
|
local todos = sidebar.chat_history.todos
|
|
if #todos == 0 then return false, "No todos found" end
|
|
for _, todo in ipairs(todos) do
|
|
if tostring(todo.id) == tostring(input.id) then
|
|
todo.status = input.status
|
|
break
|
|
end
|
|
end
|
|
sidebar:update_todos(todos)
|
|
if on_complete then
|
|
on_complete(true, nil)
|
|
return nil, nil
|
|
end
|
|
return true, nil
|
|
end
|
|
|
|
return M
|