wip

Author: olimorris

lua/codecompanion/config.lua

@@ -694,17 +694,17 @@ If you are providing code changes, use the insert_edit_into_file tool (if availa
       },
       opts = {
         context_management = {
-          trigger = 0.75, -- Compaction starts at 75% of the context window limit
-          enabled = function(adapter)
-            if adapter.type ~= "http" then
-              return false
-            end
-            -- Anthropic and OpenAI have their own server-side compaction
-            if adapter.vendor and (adapter.vendor == "anthropic" or adapter.vendor == "openai") then
-              return false
-            end
-            return true
-          end,
+          enabled = true, -- boolean, or function(adapter) -> boolean
+          editing = {
+            trigger = 0.65, -- 65% of the context window
+            keep = 5, -- last N tool results stay intact
+            exclude_tools = { "memory" }, -- tool names whose results never age out
+          },
+          compaction = {
+            trigger = 0.85, -- 85% of the context window
+            adapter = nil, -- nil | string | { name = string, model = string }
+            fallback_to_chat_adapter = false, -- on summary failure, retry with the chat adapter
+          },
         },
 
         blank_prompt = "", -- The prompt to use when the user doesn't provide a prompt
@@ -1354,6 +1354,24 @@ M.setup = function(args)
     M.config.interactions.chat.editor_context = nil
   end
 
+  -- TODO: Deprecate in v20.0.0 and remove in v21.0.0
+  -- Legacy `context_management.trigger` migrates to `context_management.compaction.trigger`
+  local context_management = args.interactions
+    and args.interactions.chat
+    and args.interactions.chat.opts
+    and args.interactions.chat.opts.context_management
+  if context_management and context_management.trigger ~= nil then
+    vim.notify(
+      "[CodeCompanion] `context_management.trigger` is deprecated. Use `context_management.compaction.trigger` instead.",
+      vim.log.levels.WARN,
+      { title = "CodeCompanion" }
+    )
+    if not (context_management.compaction and context_management.compaction.trigger ~= nil) then
+      M.config.interactions.chat.opts.context_management.compaction.trigger = context_management.trigger
+    end
+    M.config.interactions.chat.opts.context_management.trigger = nil
+  end
+
   M.config.interactions.chat.keymaps = remove_disabled_keymaps(M.config.interactions.chat.keymaps)
   M.config.interactions.cli.keymaps = remove_disabled_keymaps(M.config.interactions.cli.keymaps)
   M.config.interactions.inline.keymaps = remove_disabled_keymaps(M.config.interactions.inline.keymaps)

lua/codecompanion/interactions/chat/helpers/init.lua

@@ -345,28 +345,34 @@ function M.format_viewport_for_llm(buf_lines)
   return table.concat(formatted, "\n\n")
 end
 
----Returns the number of tokens that trigger context management
+---Returns the number of tokens that trigger context management for a given operation
 ---@param adapter CodeCompanion.HTTPAdapter
+---@param opts? { operation?: "editing"|"compaction" } defaults to "compaction"
 ---@return number
-function M.trigger_context_management(adapter)
-  if adapter.type ~= "http" then
+function M.trigger_context_management(adapter, opts)
+  opts = opts or {}
+  local operation = opts.operation or "compaction"
+
+  local context_management = config.interactions.chat.opts.context_management
+  local settings = context_management and context_management[operation]
+  local trigger = settings and settings.trigger
+  if trigger == nil then
     return 0
   end
 
-  local ok
-  local trigger_tokens = config.interactions.chat.opts.context_management.trigger
-  if trigger_tokens < 1 then
-    ok, trigger_tokens = pcall(function()
-      return math.floor(trigger_tokens * adapter.schema.model.choices[adapter.schema.model.default].meta.context_window)
+  if trigger < 1 then
+    local ok
+    ok, trigger = pcall(function()
+      return math.floor(trigger * adapter.schema.model.choices[adapter.schema.model.default].meta.context_window)
     end)
 
     if not ok then
-      log:error("Could not get evaluate the trigger for context management in the `%s` adapter", adapter.name)
+      log:error("Could not evaluate the %s trigger for context management in the `%s` adapter", operation, adapter.name)
       return 0
     end
   end
 
-  return trigger_tokens
+  return trigger
 end
 
 return M