Skip to content

Commit ba8e6fb

Browse files
committed
fix(provenance): mark legacy attachment replay fixtures
1 parent 6bfc5a5 commit ba8e6fb

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

‎apps/sim/executor/handlers/agent/agent-handler.test.ts‎

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,10 @@ describe('AgentBlockHandler', () => {
394394
expect(inputs).toEqual(original)
395395

396396
queueTableRows(schemaMock.memory, [
397-
{ data: [...stored, { role: 'assistant', content: 'First answer' }] },
397+
{
398+
secretProvenanceVersion: null,
399+
data: [...stored, { role: 'assistant', content: 'First answer' }],
400+
},
398401
])
399402
mockGetProviderFromModel.mockReturnValue('anthropic')
400403
const nextContext = { ...mockContext, executionId: 'exec-2' }
@@ -428,6 +431,7 @@ describe('AgentBlockHandler', () => {
428431
mockGetProviderFromModel.mockReturnValue('openai')
429432
queueTableRows(schemaMock.memory, [
430433
{
434+
secretProvenanceVersion: null,
431435
data: [
432436
{ role: 'user', content: 'Analyze this file', executionId: 'exec-1', files: [file] },
433437
],
@@ -448,7 +452,9 @@ describe('AgentBlockHandler', () => {
448452

449453
it('saves a new attachment appended to an existing conversation', async () => {
450454
mockGetProviderFromModel.mockReturnValue('openai')
451-
queueTableRows(schemaMock.memory, [{ data: [{ role: 'assistant', content: 'Hello' }] }])
455+
queueTableRows(schemaMock.memory, [
456+
{ secretProvenanceVersion: null, data: [{ role: 'assistant', content: 'Hello' }] },
457+
])
452458
await handler.execute({ ...mockContext, executionId: 'exec-2' }, mockBlock, {
453459
model: 'gpt-4o',
454460
memoryType: 'conversation',
@@ -468,6 +474,7 @@ describe('AgentBlockHandler', () => {
468474
const hydrate = vi.spyOn(userFileBase64, 'hydrateUserFilesWithBase64')
469475
queueTableRows(schemaMock.memory, [
470476
{
477+
secretProvenanceVersion: null,
471478
data: [
472479
{ role: 'user', content: 'Old file', files: [file] },
473480
{ role: 'assistant', content: 'Recent answer' },

0 commit comments

Comments
 (0)