From 0107a7a03d659f1c38444f576c8643b752fa9dcd Mon Sep 17 00:00:00 2001
From: Yichen Jiang <jyc132780208@sina.com>
Date: Tue, 14 Jul 2026 22:25:59 +0800
Subject: [PATCH] test(context): adapt time context to provider routing

---
 packages/context/time-context/tests/fixtures/cordis.yml  | 1 +
 packages/context/time-context/tests/time-context.spec.ts | 7 +++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/context/time-context/tests/fixtures/cordis.yml b/packages/context/time-context/tests/fixtures/cordis.yml
index e9558abec6..afd9bc24d4 100644
--- a/packages/context/time-context/tests/fixtures/cordis.yml
+++ b/packages/context/time-context/tests/fixtures/cordis.yml
@@ -11,6 +11,7 @@
 - id: stdio-agent
   name: '@deepseek-ai/dsh-stdio-agent'
   config:
+    provider: mock
     model: mock-echo
     persona: 'Test the time-context plugin.'
     welcome: 'time-context e2e ready.'
diff --git a/packages/context/time-context/tests/time-context.spec.ts b/packages/context/time-context/tests/time-context.spec.ts
index 562b002b68..b371501e67 100644
--- a/packages/context/time-context/tests/time-context.spec.ts
+++ b/packages/context/time-context/tests/time-context.spec.ts
@@ -117,6 +117,7 @@ describe('temporal section rendering', () => {
     const session = new Session(SessionId('offset'))
     session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
     session.append('assistant/message', {
+      provenance: { provider: 'mock', model: 'mock' },
       turn: 1,
       step: 1,
       content: [{ type: 'text', text: 'previous' }],
@@ -136,6 +137,7 @@ describe('temporal section rendering', () => {
     const session = new Session(SessionId('backward-duration'))
     session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
     session.append('assistant/message', {
+      provenance: { provider: 'mock', model: 'mock' },
       turn: 1,
       step: 1,
       content: [{ type: 'text', text: 'future by adjusted clock' }],
@@ -152,7 +154,7 @@ describe('temporal section rendering', () => {
       session.append('user/message', { content: [{ type: 'text', text: 'u' }], source: { kind: 'user' } }, { surfaceOp: 'append' })
     }],
     ['assistant/message', (session: Session): void => {
-      session.append('assistant/message', { turn: 1, step: 1, content: [{ type: 'text', text: 'a' }] }, { surfaceOp: 'append' })
+      session.append('assistant/message', { provenance: { provider: 'mock', model: 'mock' }, turn: 1, step: 1, content: [{ type: 'text', text: 'a' }] }, { surfaceOp: 'append' })
     }],
     ['tool/result', (session: Session): void => {
       session.append('tool/result', {
@@ -239,6 +241,7 @@ describe('refresh policy', () => {
     const first = await sectionText(ctx, agent)
     vi.setSystemTime(BASE + 1_000)
     session.append('assistant/message', {
+      provenance: { provider: 'mock', model: 'mock' },
       turn: 1,
       step: 1,
       content: [{ type: 'text', text: 'done' }],
@@ -330,7 +333,7 @@ describe('real agent-loop request logging', () => {
         return [{ type: 'text' as const, text: 'advanced' }]
       },
     }))
-    const agent = ctx.agentLoop.create(AgentId('loop'), { model: 'mock' })
+    const agent = ctx.agentLoop.create(AgentId('loop'), { provider: 'mock', model: 'mock' })
 
     agent.send([{ type: 'text', text: 'start' }])
     await agent.whenIdle()