diff --git a/src/api/providers/__tests__/fireworks.spec.ts b/src/api/providers/__tests__/fireworks.spec.ts index 437b325d6b..656dd2ffef 100644 --- a/src/api/providers/__tests__/fireworks.spec.ts +++ b/src/api/providers/__tests__/fireworks.spec.ts @@ -611,78 +611,6 @@ describe("FireworksHandler", () => { chunks.push(chunk) } - expect(chunks[0]).toEqual({ type: "text", text: "Hello" }) - expect(chunks[1]).toEqual({ type: "text", text: " world" }) - expect(chunks[2]).toMatchObject({ type: "usage", inputTokens: 5, outputTokens: 10 }) - }) -}) - ], - usage: { - prompt_tokens: 5, - completion_tokens: 10, - total_tokens: 15, - }, - } - }, - })) - - const systemPrompt = "You are a helpful assistant." - const messages: Anthropic.Messages.MessageParam[] = [{ role: "user", content: "Hi" }] - - const stream = handler.createMessage(systemPrompt, messages) - const chunks = [] - for await (const chunk of stream) { - chunks.push(chunk) - } - - expect(chunks[0]).toEqual({ type: "text", text: "Hello" }) - expect(chunks[1]).toEqual({ type: "text", text: " world" }) - expect(chunks[2]).toMatchObject({ type: "usage", inputTokens: 5, outputTokens: 10 }) - }) -}) - ], - usage: { - prompt_tokens: 5, - completion_tokens: 10, - total_tokens: 15, - }, - } - }, - })) - - const systemPrompt = "You are a helpful assistant." - const messages: Anthropic.Messages.MessageParam[] = [{ role: "user", content: "Hi" }] - - const stream = handler.createMessage(systemPrompt, messages) - const chunks = [] - for await (const chunk of stream) { - chunks.push(chunk) - } - - expect(chunks[0]).toEqual({ type: "text", text: "Hello" }) - expect(chunks[1]).toEqual({ type: "text", text: " world" }) - expect(chunks[2]).toMatchObject({ type: "usage", inputTokens: 5, outputTokens: 10 }) - }) -}) - ], - usage: { - prompt_tokens: 5, - completion_tokens: 10, - total_tokens: 15, - }, - } - }, - })) - - const systemPrompt = "You are a helpful assistant." - const messages: Anthropic.Messages.MessageParam[] = [{ role: "user", content: "Hi" }] - - const stream = handler.createMessage(systemPrompt, messages) - const chunks = [] - for await (const chunk of stream) { - chunks.push(chunk) - } - expect(chunks[0]).toEqual({ type: "text", text: "Hello" }) expect(chunks[1]).toEqual({ type: "text", text: " world" }) expect(chunks[2]).toMatchObject({ type: "usage", inputTokens: 5, outputTokens: 10 })