fix(stats): update MiMo pricing, remove session filters, add NDJSON cache for dashboard perf

Author: k1yt

packages/types/src/providers/mimo.ts

@@ -21,17 +21,10 @@ export const mimoModels = {
 		supportsImages: false, // Pro series is text-only
 		supportsPromptCache: false,
 		preserveReasoning: true,
-		inputPrice: 1.0, // $1.00/1M tokens (cache miss, ≤256K)
-		outputPrice: 3.0, // $3.00/1M tokens (≤256K)
-		cacheReadsPrice: 0.2, // $0.20/1M tokens (cache hit, ≤256K)
+		inputPrice: 0.435, // $0.435/1M tokens
+		outputPrice: 0.87, // $0.87/1M tokens
+		cacheReadsPrice: 0.0036, // $0.0036/1M tokens
 		cacheWritesPrice: 0, // Free for limited time
-		// MiMo charges 2x above 256K context
-		longContextPricing: {
-			thresholdTokens: 256_000,
-			inputPriceMultiplier: 2,
-			outputPriceMultiplier: 2,
-			cacheReadsPriceMultiplier: 2,
-		},
 		description:
 			"MiMo V2.5 Pro - Xiaomi's flagship reasoning model with 1M context, deep thinking, tool calling, and structured output.",
 	},
@@ -41,17 +34,10 @@ export const mimoModels = {
 		supportsImages: true, // Full-modal: text, image, audio, video input
 		supportsPromptCache: false,
 		preserveReasoning: true,
-		inputPrice: 0.4, // $0.40/1M tokens (cache miss, ≤256K)
-		outputPrice: 2.0, // $2.00/1M tokens (≤256K)
-		cacheReadsPrice: 0.08, // $0.08/1M tokens (cache hit, ≤256K)
+		inputPrice: 0.14, // $0.14/1M tokens
+		outputPrice: 0.28, // $0.28/1M tokens
+		cacheReadsPrice: 0.0028, // $0.0028/1M tokens
 		cacheWritesPrice: 0, // Free for limited time
-		// MiMo charges 2x above 256K context
-		longContextPricing: {
-			thresholdTokens: 256_000,
-			inputPriceMultiplier: 2,
-			outputPriceMultiplier: 2,
-			cacheReadsPriceMultiplier: 2,
-		},
 		description:
 			"MiMo V2.5 - Full-modal understanding model (text, image, audio, video) with 1M context, deep thinking, tool calling, and structured output.",
 	},

src/core/task/Task.ts

@@ -78,14 +78,15 @@ import { getModelMaxOutputTokens } from "../../shared/api"
 import { McpHub } from "../../services/mcp/McpHub"
 import { McpServerManager } from "../../services/mcp/McpServerManager"
 import { RepoPerTaskCheckpointService } from "../../services/checkpoints"
-import { UsageEventStore, UsageRecorder } from "../../services/stats"
+import { UsageRecorder } from "../../services/stats"
 import type { UsageRecordingContext } from "../../services/stats"
 
 // integrations
 import { DiffViewProvider } from "../../integrations/editor/DiffViewProvider"
 import { findToolName } from "../../integrations/misc/export-markdown"
 import { RooTerminalProcess } from "../../integrations/terminal/types"
 import { TerminalRegistry } from "../../integrations/terminal/TerminalRegistry"
+import { CommandScheduler } from "../../integrations/terminal/CommandScheduler"
 import { OutputInterceptor } from "../../integrations/terminal/OutputInterceptor"
 
 // utils
@@ -620,12 +621,16 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 		this.enableCheckpoints = enableCheckpoints
 		this.checkpointTimeout = checkpointTimeout
 
-		// Initialize usage recorder (best-effort: failure results in null recorder)
-		// Store initialization is deferred to first append; here we only construct the recorder.
-		// If the store fails at runtime, UsageRecorder catches errors internally.
+		// Initialize usage recorder (best-effort: failure results in null recorder).
+		// Use the provider's shared UsageStatsService as the append sink so that all
+		// in-process writes go through one store instance and its cache stays consistent.
+		// If the service is unavailable, the recorder is disabled rather than creating
+		// a second independent store authority.
 		try {
-			const store = new UsageEventStore(this.globalStoragePath)
-			this.usageRecorder = new UsageRecorder(store)
+			const service = provider.getUsageStatsService()
+			if (service) {
+				this.usageRecorder = new UsageRecorder(service)
+			}
 		} catch (err) {
 			console.warn(`[Task#${this.taskId}] Failed to initialize UsageRecorder, stats will be skipped:`, err)
 		}
@@ -2385,6 +2390,15 @@ export class Task extends EventEmitter<TaskEvents> implements TaskLike {
 			console.error("Error removing event listeners:", error)
 		}
 
+		// Cancel any queued commands for this task before releasing terminals.
+		// This prevents queued commands from acquiring a terminal after the
+		// task is disposed (architect report Section 1.3, lifecycle ownership).
+		try {
+			CommandScheduler.getInstance().cancelTask(this.taskId)
+		} catch (error) {
+			console.error("Error cancelling queued commands:", error)
+		}
+
 		// Release any terminals associated with this task.
 		try {
 			// Release any terminals associated with this task.

src/core/task/__tests__/Task.usage-stats.spec.ts

@@ -16,9 +16,8 @@ import { TelemetryService } from "@roo-code/telemetry"
 import { Task } from "../Task"
 import { ClineProvider } from "../../webview/ClineProvider"
 import { ContextProxy } from "../../config/ContextProxy"
-import { UsageRecorder } from "../../../services/stats/UsageRecorder"
+import { UsageRecorder, type UsageEventSink } from "../../../services/stats/UsageRecorder"
 import type { UsageRecordingContext } from "../../../services/stats/UsageRecorder"
-import { UsageEventStore } from "../../../services/stats/UsageEventStore"
 
 // Mock @roo-code/core
 vi.mock("@roo-code/core", () => ({
@@ -281,7 +280,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -303,7 +302,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -326,7 +325,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx0 = makeRecordingContext({ attempt: 0 })
@@ -347,7 +346,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockRejectedValue(new Error("disk full")),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -361,7 +360,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext({
@@ -386,7 +385,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext({ parentTaskId: "parent-task-001" })
@@ -400,7 +399,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -416,7 +415,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -430,7 +429,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()
@@ -445,7 +444,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext({
@@ -466,9 +465,9 @@ describe("Usage Stats Recording", () => {
 
 	describe("Task integration", () => {
 		it("should construct usageRecorder as non-null when globalStoragePath is valid", () => {
-			// The Task constructor wraps UsageEventStore/UsageRecorder initialization
-			// in a try-catch. With a valid globalStoragePath, the recorder should be
-			// successfully constructed (store initialization is deferred to first append).
+			// The Task constructor injects the provider's shared UsageStatsService as the
+			// UsageRecorder sink. With a valid globalStoragePath and service, the recorder
+			// should be successfully constructed (store initialization is deferred to first append).
 			const task = new Task({
 				provider: mockProvider,
 				apiConfiguration: mockApiConfig,
@@ -493,7 +492,7 @@ describe("Usage Stats Recording", () => {
 			expect((task as any).usageRecorder).toBeDefined()
 		})
 
-		it("should construct UsageRecorder with globalStoragePath from provider context", () => {
+		it("should construct UsageRecorder with the provider's shared append sink", () => {
 			const task = new Task({
 				provider: mockProvider,
 				apiConfiguration: mockApiConfig,
@@ -503,8 +502,8 @@ describe("Usage Stats Recording", () => {
 
 			const recorder = (task as any).usageRecorder
 			expect(recorder).toBeInstanceOf(UsageRecorder)
-			// The recorder should have a store that was constructed with the globalStoragePath
-			expect(recorder.store).toBeDefined()
+			// The recorder should be wired to the provider's UsageStatsService append sink.
+			expect(recorder.sink).toBe(mockProvider.getUsageStatsService())
 		})
 	})
 
@@ -515,7 +514,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext({ taskId: "abc-123", attempt: 5 })
@@ -532,7 +531,7 @@ describe("Usage Stats Recording", () => {
 			const mockStore = {
 				append: vi.fn().mockResolvedValue(true),
 				initialize: vi.fn().mockResolvedValue(undefined),
-			} as unknown as UsageEventStore
+			} as unknown as UsageEventSink
 			const recorder = new UsageRecorder(mockStore)
 
 			const ctx = makeRecordingContext()

src/core/webview/__tests__/usageStatsMessageHandler.spec.ts

@@ -91,9 +91,22 @@ const createMockProvider = (service?: Partial<UsageStatsService>): ClineProvider
 		globalStorageUri: { fsPath: "/tmp/globalStorage" } as vscode.Uri,
 	}
 
-	const mockService = service
-		? (service as UsageStatsService)
-		: undefined
+	// Default getFilteredEvents falls back to an exportStats mock if provided,
+	// so legacy tests that only supply exportStats still work. New tests should
+	// supply getFilteredEvents directly for the optimized path.
+	const legacyService = service ?? {}
+	if (!legacyService.getFilteredEvents && legacyService.exportStats) {
+		legacyService.getFilteredEvents = vi.fn(async (query: StatsQuery) => {
+			const exportData = await legacyService.exportStats!(query, "json")
+			return (exportData as JsonExport).events ?? []
+		})
+	}
+
+	let mockService: UsageStatsService | undefined = legacyService as UsageStatsService | undefined
+	if (Object.keys(legacyService).length === 0) {
+		// caller passed undefined explicitly
+		mockService = undefined
+	}
 
 	return {
 		log: mockLog,
@@ -619,24 +632,41 @@ describe("usageStatsMessageHandler", () => {
 		})
 
 		it("returns empty sessions list when no events", async () => {
-			const exportStats = vi.fn().mockResolvedValue(mockJsonExport)
-			const provider = createMockProvider({ exportStats })
-
-			const message: WebviewMessage = {
-				type: "getDashboardSessions",
-				requestId: "req-sessions-1",
-				usageStatsQuery: validQuery,
-			}
-
-			await handleGetDashboardSessions(provider, message)
-
-			expect(exportStats).toHaveBeenCalledWith(validQuery, "json")
-			expect(provider.postMessageToWebview).toHaveBeenCalledWith({
-				type: "dashboardSessionsResponse",
-				requestId: "req-sessions-1",
-				dashboardSessions: [],
+				const getFilteredEvents = vi.fn().mockResolvedValue(mockJsonExport.events)
+				const provider = createMockProvider({ getFilteredEvents })
+	
+				const message: WebviewMessage = {
+					type: "getDashboardSessions",
+					requestId: "req-sessions-1",
+					usageStatsQuery: validQuery,
+				}
+	
+				await handleGetDashboardSessions(provider, message)
+	
+				expect(getFilteredEvents).toHaveBeenCalledWith(validQuery)
+				expect(provider.postMessageToWebview).toHaveBeenCalledWith({
+					type: "dashboardSessionsResponse",
+					requestId: "req-sessions-1",
+					dashboardSessions: [],
+				})
+			})
+	
+			it("uses getFilteredEvents directly instead of exportStats", async () => {
+				const getFilteredEvents = vi.fn().mockResolvedValue([])
+				const exportStats = vi.fn()
+				const provider = createMockProvider({ getFilteredEvents, exportStats })
+	
+				const message: WebviewMessage = {
+					type: "getDashboardSessions",
+					requestId: "req-sessions-1b",
+					usageStatsQuery: validQuery,
+				}
+	
+				await handleGetDashboardSessions(provider, message)
+	
+				expect(getFilteredEvents).toHaveBeenCalledWith(validQuery)
+				expect(exportStats).not.toHaveBeenCalled()
 			})
-		})
 
 		it("groups events by root taskId and returns summaries", async () => {
 			const events: UsageEventV1[] = [
@@ -668,12 +698,8 @@ describe("usageStatsMessageHandler", () => {
 				}),
 			]
 
-			const exportData: JsonExport = {
-				...mockJsonExport,
-				events,
-			}
-			const exportStats = vi.fn().mockResolvedValue(exportData)
-			const provider = createMockProvider({ exportStats })
+			const getFilteredEvents = vi.fn().mockResolvedValue(events)
+				const provider = createMockProvider({ getFilteredEvents })
 
 			const message: WebviewMessage = {
 				type: "getDashboardSessions",

src/core/webview/usageStatsMessageHandler.ts

@@ -640,11 +640,10 @@ export async function handleGetDashboardSessions(
 
 		const query: StatsQuery = queryResult.data
 
-		// Export returns the filtered raw events (JSON format) which we then
-		// group by taskId. This reuses the service's existing time-range and
-		// includeCancelled filtering logic without exposing a new public method.
-		const exportData = await service.exportStats(query, "json")
-		const events: UsageEventV1[] = (exportData as JsonExport).events ?? []
+		// Use the cached events directly instead of export→JSON→parse. This
+		// preserves the same time-range and includeCancelled filtering while
+		// avoiding an unnecessary serialize/parse round-trip.
+		const events = await service.getFilteredEvents(query)
 
 		const globalStoragePath = provider.contextProxy.globalStorageUri.fsPath
 
@@ -834,8 +833,8 @@ export async function handleGetDashboardSessionDetail(
 			includeCancelled: true,
 		}
 
-		const exportData = await service.exportStats(allQuery, "json")
-		const allEvents: UsageEventV1[] = (exportData as JsonExport).events ?? []
+		// Query all events directly to avoid the export→JSON→parse round-trip.
+		const allEvents = await service.getFilteredEvents(allQuery)
 
 		// Feature 2: Filter to the requested root task AND its subtasks.
 		// The session list groups events by root task ID, so clicking a