make compaction work for every model (#19)

Author: nexxeln

src/index.ts

@@ -46,9 +46,37 @@ export const SupermemoryPlugin: Plugin = async (ctx: PluginInput) => {
     log("Plugin disabled - SUPERMEMORY_API_KEY not set");
   }
 
+  // Fetch model limits once at plugin init
+  const modelLimits = new Map<string, number>();
+
+  (async () => {
+    try {
+      const response = await ctx.client.provider.list();
+      if (response.data?.all) {
+        for (const provider of response.data.all) {
+          if (provider.models) {
+            for (const [modelId, model] of Object.entries(provider.models)) {
+              if (model.limit?.context) {
+                modelLimits.set(`${provider.id}/${modelId}`, model.limit.context);
+              }
+            }
+          }
+        }
+      }
+      log("Model limits loaded", { count: modelLimits.size });
+    } catch (error) {
+      log("Failed to fetch model limits", { error: String(error) });
+    }
+  })();
+
+  const getModelLimit = (providerID: string, modelID: string): number | undefined => {
+    return modelLimits.get(`${providerID}/${modelID}`);
+  };
+
   const compactionHook = isConfigured() && ctx.client
     ? createCompactionHook(ctx as CompactionContext, tags, {
         threshold: CONFIG.compactionThreshold,
+        getModelLimit,
       })
     : null;
 

src/services/compaction.ts

@@ -11,8 +11,7 @@ const PART_STORAGE = join(homedir(), ".opencode", "parts");
 const DEFAULT_THRESHOLD = 0.80;
 const MIN_TOKENS_FOR_COMPACTION = 50_000;
 const COMPACTION_COOLDOWN_MS = 30_000;
-const CLAUDE_DEFAULT_CONTEXT_LIMIT = 200_000;
-const CLAUDE_MODEL_PATTERN = /claude-(opus|sonnet|haiku)/i;
+const DEFAULT_CONTEXT_LIMIT = 200_000;
 
 interface CompactionState {
   lastCompactionTime: Map<string, number>;
@@ -98,10 +97,6 @@ This context is critical for maintaining continuity after compaction.
 `;
 }
 
-function isSupportedModel(modelID: string): boolean {
-  return CLAUDE_MODEL_PATTERN.test(modelID);
-}
-
 function getMessageDir(sessionID: string): string | null {
   if (!existsSync(MESSAGE_STORAGE)) return null;
 
@@ -344,13 +339,8 @@ export function createCompactionHook(
     }
     agent = storedMessage?.agent;
 
-    if (!isSupportedModel(modelID)) {
-      log("[compaction] skipping unsupported model", { modelID });
-      return;
-    }
-
     const configLimit = getModelLimit?.(providerID, modelID);
-    const contextLimit = configLimit ?? CLAUDE_DEFAULT_CONTEXT_LIMIT;
+    const contextLimit = configLimit ?? DEFAULT_CONTEXT_LIMIT;
     const totalUsed = tokens.input + tokens.cache.read + tokens.output;
 
     if (totalUsed < MIN_TOKENS_FOR_COMPACTION) return;