From 64a579e965f75df813f990095c9d1d7c18cb7bc9 Mon Sep 17 00:00:00 2001 From: Pugazhendhi Date: Fri, 7 Feb 2025 19:31:36 +0530 Subject: [PATCH] Fixes default model in test file --- src/api/providers/__tests__/unbound.test.ts | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/src/api/providers/__tests__/unbound.test.ts b/src/api/providers/__tests__/unbound.test.ts index e273c7d3e8..790c0f0129 100644 --- a/src/api/providers/__tests__/unbound.test.ts +++ b/src/api/providers/__tests__/unbound.test.ts @@ -214,6 +214,15 @@ describe("UnboundHandler", () => { const nonAnthropicOptions = { apiModelId: "openai/gpt-4o", unboundApiKey: "test-key", + unboundModelId: "openai/gpt-4o", + unboundModelInfo: { + description: "OpenAI's GPT-4", + maxTokens: undefined, + contextWindow: 128000, + supportsPromptCache: true, + inputPrice: 0.01, + outputPrice: 0.03, + }, } const nonAnthropicHandler = new UnboundHandler(nonAnthropicOptions) @@ -243,7 +252,7 @@ describe("UnboundHandler", () => { unboundModelInfo: undefined, }) const modelInfo = handlerWithInvalidModel.getModel() - expect(modelInfo.id).toBe("openai/gpt-4o") // Default model + expect(modelInfo.id).toBe("anthropic/claude-3-5-sonnet-20241022") // Default model expect(modelInfo.info).toBeDefined() }) })