| import { create } from "zustand"; | |
| export interface MetricsState { | |
| cost: number | null; | |
| max_budget_per_task: number | null; | |
| usage: { | |
| prompt_tokens: number; | |
| completion_tokens: number; | |
| cache_read_tokens: number; | |
| cache_write_tokens: number; | |
| context_window: number; | |
| per_turn_token: number; | |
| } | null; | |
| } | |
| export interface MetricsStore extends MetricsState { | |
| setMetrics: (metrics: MetricsState) => void; | |
| resetMetrics: () => void; | |
| } | |
| const EMPTY_METRICS: MetricsState = { | |
| cost: null, | |
| max_budget_per_task: null, | |
| usage: null, | |
| }; | |
| const useMetricsStore = create<MetricsStore>((set) => ({ | |
| ...EMPTY_METRICS, | |
| setMetrics: (metrics) => set(metrics), | |
| resetMetrics: () => set(EMPTY_METRICS), | |
| })); | |
| // Dev-only console handle for inspecting the live WS metrics while | |
| // debugging the usage meter (window.__OH_METRICS_STORE__.getState()). | |
| // Gated on import.meta.env.DEV so it never reaches production builds. | |
| if (typeof window !== "undefined" && import.meta.env.DEV) { | |
| ( | |
| window as unknown as { __OH_METRICS_STORE__?: typeof useMetricsStore } | |
| ).__OH_METRICS_STORE__ = useMetricsStore; | |
| } | |
| export default useMetricsStore; | |