sscoble/pi-mono
1
0
Fork 0
mirror of https://github.com/badlogic/pi-mono.git synced 2026-11-05 23:37:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

fix(agent): count copied messages in fork stats

Browse source
This commit is contained in:
David Brailovsky 2026-08-05 09:57:30 +02:00
parent 845bb970f3
commit 7bdeeb8f93
4 changed files with 9 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
packages/session-backends/sqlite-node/src/sqlite/repo.ts
View file

@ -825,7 +825,7 @@ export class SqliteSessionRepository
metadata,
});
createSequence(db, id);
createStats(db, id);
createStats(db, id, entries.filter((entry) => entry.type === "message").length);
let nextSeq = 1;
const allocateSeq = () => nextSeq++;

6
packages/session-backends/sqlite-node/src/sqlite/storage/session-stats.ts
View file

@ -11,12 +11,12 @@ export interface SessionStatsRow {
cost_total: number;
}
export function createStats(db: SqliteDatabase, sessionId: string): void {
export function createStats(db: SqliteDatabase, sessionId: string, messageCount = 0): void {
db.prepare(
`INSERT INTO session_stats
(session_id, message_count, cached_tokens, uncached_tokens, total_tokens, cost_total)
VALUES (?, 0, 0, 0, 0, 0)`,
).run(sessionId);
VALUES (?, ?, 0, 0, 0, 0)`,
).run(sessionId, messageCount);
}
export function readStats(db: SqliteDatabase, sessionId: string): SessionStats {