Skip to content

Commit fc9d1cd

Browse files
committed
test(providers): add new models.ts exports to provider mocks
attachments.ts now reads getProviderFileAttachment / INLINE_ATTACHMENT_MAX_BYTES from @/providers/models; the provider unit tests that fully mock that module need both exports or attachments.ts fails to load.
1 parent 9c4ad83 commit fc9d1cd

12 files changed

Lines changed: 48 additions & 0 deletions

File tree

apps/sim/blocks/utils.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,10 @@ vi.mock('@/lib/core/config/env-flags', () => ({
4747
}))
4848

4949
vi.mock('@/providers/models', () => ({
50+
getProviderFileAttachment: vi
51+
.fn()
52+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
53+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
5054
getHostedModels: mockGetHostedModels,
5155
getProviderModels: mockGetProviderModels,
5256
getProviderIcon: mockGetProviderIcon,

apps/sim/lib/api-key/byok.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@ vi.mock('@/lib/core/config/env-flags', () => ({
4040
}))
4141

4242
vi.mock('@/providers/models', () => ({
43+
getProviderFileAttachment: vi
44+
.fn()
45+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
46+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
4347
getHostedModels: vi.fn(() => []),
4448
}))
4549

apps/sim/providers/azure-anthropic/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,10 @@ vi.mock('@/providers/anthropic/core', () => ({
4444
executeAnthropicProviderRequest: mockExecuteAnthropic,
4545
}))
4646
vi.mock('@/providers/models', () => ({
47+
getProviderFileAttachment: vi
48+
.fn()
49+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
50+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
4751
getProviderModels: vi.fn(() => []),
4852
getProviderDefaultModel: vi.fn(() => 'azure-anthropic/claude'),
4953
}))

apps/sim/providers/azure-openai/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,10 @@ vi.mock('@/providers/azure-openai/utils', () => ({
6262
checkForForcedToolUsage: vi.fn(() => ({ hasUsedForcedTool: false, usedForcedTools: [] })),
6363
}))
6464
vi.mock('@/providers/models', () => ({
65+
getProviderFileAttachment: vi
66+
.fn()
67+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
68+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
6569
getProviderModels: vi.fn(() => []),
6670
getProviderDefaultModel: vi.fn(() => 'azure/gpt-4o'),
6771
}))

apps/sim/providers/baseten/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@ vi.mock('openai', () => ({
2626
vi.mock('@/providers', () => ({ MAX_TOOL_ITERATIONS: 5 }))
2727

2828
vi.mock('@/providers/models', () => ({
29+
getProviderFileAttachment: vi
30+
.fn()
31+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
32+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
2933
getProviderModels: vi.fn().mockReturnValue([]),
3034
getProviderDefaultModel: vi.fn().mockReturnValue('openai/gpt-oss-120b'),
3135
}))

apps/sim/providers/bedrock/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,10 @@ vi.mock('@/providers/bedrock/utils', () => ({
2525
}))
2626

2727
vi.mock('@/providers/models', () => ({
28+
getProviderFileAttachment: vi
29+
.fn()
30+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
31+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
2832
getProviderModels: vi.fn().mockReturnValue([]),
2933
getProviderDefaultModel: vi.fn().mockReturnValue('us.anthropic.claude-3-5-sonnet-20241022-v2:0'),
3034
}))

apps/sim/providers/fireworks/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@ vi.mock('openai', () => ({
2626
vi.mock('@/providers', () => ({ MAX_TOOL_ITERATIONS: 5 }))
2727

2828
vi.mock('@/providers/models', () => ({
29+
getProviderFileAttachment: vi
30+
.fn()
31+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
32+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
2933
getProviderModels: vi.fn().mockReturnValue([]),
3034
getProviderDefaultModel: vi.fn().mockReturnValue('llama-v3p1-70b-instruct'),
3135
}))

apps/sim/providers/litellm/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,10 @@ vi.mock('@/stores/providers', () => ({
2929
}))
3030

3131
vi.mock('@/providers/models', () => ({
32+
getProviderFileAttachment: vi
33+
.fn()
34+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
35+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
3236
getProviderModels: () => [],
3337
getProviderDefaultModel: () => '',
3438
}))

apps/sim/providers/ollama-cloud/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,10 @@ vi.mock('openai', () => {
4545

4646
vi.mock('@/providers', () => ({ MAX_TOOL_ITERATIONS: 20 }))
4747
vi.mock('@/providers/models', () => ({
48+
getProviderFileAttachment: vi
49+
.fn()
50+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
51+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
4852
getProviderModels: vi.fn().mockReturnValue([]),
4953
getProviderDefaultModel: vi.fn().mockReturnValue(''),
5054
}))

apps/sim/providers/openrouter/index.test.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,10 @@ vi.mock('@/providers', () => ({ MAX_TOOL_ITERATIONS: 10 }))
3737
vi.mock('@/tools', () => ({ executeTool: mockExecuteTool }))
3838

3939
vi.mock('@/providers/models', () => ({
40+
getProviderFileAttachment: vi
41+
.fn()
42+
.mockReturnValue({ maxBytes: 10 * 1024 * 1024, strategy: 'inline' }),
43+
INLINE_ATTACHMENT_MAX_BYTES: 10 * 1024 * 1024,
4044
getProviderModels: vi.fn().mockReturnValue([]),
4145
getProviderDefaultModel: vi.fn().mockReturnValue(''),
4246
}))

0 commit comments

Comments
 (0)