Pine/deepseek-harness
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases 1 Wiki Activity

test(llm): adapt retry coverage to followup API

Browse Source
This commit is contained in:
Turtle
2026-07-25 10:36:03 +08:00
parent a193ea7092
commit 4c6618c26d
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/llm/llm-retry/tests/retry.spec.ts
+10 -10
View File
@@ -352,7 +352,7 @@ describe('provider-routed retry policy', () => {
}) })
const scheduled = waitForRetry(context, agent, 1) const scheduled = waitForRetry(context, agent, 1)
agent.send([{ type: 'text', text: 'go' }]) agent.followup([{ type: 'text', text: 'go' }])
expect((await scheduled).data.delayMs).toBe(3) expect((await scheduled).data.delayMs).toBe(3)
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
await vi.advanceTimersByTimeAsync(3) await vi.advanceTimersByTimeAsync(3)
@@ -390,7 +390,7 @@ describe('provider-routed retry policy', () => {
model: 'mock', model: 'mock',
}) })
const normalIdle = waitForIdle(context, normalAgent) const normalIdle = waitForIdle(context, normalAgent)
normalAgent.send([{ type: 'text', text: 'normal' }]) normalAgent.followup([{ type: 'text', text: 'normal' }])
await normalIdle await normalIdle
expect(normalAgent.session.events.some(event => event.type === 'llm/retry')).toBe(false) expect(normalAgent.session.events.some(event => event.type === 'llm/retry')).toBe(false)
@@ -399,7 +399,7 @@ describe('provider-routed retry policy', () => {
model: 'mock', model: 'mock',
}) })
const scheduled = waitForRetry(context, alwaysAgent, 1) const scheduled = waitForRetry(context, alwaysAgent, 1)
alwaysAgent.send([{ type: 'text', text: 'always' }]) alwaysAgent.followup([{ type: 'text', text: 'always' }])
expect((await scheduled).data).toMatchObject({ expect((await scheduled).data).toMatchObject({
provider: 'other', provider: 'other',
mode: 'always', mode: 'always',
@@ -433,7 +433,7 @@ describe('provider-routed retry policy', () => {
}) })
const scheduled = waitForRetry(context, agent, 1) const scheduled = waitForRetry(context, agent, 1)
agent.send([{ type: 'text', text: 'reroute' }]) agent.followup([{ type: 'text', text: 'reroute' }])
expect((await scheduled).data).toMatchObject({ provider: 'other', mode: 'always' }) expect((await scheduled).data).toMatchObject({ provider: 'other', mode: 'always' })
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
await vi.advanceTimersByTimeAsync(1) await vi.advanceTimersByTimeAsync(1)
@@ -462,7 +462,7 @@ describe('provider-routed retry policy', () => {
}) })
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
agent.send([{ type: 'text', text: 'keep trying' }]) agent.followup([{ type: 'text', text: 'keep trying' }])
await vi.runAllTimersAsync() await vi.runAllTimersAsync()
await idle await idle
@@ -499,7 +499,7 @@ describe('provider-routed retry policy', () => {
}) })
const scheduled = waitForRetry(context, agent, 1) const scheduled = waitForRetry(context, agent, 1)
agent.send([{ type: 'text', text: 'safe input' }]) agent.followup([{ type: 'text', text: 'safe input' }])
await scheduled await scheduled
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
await vi.advanceTimersByTimeAsync(1) await vi.advanceTimersByTimeAsync(1)
@@ -528,7 +528,7 @@ describe('provider-routed retry policy', () => {
}) })
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
agent.send([{ type: 'text', text: 'recover' }]) agent.followup([{ type: 'text', text: 'recover' }])
await idle await idle
expect(adapter.requests).toHaveLength(2) expect(adapter.requests).toHaveLength(2)
@@ -555,7 +555,7 @@ describe('provider-routed retry policy', () => {
}) })
const scheduled = waitForRetry(context, agent, 1) const scheduled = waitForRetry(context, agent, 1)
agent.send([{ type: 'text', text: 'recover' }]) agent.followup([{ type: 'text', text: 'recover' }])
await scheduled await scheduled
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
await vi.advanceTimersByTimeAsync(1) await vi.advanceTimersByTimeAsync(1)
@@ -635,7 +635,7 @@ describe('provider-routed retry policy', () => {
model: 'mock', model: 'mock',
}) })
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
agent.send([{ type: 'text', text: 'go' }]) agent.followup([{ type: 'text', text: 'go' }])
await entered.promise await entered.promise
agent.cancel({ kind: 'user' }) agent.cancel({ kind: 'user' })
@@ -664,7 +664,7 @@ describe('provider-routed retry policy', () => {
}) })
const idle = waitForIdle(context, agent) const idle = waitForIdle(context, agent)
agent.send([{ type: 'text', text: 'go' }]) agent.followup([{ type: 'text', text: 'go' }])
await idle await idle
downstream.resolve({ action: 'fail' }) downstream.resolve({ action: 'fail' })