Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix type issue rec

This commit is contained in:
dal 2025-09-16 23:52:38 -06:00
parent 46bafb36d4
commit a29f357324
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
packages/ai/src/utils/with-agent-retry.test.ts
View File

@ -1,6 +1,7 @@
import type { ModelMessage } from 'ai';
import { beforeEach, describe, expect, it, vi } from 'vitest';
import {
type StreamExecutor,
analyzeError,
calculateBackoffDelay,
composeMiddleware,
@ -427,14 +428,14 @@ describe('with-agent-retry', () => {
const messages: ModelMessage[] = [];
const callOrder: string[] = [];
const middleware1 = (next: any) => async (msgs: ModelMessage[]) => {
const middleware1 = (next: StreamExecutor<unknown>) => async (msgs: ModelMessage[]) => {
callOrder.push('middleware1-before');
const result = await next(msgs);
callOrder.push('middleware1-after');
return result;
};
const middleware2 = (next: any) => async (msgs: ModelMessage[]) => {
const middleware2 = (next: StreamExecutor<unknown>) => async (msgs: ModelMessage[]) => {
callOrder.push('middleware2-before');
const result = await next(msgs);
callOrder.push('middleware2-after');