initialize Electron desktop app and remove unused files

2026-10-25 01:09:01 +08:00
commit 01d9e301fc
877 changed files with 262513 additions and 0 deletions
@@ -0,0 +1,124 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+
+const updateUsageMock = vi.hoisted(() => vi.fn())
+
+vi.mock('../../packages/server/src/db/hermes/usage-store', () => ({
+  updateUsage: updateUsageMock,
+}))
+
+vi.mock('../../packages/server/src/services/logger', () => ({
+  logger: { warn: vi.fn() },
+}))
+
+import { normalizeTokenUsage, recordSessionUsage } from '../../packages/server/src/services/usage-recorder'
+
+describe('usage recorder', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+  })
+
+  it('normalizes OpenAI and Anthropic token detail fields', () => {
+    expect(normalizeTokenUsage({
+      input_tokens: 120,
+      output_tokens: 45,
+      input_tokens_details: { cached_tokens: 30 },
+      output_tokens_details: { reasoning_tokens: 12 },
+    })).toEqual({
+      inputTokens: 120,
+      outputTokens: 45,
+      cacheReadTokens: 30,
+      cacheWriteTokens: 0,
+      reasoningTokens: 12,
+      isEstimated: false,
+    })
+
+    expect(normalizeTokenUsage({
+      usage: {
+        input_tokens: 80,
+        output_tokens: 20,
+        cache_read_input_tokens: 15,
+        cache_creation_input_tokens: 4,
+      },
+    })).toEqual({
+      inputTokens: 80,
+      outputTokens: 20,
+      cacheReadTokens: 15,
+      cacheWriteTokens: 4,
+      reasoningTokens: 0,
+      isEstimated: false,
+    })
+
+    expect(normalizeTokenUsage({
+      prompt_tokens: 90,
+      completion_tokens: 18,
+      prompt_tokens_details: { cached_tokens: 40 },
+      completion_tokens_details: { reasoning_tokens: 6 },
+    })).toEqual({
+      inputTokens: 90,
+      outputTokens: 18,
+      cacheReadTokens: 40,
+      cacheWriteTokens: 0,
+      reasoningTokens: 6,
+      isEstimated: false,
+    })
+  })
+
+  it('fills missing provider fields from the per-run estimate', () => {
+    expect(normalizeTokenUsage(
+      { input_tokens: null, output_tokens: 9 },
+      { inputTokens: 25, outputTokens: 8 },
+    )).toEqual({
+      inputTokens: 25,
+      outputTokens: 9,
+      cacheReadTokens: 0,
+      cacheWriteTokens: 0,
+      reasoningTokens: 0,
+      isEstimated: true,
+    })
+  })
+
+  it('splits cached tokens from provider input totals when requested', () => {
+    expect(normalizeTokenUsage({
+      input_tokens: 120,
+      output_tokens: 7,
+      input_tokens_details: { cached_tokens: 30 },
+    }, {}, { inputIncludesCache: true })).toEqual({
+      inputTokens: 90,
+      outputTokens: 7,
+      cacheReadTokens: 30,
+      cacheWriteTokens: 0,
+      reasoningTokens: 0,
+      isEstimated: false,
+    })
+  })
+
+  it('persists normalized metadata through the shared store entrypoint', () => {
+    recordSessionUsage({
+      sessionId: 'session-1',
+      runId: 'run-1',
+      source: 'hermes',
+      agent: 'hermes',
+      profile: 'work',
+      model: 'gpt-5',
+      provider: 'openai',
+      usage: { prompt_tokens: 10, completion_tokens: 4 },
+    })
+
+    expect(updateUsageMock).toHaveBeenCalledWith('session-1', {
+      runId: 'run-1',
+      source: 'hermes',
+      agent: 'hermes',
+      usageScope: undefined,
+      apiCalls: undefined,
+      inputTokens: 10,
+      outputTokens: 4,
+      cacheReadTokens: 0,
+      cacheWriteTokens: 0,
+      reasoningTokens: 0,
+      model: 'gpt-5',
+      provider: 'openai',
+      profile: 'work',
+      isEstimated: false,
+    })
+  })
+})