Просмотр исходного кода

Exercise normalized goal-round rate limits

Tianyi Cui 2 месяцев назад
Родитель
Сommit
de72f972b7
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      packages/goal/goal-session/tests/goal-session.spec.ts

+ 2 - 2
packages/goal/goal-session/tests/goal-session.spec.ts

@@ -6,7 +6,7 @@ import AgentLoop from '@deepseek-ai/dsh-agent-loop'
 import { mountAgentLoopTestDependencies } from '@deepseek-ai/dsh-agent-loop-testkit'
 import GoalService, { GoalId } from '@deepseek-ai/dsh-goal'
 import type { GoalView } from '@deepseek-ai/dsh-goal'
-import { LlmAdapter } from '@deepseek-ai/dsh-llm'
+import { LlmAdapter, LlmError } from '@deepseek-ai/dsh-llm'
 import type { GenerateOptions, StreamChunk } from '@deepseek-ai/dsh-llm'
 import { SessionId } from '@deepseek-ai/dsh-session'
 import type { TurnEndReason } from '@deepseek-ai/dsh-session'
@@ -232,7 +232,7 @@ describe('same-session goal driving', () => {
   })
 
   it.each([
-    ['rate limit', Object.assign(new Error('slow down'), { code: 'RATE_LIMIT' }), 'usage-limited'],
+    ['rate limit', new LlmError('slow down', 'RATE_LIMIT'), 'usage-limited'],
     ['request error', new Error('provider broke'), 'turn-error'],
     ['max tokens', maxTokensResponse('unfinished'), 'max-tokens'],
   ] as const)('stops after a %s without an automatic retry', async (_label, response, code) => {