Skip to content

feat: add on_complete_callback to gp.Prompt #145

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Robitx merged 4 commits into from
Jul 17, 2024
Merged

feat: add on_complete_callback to gp.Prompt #145

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
796cdf1
feat: add on_complete_callback to query and Prompt
1orZero Jul 6, 2024
8bf9b25
Merge remote-tracking branch 'upstream/main' into feature/on_complete…
1orZero Jul 10, 2024
1834a41
fix: add back provider parameter to M.query function
1orZero Jul 10, 2024
336cbb9
Merge branch 'main' into feature/on_complete_callback
Robitx Jul 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 13 additions & 3 deletions lua/gp/init.lua
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1310,7 +1310,8 @@ end
---@param payload table # payload for api
---@param handler function # response handler
---@param on_exit function | nil # optional on_exit handler
M.query = function(buf, provider, payload, handler, on_exit)
---@param on_complete_callback function | nil # optional on_complete_callback handler
M.query = function(buf, provider, payload, handler, on_exit, on_complete_callback)
-- make sure handler is a function
if type(handler) ~= "function" then
M.error(
Expand Down Expand Up @@ -1430,6 +1431,13 @@ M.query = function(buf, provider, payload, handler, on_exit)
end)
end
end

-- optional on_complete_callback handler
if type(on_complete_callback) == "function" then
vim.schedule(function()
on_complete_callback(qt.response)
end)
end
end
end
end
Expand Down Expand Up @@ -2958,7 +2966,8 @@ end,
---@param template string # template with model instructions
---@param prompt string | nil # nil for non interactive commads
---@param whisper string | nil # predefined input (e.g. obtained from Whisper)
M.Prompt = function(params, target, agent, template, prompt, whisper)
---@param callback function | nil # callback after completing the prompt
M.Prompt = function(params, target, agent, template, prompt, whisper, callback)
if not agent or not type(agent) == "table" or not agent.provider then
M.warning(
"The `gp.Prompt` method signature has changed.\n"
Expand Down Expand Up @@ -3241,7 +3250,8 @@ M.Prompt = function(params, target, agent, template, prompt, whisper)
vim.schedule_wrap(function(qid)
on_exit(qid)
vim.cmd("doautocmd User GpDone")
end)
end),
on_complete_callback
)
end

Expand Down