small refactor and prepare strategies

Author: jorgenwh

lib/core/strategies/deduplication.ts

@@ -2,6 +2,7 @@ import type { SessionState, WithParts } from "./state"
 import type { Logger } from "./logger"
 import type { PluginConfig } from "./config"
 import { syncToolCache } from "./state/tool-cache"
+import { deduplicate } from "./strategies"
 
 
 export function createChatMessageTransformHandler(
@@ -15,6 +16,9 @@ export function createChatMessageTransformHandler(
         output: { messages: WithParts[] }
     ) => {
         syncToolCache(state, logger, output.messages);
+
+        deduplicate(state, logger, config, output.messages)
+        pruneTool(state, logger, config, output.messages)
     }
 }
 

lib/core/strategies/index.ts

@@ -2,7 +2,7 @@ import { readFileSync } from "fs"
 import { join } from "path"
 
 export function loadPrompt(name: string, vars?: Record<string, string>): string {
-    const filePath = join(__dirname, "..", "prompts", `${name}.txt`)
+    const filePath = join(__dirname, "prompts", `${name}.txt`)
     let content = readFileSync(filePath, "utf8").trim()
     if (vars) {
         for (const [key, value] of Object.entries(vars)) {

lib/hooks.ts

@@ -8,7 +8,7 @@ import { ensureSessionInitialized } from "./state"
 import { saveSessionState } from "./state/persistence"
 import type { Logger } from "./logger"
 import { estimateTokensBatch } from "./tokenizer"
-import { loadPrompt } from "./core/prompt"
+import { loadPrompt } from "./prompt"
 
 /** Tool description loaded from prompts/tool.txt */
 const TOOL_DESCRIPTION = loadPrompt("tool")

lib/core/prompt.ts lib/prompt.tslib/core/prompt.ts renamed to lib/prompt.ts

@@ -26,9 +26,13 @@ export interface SessionStats {
     totalGCTools: number
 }
 
+export interface Prune {
+    toolIds: string[]
+}
+
 export interface SessionState {
     sessionId: string | null
-    prunedIds: string[]
+    prune: Prune
     stats: SessionStats
     gcPending: GCStats
     toolParameters: Map<string, ToolParameterEntry>

lib/pruning-tool.ts

@@ -0,0 +1,12 @@
+import { PluginConfig } from "../config"
+import { Logger } from "../logger"
+import type { SessionState, WithParts } from "../state"
+
+export const deduplicate = (
+    state: SessionState,
+    logger: Logger,
+    config: PluginConfig,
+    messages: WithParts[]
+) => {
+
+}

lib/state/types.ts

@@ -0,0 +1,2 @@
+export { deduplicate } from "./deduplication"
+