Skip to content

Commit 58f5cd7

Browse files
canrobins13canrobins13
committed
test typing fixes
1 parent 2d544d5 commit 58f5cd7

File tree

1 file changed

+39
-0
lines changed

1 file changed

+39
-0
lines changed

‎src/core/sliding-window/__tests__/sliding-window.test.ts‎

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -248,6 +248,8 @@ describe("truncateConversationIfNeeded", () => {
248248
contextWindow: modelInfo.contextWindow,
249249
maxTokens: modelInfo.maxTokens,
250250
apiHandler: mockApiHandler,
251+
autoCondenseContext: false,
252+
autoCondenseContextPercent: 100,
251253
systemPrompt: "System prompt",
252254
})
253255

@@ -277,6 +279,8 @@ describe("truncateConversationIfNeeded", () => {
277279
contextWindow: modelInfo.contextWindow,
278280
maxTokens: modelInfo.maxTokens,
279281
apiHandler: mockApiHandler,
282+
autoCondenseContext: false,
283+
autoCondenseContextPercent: 100,
280284
systemPrompt: "System prompt",
281285
})
282286

@@ -304,6 +308,8 @@ describe("truncateConversationIfNeeded", () => {
304308
contextWindow: modelInfo1.contextWindow,
305309
maxTokens: modelInfo1.maxTokens,
306310
apiHandler: mockApiHandler,
311+
autoCondenseContext: false,
312+
autoCondenseContextPercent: 100,
307313
systemPrompt: "System prompt",
308314
})
309315

@@ -313,6 +319,8 @@ describe("truncateConversationIfNeeded", () => {
313319
contextWindow: modelInfo2.contextWindow,
314320
maxTokens: modelInfo2.maxTokens,
315321
apiHandler: mockApiHandler,
322+
autoCondenseContext: false,
323+
autoCondenseContextPercent: 100,
316324
systemPrompt: "System prompt",
317325
})
318326

@@ -329,6 +337,8 @@ describe("truncateConversationIfNeeded", () => {
329337
contextWindow: modelInfo1.contextWindow,
330338
maxTokens: modelInfo1.maxTokens,
331339
apiHandler: mockApiHandler,
340+
autoCondenseContext: false,
341+
autoCondenseContextPercent: 100,
332342
systemPrompt: "System prompt",
333343
})
334344

@@ -338,6 +348,8 @@ describe("truncateConversationIfNeeded", () => {
338348
contextWindow: modelInfo2.contextWindow,
339349
maxTokens: modelInfo2.maxTokens,
340350
apiHandler: mockApiHandler,
351+
autoCondenseContext: false,
352+
autoCondenseContextPercent: 100,
341353
systemPrompt: "System prompt",
342354
})
343355

@@ -369,6 +381,8 @@ describe("truncateConversationIfNeeded", () => {
369381
contextWindow: modelInfo.contextWindow,
370382
maxTokens,
371383
apiHandler: mockApiHandler,
384+
autoCondenseContext: false,
385+
autoCondenseContextPercent: 100,
372386
systemPrompt: "System prompt",
373387
})
374388
expect(resultWithSmall).toEqual({
@@ -399,6 +413,8 @@ describe("truncateConversationIfNeeded", () => {
399413
contextWindow: modelInfo.contextWindow,
400414
maxTokens,
401415
apiHandler: mockApiHandler,
416+
autoCondenseContext: false,
417+
autoCondenseContextPercent: 100,
402418
systemPrompt: "System prompt",
403419
})
404420
expect(resultWithLarge.messages).not.toEqual(messagesWithLargeContent) // Should truncate
@@ -422,6 +438,8 @@ describe("truncateConversationIfNeeded", () => {
422438
contextWindow: modelInfo.contextWindow,
423439
maxTokens,
424440
apiHandler: mockApiHandler,
441+
autoCondenseContext: false,
442+
autoCondenseContextPercent: 100,
425443
systemPrompt: "System prompt",
426444
})
427445
expect(resultWithVeryLarge.messages).not.toEqual(messagesWithVeryLargeContent) // Should truncate
@@ -448,6 +466,8 @@ describe("truncateConversationIfNeeded", () => {
448466
contextWindow: modelInfo.contextWindow,
449467
maxTokens: modelInfo.maxTokens,
450468
apiHandler: mockApiHandler,
469+
autoCondenseContext: false,
470+
autoCondenseContextPercent: 100,
451471
systemPrompt: "System prompt",
452472
})
453473
expect(result).toEqual({
@@ -488,6 +508,7 @@ describe("truncateConversationIfNeeded", () => {
488508
maxTokens: modelInfo.maxTokens,
489509
apiHandler: mockApiHandler,
490510
autoCondenseContext: true,
511+
autoCondenseContextPercent: 100,
491512
systemPrompt: "System prompt",
492513
})
493514

@@ -534,6 +555,7 @@ describe("truncateConversationIfNeeded", () => {
534555
maxTokens: modelInfo.maxTokens,
535556
apiHandler: mockApiHandler,
536557
autoCondenseContext: true,
558+
autoCondenseContextPercent: 100,
537559
systemPrompt: "System prompt",
538560
})
539561

@@ -570,6 +592,7 @@ describe("truncateConversationIfNeeded", () => {
570592
maxTokens: modelInfo.maxTokens,
571593
apiHandler: mockApiHandler,
572594
autoCondenseContext: false,
595+
autoCondenseContextPercent: 100,
573596
systemPrompt: "System prompt",
574597
})
575598

@@ -624,6 +647,8 @@ describe("getMaxTokens", () => {
624647
contextWindow: modelInfo.contextWindow,
625648
maxTokens: modelInfo.maxTokens,
626649
apiHandler: mockApiHandler,
650+
autoCondenseContext: false,
651+
autoCondenseContextPercent: 100,
627652
systemPrompt: "System prompt",
628653
})
629654
expect(result1).toEqual({
@@ -640,6 +665,8 @@ describe("getMaxTokens", () => {
640665
contextWindow: modelInfo.contextWindow,
641666
maxTokens: modelInfo.maxTokens,
642667
apiHandler: mockApiHandler,
668+
autoCondenseContext: false,
669+
autoCondenseContextPercent: 100,
643670
systemPrompt: "System prompt",
644671
})
645672
expect(result2.messages).not.toEqual(messagesWithSmallContent)
@@ -664,6 +691,8 @@ describe("getMaxTokens", () => {
664691
contextWindow: modelInfo.contextWindow,
665692
maxTokens: modelInfo.maxTokens,
666693
apiHandler: mockApiHandler,
694+
autoCondenseContext: false,
695+
autoCondenseContextPercent: 100,
667696
systemPrompt: "System prompt",
668697
})
669698
expect(result1).toEqual({
@@ -680,6 +709,8 @@ describe("getMaxTokens", () => {
680709
contextWindow: modelInfo.contextWindow,
681710
maxTokens: modelInfo.maxTokens,
682711
apiHandler: mockApiHandler,
712+
autoCondenseContext: false,
713+
autoCondenseContextPercent: 100,
683714
systemPrompt: "System prompt",
684715
})
685716
expect(result2.messages).not.toEqual(messagesWithSmallContent)
@@ -703,6 +734,8 @@ describe("getMaxTokens", () => {
703734
contextWindow: modelInfo.contextWindow,
704735
maxTokens: modelInfo.maxTokens,
705736
apiHandler: mockApiHandler,
737+
autoCondenseContext: false,
738+
autoCondenseContextPercent: 100,
706739
systemPrompt: "System prompt",
707740
})
708741
expect(result1.messages).toEqual(messagesWithSmallContent)
@@ -714,6 +747,8 @@ describe("getMaxTokens", () => {
714747
contextWindow: modelInfo.contextWindow,
715748
maxTokens: modelInfo.maxTokens,
716749
apiHandler: mockApiHandler,
750+
autoCondenseContext: false,
751+
autoCondenseContextPercent: 100,
717752
systemPrompt: "System prompt",
718753
})
719754
expect(result2).not.toEqual(messagesWithSmallContent)
@@ -735,6 +770,8 @@ describe("getMaxTokens", () => {
735770
contextWindow: modelInfo.contextWindow,
736771
maxTokens: modelInfo.maxTokens,
737772
apiHandler: mockApiHandler,
773+
autoCondenseContext: false,
774+
autoCondenseContextPercent: 100,
738775
systemPrompt: "System prompt",
739776
})
740777
expect(result1.messages).toEqual(messagesWithSmallContent)
@@ -746,6 +783,8 @@ describe("getMaxTokens", () => {
746783
contextWindow: modelInfo.contextWindow,
747784
maxTokens: modelInfo.maxTokens,
748785
apiHandler: mockApiHandler,
786+
autoCondenseContext: false,
787+
autoCondenseContextPercent: 100,
749788
systemPrompt: "System prompt",
750789
})
751790
expect(result2).not.toEqual(messagesWithSmallContent)

0 commit comments

Comments
 (0)