|
|
@@ -163,6 +163,28 @@ describe("ChutesHandler", () => {
|
|
|
expect(model.info).toEqual(expect.objectContaining(chutesModels[testModelId]))
|
|
|
})
|
|
|
|
|
|
+ it("should return Qwen3-235B-A22B-Instruct-2507 model with correct configuration", () => {
|
|
|
+ const testModelId: ChutesModelId = "Qwen/Qwen3-235B-A22B-Instruct-2507"
|
|
|
+ const handlerWithModel = new ChutesHandler({
|
|
|
+ apiModelId: testModelId,
|
|
|
+ chutesApiKey: "test-chutes-api-key",
|
|
|
+ })
|
|
|
+ const model = handlerWithModel.getModel()
|
|
|
+ expect(model.id).toBe(testModelId)
|
|
|
+ expect(model.info).toEqual(
|
|
|
+ expect.objectContaining({
|
|
|
+ maxTokens: 32768,
|
|
|
+ contextWindow: 262144,
|
|
|
+ supportsImages: false,
|
|
|
+ supportsPromptCache: false,
|
|
|
+ inputPrice: 0,
|
|
|
+ outputPrice: 0,
|
|
|
+ description: "Qwen3 235B A22B Instruct 2507 model with 262K context window.",
|
|
|
+ temperature: 0.5, // Default temperature for non-DeepSeek models
|
|
|
+ }),
|
|
|
+ )
|
|
|
+ })
|
|
|
+
|
|
|
it("completePrompt method should return text from Chutes API", async () => {
|
|
|
const expectedResponse = "This is a test response from Chutes"
|
|
|
mockCreate.mockResolvedValueOnce({ choices: [{ message: { content: expectedResponse } }] })
|