Merge pull request #386 from Opencode-DCP/cleanup/context-limit-hints

Cleanup/context limit hints

Author: Tarquinen

lib/commands/sweep.ts

@@ -128,7 +128,7 @@ export async function handleSweepCommand(ctx: SweepCommandContext): Promise<void
     const protectedTools = config.commands.protectedTools
 
     syncToolCache(state, config, logger, messages)
-    buildToolIdList(state, messages, logger)
+    buildToolIdList(state, messages)
 
     // Parse optional numeric argument
     const numArg = args[0] ? parseInt(args[0], 10) : null

lib/hooks.ts

@@ -4,7 +4,7 @@ import type { PluginConfig } from "./config"
 import { syncToolCache } from "./state/tool-cache"
 import { deduplicate, supersedeWrites, purgeErrors } from "./strategies"
 import { prune, insertCompressToolContext } from "./messages"
-import { buildToolIdList, isIgnoredUserMessage, annotateContext } from "./messages/utils"
+import { buildToolIdList, isIgnoredUserMessage } from "./messages/utils"
 import { checkSession } from "./state"
 import { renderSystemPrompt } from "./prompts"
 import { handleStatsCommand } from "./commands/stats"
@@ -108,7 +108,7 @@ export function createChatMessageTransformHandler(
         }
 
         syncToolCache(state, config, logger, output.messages)
-        buildToolIdList(state, output.messages, logger)
+        buildToolIdList(state, output.messages)
 
         deduplicate(state, logger, config, output.messages)
         supersedeWrites(state, logger, config, output.messages)
@@ -118,7 +118,6 @@ export function createChatMessageTransformHandler(
 
         insertCompressToolContext(state, config, logger, output.messages)
 
-        annotateContext(output.messages)
         applyPendingManualTriggerPrompt(state, output.messages, logger)
 
         if (state.sessionId) {

lib/messages/index.ts

@@ -1,2 +1,2 @@
 export { prune } from "./prune"
-export { insertCompressToolContext } from "./inject"
+export { insertCompressToolContext } from "./inject/inject"

lib/messages/inject.ts

@@ -0,0 +1,57 @@
+import type { SessionState, WithParts } from "../../state"
+import type { Logger } from "../../logger"
+import type { PluginConfig } from "../../config"
+import {
+    addAnchor,
+    applyAnchoredHints,
+    findLastNonIgnoredMessage,
+    getLimitNudgeInterval,
+    getModelInfo,
+    isContextOverLimit,
+    messageHasCompress,
+    persistAnchors,
+} from "./utils"
+
+const CONTEXT_LIMIT_HINT_TEXT = "your context exceeds the context limit, you must compress"
+
+export const insertCompressToolContext = (
+    state: SessionState,
+    config: PluginConfig,
+    logger: Logger,
+    messages: WithParts[],
+): void => {
+    if (config.tools.compress.permission === "deny") {
+        return
+    }
+
+    const lastAssistantMessage = messages.findLast((message) => message.info.role === "assistant")
+    if (lastAssistantMessage && messageHasCompress(lastAssistantMessage)) {
+        return
+    }
+
+    const { providerId, modelId } = getModelInfo(messages)
+    let anchorsChanged = false
+
+    if (isContextOverLimit(config, state, providerId, modelId, messages)) {
+        const lastNonIgnoredMessage = findLastNonIgnoredMessage(messages)
+        if (lastNonIgnoredMessage) {
+            const interval = getLimitNudgeInterval(config)
+            const added = addAnchor(
+                state.contextLimitAnchors,
+                lastNonIgnoredMessage.message.info.id,
+                lastNonIgnoredMessage.index,
+                messages,
+                interval,
+            )
+            if (added) {
+                anchorsChanged = true
+            }
+        }
+    }
+
+    applyAnchoredHints(state.contextLimitAnchors, messages, modelId, CONTEXT_LIMIT_HINT_TEXT)
+
+    if (anchorsChanged) {
+        persistAnchors(state, logger)
+    }
+}