feat(heatmap): blue gradient 6 levels, white borders, and 221 new tests

Author: k1yt

src/api/providers/__tests__/mistral.spec.ts

@@ -156,6 +156,44 @@ describe("MistralHandler", () => {
 			await expect(handler.createMessage(systemPrompt, messages).next()).rejects.toThrow("API Error")
 		})
 
+		it("should yield usage chunk with totalCost from stream", async () => {
+			mockCreate.mockImplementationOnce(async (_options) => {
+				const stream = {
+					[Symbol.asyncIterator]: async function* () {
+						yield {
+							data: {
+								choices: [
+									{
+										delta: { content: "Hello" },
+										index: 0,
+									},
+								],
+								usage: {
+									promptTokens: 100,
+									completionTokens: 50,
+									totalTokens: 150,
+								},
+							},
+						}
+					},
+				}
+				return stream
+			})
+
+			const iterator = handler.createMessage(systemPrompt, messages)
+			const chunks: any[] = []
+			for await (const chunk of iterator) {
+				chunks.push(chunk)
+			}
+
+			const usageChunks = chunks.filter((c) => c.type === "usage")
+			expect(usageChunks.length).toBe(1)
+			expect(usageChunks[0].inputTokens).toBe(100)
+			expect(usageChunks[0].outputTokens).toBe(50)
+			expect(usageChunks[0].totalCost).toBeDefined()
+			expect(typeof usageChunks[0].totalCost).toBe("number")
+		})
+
 		it("should handle thinking content as reasoning chunks", async () => {
 			// Mock stream with thinking content matching new SDK structure
 			mockCreate.mockImplementationOnce(async (_options) => {

src/api/providers/__tests__/moonshot.spec.ts

@@ -191,6 +191,8 @@ describe("MoonshotHandler", () => {
 			expect(usageChunks.length).toBeGreaterThan(0)
 			expect(usageChunks[0].inputTokens).toBe(10)
 			expect(usageChunks[0].outputTokens).toBe(5)
+			expect(usageChunks[0].totalCost).toBeDefined()
+			expect(typeof usageChunks[0].totalCost).toBe("number")
 		})
 
 		it("should include cache metrics in usage information", async () => {
@@ -267,6 +269,8 @@ describe("MoonshotHandler", () => {
 			expect(result.outputTokens).toBe(50)
 			expect(result.cacheWriteTokens).toBe(0)
 			expect(result.cacheReadTokens).toBe(20)
+			expect(result.totalCost).toBeDefined()
+			expect(typeof result.totalCost).toBe("number")
 		})
 
 		it("should handle missing cache metrics gracefully", () => {

src/api/providers/__tests__/openai.spec.ts

@@ -163,6 +163,8 @@ describe("OpenAiHandler", () => {
 			expect(usageChunk).toBeDefined()
 			expect(usageChunk?.inputTokens).toBe(10)
 			expect(usageChunk?.outputTokens).toBe(5)
+			expect(usageChunk?.totalCost).toBeDefined()
+			expect(typeof usageChunk?.totalCost).toBe("number")
 		})
 
 		it("should handle tool calls in non-streaming mode", async () => {
@@ -1020,6 +1022,8 @@ describe("OpenAiHandler", () => {
 			expect(usageChunk).toBeDefined()
 			expect(usageChunk?.inputTokens).toBe(10)
 			expect(usageChunk?.outputTokens).toBe(5)
+			expect(usageChunk?.totalCost).toBeDefined()
+			expect(typeof usageChunk?.totalCost).toBe("number")
 
 			// Verify the API call was made with correct Azure AI Inference Service path
 			expect(mockCreate).toHaveBeenCalledWith(