From 2991ca69a04ef5f92874ab001069fa00ae77c021 Mon Sep 17 00:00:00 2001 From: Omar Crespo Date: Wed, 19 Feb 2025 23:02:34 -0500 Subject: [PATCH] fix: include cursor-planning in the whitelist for dualboost (#1312) --- lua/avante/llm.lua | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/lua/avante/llm.lua b/lua/avante/llm.lua index f75d6b2..8a6bef8 100644 --- a/lua/avante/llm.lua +++ b/lua/avante/llm.lua @@ -470,7 +470,15 @@ M.stream = function(opts) return original_on_stop(stop_opts) end) end - if Config.dual_boost.enabled and opts.mode == "planning" then + + local valid_dual_boost_modes = { + planning = true, + ["cursor-planning"] = true, + } + + opts.mode = opts.mode or "planning" + + if Config.dual_boost.enabled and valid_dual_boost_modes[opts.mode] then M._dual_boost_stream( opts, Providers[Config.dual_boost.first_provider],