fix(cmd): strategy now respects vim.g.codecompanion_adapter

olimorris · Dec 9, 2024 · b629005 · b629005
1 parent 02c1bc0
commit b629005
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/lua/codecompanion/strategies/chat/init.lua b/lua/codecompanion/strategies/chat/init.lua
@@ -634,8 +634,7 @@ function Chat:submit(opts)
 
   self:apply_tools_and_variables(message)
 
-  -- Check if the user has manually overriden the adapter. This is useful if the
-  -- user loses their internet connection and wants to switch to a local LLM
+  -- Check if the user has manually overriden the adapter
   if vim.g.codecompanion_adapter and self.adapter.name ~= vim.g.codecompanion_adapter then
     self.adapter = adapters.resolve(config.adapters[vim.g.codecompanion_adapter])
   end

diff --git a/lua/codecompanion/strategies/cmd.lua b/lua/codecompanion/strategies/cmd.lua
@@ -20,6 +20,11 @@ function Cmd.new(args)
     return log:error("No adapter found")
   end
 
+  -- Check if the user has manually overriden the adapter
+  if vim.g.codecompanion_adapter and self.adapter.name ~= vim.g.codecompanion_adapter then
+    self.adapter = adapters.resolve(config.adapters[vim.g.codecompanion_adapter])
+  end
+
   return self
 end
 

diff --git a/lua/codecompanion/strategies/inline.lua b/lua/codecompanion/strategies/inline.lua
@@ -135,8 +135,7 @@ function Inline.new(args)
     return log:error("No adapter found")
   end
 
-  -- Check if the user has manually overriden the adapter. This is useful if the
-  -- user loses their internet connection and wants to switch to a local LLM
+  -- Check if the user has manually overriden the adapter
   if vim.g.codecompanion_adapter and self.adapter.name ~= vim.g.codecompanion_adapter then
     self.adapter = adapters.resolve(config.adapters[vim.g.codecompanion_adapter])
   end