Skip to content

Commit 64a579e

Browse files
committed
Fixes default model in test file
1 parent ddc6b9c commit 64a579e

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/api/providers/__tests__/unbound.test.ts

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,15 @@ describe("UnboundHandler", () => {
214214
const nonAnthropicOptions = {
215215
apiModelId: "openai/gpt-4o",
216216
unboundApiKey: "test-key",
217+
unboundModelId: "openai/gpt-4o",
218+
unboundModelInfo: {
219+
description: "OpenAI's GPT-4",
220+
maxTokens: undefined,
221+
contextWindow: 128000,
222+
supportsPromptCache: true,
223+
inputPrice: 0.01,
224+
outputPrice: 0.03,
225+
},
217226
}
218227
const nonAnthropicHandler = new UnboundHandler(nonAnthropicOptions)
219228

@@ -243,7 +252,7 @@ describe("UnboundHandler", () => {
243252
unboundModelInfo: undefined,
244253
})
245254
const modelInfo = handlerWithInvalidModel.getModel()
246-
expect(modelInfo.id).toBe("openai/gpt-4o") // Default model
255+
expect(modelInfo.id).toBe("anthropic/claude-3-5-sonnet-20241022") // Default model
247256
expect(modelInfo.info).toBeDefined()
248257
})
249258
})

0 commit comments

Comments
 (0)