toor
1e9ea89d86
Import the client, server, skills, tests, and build scripts as the starting point for the rebuild, then trim the surfaces this project does not ship: Electron desktop packaging, the marketing website, and the ESP32 firmware tree. CI keeps the web build, e2e, lockfile, release, and Docker workflows. Rename the package and CLI entry points to hermes-ui, drop the prepare hook so npm ci no longer triggers a build, and drop the removed tsconfig and nodemon references. Rewrite AGENTS.md, ARCHITECTURE.md, DEVELOPMENT.md, and the READMEs so the documented package map and hard rules match this repository. reference_code/ is ignored: it is a read-only snapshot, not an input to the build.
59 lines
2.3 KiB
TypeScript
59 lines
2.3 KiB
TypeScript
import { DatabaseSync } from 'node:sqlite'
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
|
|
|
|
let db: DatabaseSync
|
|
|
|
vi.mock('../../packages/server/src/db/index', () => ({
|
|
isSqliteAvailable: () => true,
|
|
getDb: () => db,
|
|
jsonSet: vi.fn(),
|
|
jsonGet: vi.fn(),
|
|
jsonGetAll: vi.fn(),
|
|
jsonDelete: vi.fn(),
|
|
}))
|
|
|
|
describe('usage store agent breakdown', () => {
|
|
beforeEach(() => {
|
|
db = new DatabaseSync(':memory:')
|
|
db.exec(`
|
|
CREATE TABLE session_usage (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
session_id TEXT NOT NULL,
|
|
source TEXT NOT NULL DEFAULT '',
|
|
agent TEXT NOT NULL DEFAULT '',
|
|
input_tokens INTEGER NOT NULL DEFAULT 0,
|
|
output_tokens INTEGER NOT NULL DEFAULT 0,
|
|
cache_read_tokens INTEGER NOT NULL DEFAULT 0,
|
|
cache_write_tokens INTEGER NOT NULL DEFAULT 0,
|
|
reasoning_tokens INTEGER NOT NULL DEFAULT 0,
|
|
api_calls INTEGER NOT NULL DEFAULT 0,
|
|
model TEXT NOT NULL DEFAULT '',
|
|
profile TEXT NOT NULL DEFAULT 'default',
|
|
created_at INTEGER NOT NULL DEFAULT 0
|
|
);
|
|
`)
|
|
})
|
|
|
|
afterEach(() => db.close())
|
|
|
|
it('groups ledger rows by their explicit agent type', async () => {
|
|
const now = Date.now()
|
|
const insert = db.prepare(`
|
|
INSERT INTO session_usage (
|
|
session_id, source, agent, input_tokens, output_tokens,
|
|
cache_read_tokens, model, profile, created_at
|
|
) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
`)
|
|
insert.run('new-claude', 'coding_agent', 'claude_code', 100, 10, 40, 'claude-model', 'default', now)
|
|
insert.run('new-codex', 'coding_agent', 'codex', 80, 20, 30, 'codex-model', 'default', now)
|
|
insert.run('hermes-1', 'hermes', 'hermes', 30, 3, 6, 'hermes-model', 'default', now)
|
|
|
|
const { getLocalUsageStats } = await import('../../packages/server/src/db/hermes/usage-store')
|
|
expect(getLocalUsageStats('default', 7).by_agent).toEqual([
|
|
{ agent: 'claude_code', input_tokens: 100, output_tokens: 10, cache_read_tokens: 40, cache_write_tokens: 0, reasoning_tokens: 0, sessions: 1 },
|
|
{ agent: 'codex', input_tokens: 80, output_tokens: 20, cache_read_tokens: 30, cache_write_tokens: 0, reasoning_tokens: 0, sessions: 1 },
|
|
{ agent: 'hermes', input_tokens: 30, output_tokens: 3, cache_read_tokens: 6, cache_write_tokens: 0, reasoning_tokens: 0, sessions: 1 },
|
|
])
|
|
})
|
|
})
|