@halo-sdk/strategies 1.0.0

package/README.md

@@ -0,0 +1,6 @@
+# @halo-sdk/strategies
+
+Pluggable strategies for Halo AI SDK.
+
+- `TruncateStrategy` — context window truncation
+- `BasicRepair` — truncated JSON repair

package/dist/basic-repair.d.ts

@@ -0,0 +1,5 @@
+import type { ToolCall, RepairStrategy, RepairResult } from "@halo-sdk/core";
+export declare class BasicRepair implements RepairStrategy {
+    repair(toolCalls: ToolCall[], _rawContent: string): RepairResult;
+}
+//# sourceMappingURL=basic-repair.d.ts.map

package/dist/basic-repair.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"basic-repair.d.ts","sourceRoot":"","sources":["../src/basic-repair.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,QAAQ,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE7E,qBAAa,WAAY,YAAW,cAAc;IAChD,MAAM,CAAC,SAAS,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,MAAM,GAAG,YAAY;CAyBjE"}

package/dist/basic-repair.js

@@ -0,0 +1,101 @@
+export class BasicRepair {
+    repair(toolCalls, _rawContent) {
+        const repaired = [];
+        let fixed = 0;
+        for (const call of toolCalls) {
+            const args = call.function?.arguments ?? "";
+            const fix = tryFixTruncatedJson(args);
+            if (fix !== args) {
+                fixed++;
+                repaired.push({
+                    ...call,
+                    function: { ...call.function, arguments: fix },
+                });
+            }
+            else {
+                repaired.push(call);
+            }
+        }
+        return {
+            toolCalls: repaired,
+            fixed,
+            suppressed: 0,
+            notes: fixed > 0 ? [`fixed ${fixed} truncated JSON argument(s)`] : [],
+        };
+    }
+}
+function tryFixTruncatedJson(raw) {
+    if (!raw || !raw.trim())
+        return raw;
+    // Try direct parse first.
+    try {
+        JSON.parse(raw);
+        return raw;
+    }
+    catch {
+        /* needs fixing */
+    }
+    // Count unbalanced brackets and braces.
+    let braces = 0;
+    let brackets = 0;
+    let inString = false;
+    let escaped = false;
+    for (const ch of raw) {
+        if (escaped) {
+            escaped = false;
+            continue;
+        }
+        if (ch === "\\") {
+            escaped = true;
+            continue;
+        }
+        if (ch === '"') {
+            inString = !inString;
+            continue;
+        }
+        if (inString)
+            continue;
+        if (ch === "{")
+            braces++;
+        if (ch === "}")
+            braces--;
+        if (ch === "[")
+            brackets++;
+        if (ch === "]")
+            brackets--;
+    }
+    // Also fix unterminated string.
+    if (inString) {
+        let fixed = raw;
+        // Close the string before closing brackets.
+        if (braces > 0 || brackets > 0)
+            fixed += '"';
+        while (braces > 0) {
+            fixed += "}";
+            braces--;
+        }
+        while (brackets > 0) {
+            fixed += "]";
+            brackets--;
+        }
+        return fixed;
+    }
+    let fixed = raw;
+    while (braces > 0) {
+        fixed += "}";
+        braces--;
+    }
+    while (brackets > 0) {
+        fixed += "]";
+        brackets--;
+    }
+    // Verify the fix is valid JSON.
+    try {
+        JSON.parse(fixed);
+        return fixed;
+    }
+    catch {
+        return raw;
+    }
+}
+//# sourceMappingURL=basic-repair.js.map

package/dist/basic-repair.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"basic-repair.js","sourceRoot":"","sources":["../src/basic-repair.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,WAAW;IACtB,MAAM,CAAC,SAAqB,EAAE,WAAmB;QAC/C,MAAM,QAAQ,GAAe,EAAE,CAAC;QAChC,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,SAAS,IAAI,EAAE,CAAC;YAC5C,MAAM,GAAG,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;gBACjB,KAAK,EAAE,CAAC;gBACR,QAAQ,CAAC,IAAI,CAAC;oBACZ,GAAG,IAAI;oBACP,QAAQ,EAAE,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,SAAS,EAAE,GAAG,EAAE;iBAC/C,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;QAED,OAAO;YACL,SAAS,EAAE,QAAQ;YACnB,KAAK;YACL,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,KAAK,6BAA6B,CAAC,CAAC,CAAC,CAAC,EAAE;SACtE,CAAC;IACJ,CAAC;CACF;AAED,SAAS,mBAAmB,CAAC,GAAW;IACtC,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE;QAAE,OAAO,GAAG,CAAC;IAEpC,0BAA0B;IAC1B,IAAI,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAChB,OAAO,GAAG,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACP,kBAAkB;IACpB,CAAC;IAED,wCAAwC;IACxC,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,QAAQ,GAAG,CAAC,CAAC;IACjB,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,OAAO,GAAG,KAAK,CAAC;IAEpB,KAAK,MAAM,EAAE,IAAI,GAAG,EAAE,CAAC;QACrB,IAAI,OAAO,EAAE,CAAC;YACZ,OAAO,GAAG,KAAK,CAAC;YAChB,SAAS;QACX,CAAC;QACD,IAAI,EAAE,KAAK,IAAI,EAAE,CAAC;YAChB,OAAO,GAAG,IAAI,CAAC;YACf,SAAS;QACX,CAAC;QACD,IAAI,EAAE,KAAK,GAAG,EAAE,CAAC;YACf,QAAQ,GAAG,CAAC,QAAQ,CAAC;YACrB,SAAS;QACX,CAAC;QACD,IAAI,QAAQ;YAAE,SAAS;QACvB,IAAI,EAAE,KAAK,GAAG;YAAE,MAAM,EAAE,CAAC;QACzB,IAAI,EAAE,KAAK,GAAG;YAAE,MAAM,EAAE,CAAC;QACzB,IAAI,EAAE,KAAK,GAAG;YAAE,QAAQ,EAAE,CAAC;QAC3B,IAAI,EAAE,KAAK,GAAG;YAAE,QAAQ,EAAE,CAAC;IAC7B,CAAC;IAED,gCAAgC;IAChC,IAAI,QAAQ,EAAE,CAAC;QACb,IAAI,KAAK,GAAG,GAAG,CAAC;QAChB,4CAA4C;QAC5C,IAAI,MAAM,GAAG,CAAC,IAAI,QAAQ,GAAG,CAAC;YAAE,KAAK,IAAI,GAAG,CAAC;QAC7C,OAAO,MAAM,GAAG,CAAC,EAAE,CAAC;YAClB,KAAK,IAAI,GAAG,CAAC;YACb,MAAM,EAAE,CAAC;QACX,CAAC;QACD,OAAO,QAAQ,GAAG,CAAC,EAAE,CAAC;YACpB,KAAK,IAAI,GAAG,CAAC;YACb,QAAQ,EAAE,CAAC;QACb,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,KAAK,GAAG,GAAG,CAAC;IAChB,OAAO,MAAM,GAAG,CAAC,EAAE,CAAC;QAClB,KAAK,IAAI,GAAG,CAAC;QACb,MAAM,EAAE,CAAC;IACX,CAAC;IACD,OAAO,QAAQ,GAAG,CAAC,EAAE,CAAC;QACpB,KAAK,IAAI,GAAG,CAAC;QACb,QAAQ,EAAE,CAAC;IACb,CAAC;IAED,gCAAgC;IAChC,IAAI,CAAC;QACH,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClB,OAAO,KAAK,CAAC;IACf,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,CAAC;IACb,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+export { TruncateStrategy } from "./truncate.js";
+export { BasicRepair } from "./basic-repair.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/index.js

@@ -0,0 +1,3 @@
+export { TruncateStrategy } from "./truncate.js";
+export { BasicRepair } from "./basic-repair.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/truncate.d.ts

@@ -0,0 +1,14 @@
+import type { ChatMessage, ContextStrategy } from "@halo-sdk/core";
+export declare class TruncateStrategy implements ContextStrategy {
+    private _maxTokens;
+    constructor(opts?: {
+        maxTokens?: number;
+    });
+    prepare(prefix: ChatMessage[], history: ChatMessage[], _ctxMax: number): {
+        history: ChatMessage[];
+        modified: boolean;
+        summary?: string;
+        droppedCount: number;
+    };
+}
+//# sourceMappingURL=truncate.d.ts.map

package/dist/truncate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"truncate.d.ts","sourceRoot":"","sources":["../src/truncate.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEnE,qBAAa,gBAAiB,YAAW,eAAe;IACtD,OAAO,CAAC,UAAU,CAAS;gBAEf,IAAI,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE;IAKzC,OAAO,CACL,MAAM,EAAE,WAAW,EAAE,EACrB,OAAO,EAAE,WAAW,EAAE,EACtB,OAAO,EAAE,MAAM,GACd;QACD,OAAO,EAAE,WAAW,EAAE,CAAC;QACvB,QAAQ,EAAE,OAAO,CAAC;QAClB,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,YAAY,EAAE,MAAM,CAAC;KACtB;CA6BF"}

package/dist/truncate.js

@@ -0,0 +1,34 @@
+export class TruncateStrategy {
+    _maxTokens;
+    constructor(opts) {
+        // Default: 80% of a 128K context window.
+        this._maxTokens = opts?.maxTokens ?? 102_400;
+    }
+    prepare(prefix, history, _ctxMax) {
+        // Prefix is read-only — only history may be truncated.
+        const histEstimate = history.reduce((sum, m) => sum + m.content.length, 0);
+        const prefixEstimate = prefix.reduce((sum, m) => sum + m.content.length, 0);
+        const budgetChars = (this._maxTokens - prefixEstimate / 4) * 4;
+        if (histEstimate <= budgetChars * 0.9) {
+            return { history, modified: false, droppedCount: 0 };
+        }
+        // Keep the most recent messages that fit the remaining budget.
+        let tailTokens = 0;
+        const tail = [];
+        for (let i = history.length - 1; i >= 0; i--) {
+            const msg = history[i];
+            const size = msg.content.length;
+            if (tailTokens + size > budgetChars * 0.9)
+                break;
+            tailTokens += size;
+            tail.unshift(msg);
+        }
+        const droppedCount = history.length - tail.length;
+        return {
+            history: tail,
+            modified: droppedCount > 0,
+            droppedCount,
+        };
+    }
+}
+//# sourceMappingURL=truncate.js.map

package/dist/truncate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"truncate.js","sourceRoot":"","sources":["../src/truncate.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,gBAAgB;IACnB,UAAU,CAAS;IAE3B,YAAY,IAA6B;QACvC,yCAAyC;QACzC,IAAI,CAAC,UAAU,GAAG,IAAI,EAAE,SAAS,IAAI,OAAO,CAAC;IAC/C,CAAC;IAED,OAAO,CACL,MAAqB,EACrB,OAAsB,EACtB,OAAe;QAOf,uDAAuD;QACvD,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC3E,MAAM,cAAc,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC5E,MAAM,WAAW,GAAG,CAAC,IAAI,CAAC,UAAU,GAAG,cAAc,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;QAE/D,IAAI,YAAY,IAAI,WAAW,GAAG,GAAG,EAAE,CAAC;YACtC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;QACvD,CAAC;QAED,+DAA+D;QAC/D,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,MAAM,IAAI,GAAkB,EAAE,CAAC;QAC/B,KAAK,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,GAAG,GAAG,OAAO,CAAC,CAAC,CAAE,CAAC;YACxB,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC;YAChC,IAAI,UAAU,GAAG,IAAI,GAAG,WAAW,GAAG,GAAG;gBAAE,MAAM;YACjD,UAAU,IAAI,IAAI,CAAC;YACnB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,YAAY,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAElD,OAAO;YACL,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,YAAY,GAAG,CAAC;YAC1B,YAAY;SACb,CAAC;IACJ,CAAC;CACF"}

package/package.json

@@ -0,0 +1,47 @@
+{
+  "name": "@halo-sdk/strategies",
+  "version": "1.0.0",
+  "description": "Pluggable strategies for Halo AI SDK — context window truncation and JSON repair",
+  "license": "MIT",
+  "keywords": [
+    "ai",
+    "llm",
+    "context-window",
+    "truncation",
+    "json-repair"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/halo-sdk/halo-ai",
+    "directory": "packages/strategies"
+  },
+  "files": [
+    "dist"
+  ],
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "peerDependencies": {
+    "@halo-sdk/core": ">=1.0.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.8.0",
+    "vitest": "^3.0.0",
+    "@halo-sdk/core": "1.0.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  }
+}