diff --git a/.changeset/early-spoons-scream.md b/.changeset/early-spoons-scream.md new file mode 100644 index 0000000000000000000000000000000000000000..8fe79dc4ec3242beab6b2bc22df886f7bd77b5bb --- /dev/null +++ b/.changeset/early-spoons-scream.md @@ -0,0 +1,5 @@ +--- +"llamaindex": minor +--- + +fix bug with node decorator diff --git a/packages/core/src/internal/decorator/node.ts b/packages/core/src/internal/decorator/node.ts index be52eac9f1a17a9ae28fa1dc6fd0a0e7629e5a3e..25ec24156448e8d81805050b3bbe1bcf877c4ce0 100644 --- a/packages/core/src/internal/decorator/node.ts +++ b/packages/core/src/internal/decorator/node.ts @@ -4,12 +4,19 @@ import { getChunkSize } from "../settings/chunk-size.js"; const emitOnce = false; -export function chunkSizeCheck( - contentGetter: () => string, - _context: ClassMethodDecoratorContext | ClassGetterDecoratorContext, +export function chunkSizeCheck< + This extends BaseNode, + Args extends any[], + Return, +>( + contentGetter: (this: This, ...args: Args) => string, + _context: ClassMethodDecoratorContext< + This, + (this: This, ...args: Args) => Return + >, ) { - return function <Node extends BaseNode>(this: Node) { - const content = contentGetter.call(this); + return function (this: This, ...args: Args) { + const content = contentGetter.call(this, ...args); const chunkSize = getChunkSize(); const enableChunkSizeCheck = getEnv("ENABLE_CHUNK_SIZE_CHECK") === "true"; if ( diff --git a/packages/core/tests/Node.test.ts b/packages/core/tests/Node.test.ts index 0bbfab42a527c91d8dea1f1ba99f8061b5fd37f5..08ad45690290a8b2d54631759655148e14d84077 100644 --- a/packages/core/tests/Node.test.ts +++ b/packages/core/tests/Node.test.ts @@ -53,7 +53,7 @@ describe("TextNode", () => { "endCharIdx": undefined, "excludedEmbedMetadataKeys": [], "excludedLlmMetadataKeys": [], - "hash": "nTSKdUTYqR52MPv/brvb4RTGeqedTEqG9QN8KSAj2Do=", + "hash": "Z6SWgFPlalaeblMGQGw0KS3qKgmZdEWXKfzEp/K+QN0=", "id_": Any<String>, "metadata": { "something": 1,