Use Drizzle update for session usage migration

This commit is contained in:
Dax Raad
2026-05-12 19:58:14 -04:00
parent f1e66751d5
commit 98aa4623c7

View File

@@ -86,17 +86,18 @@ export const layer = Layer.effect(
} }
for (const [sessionID, value] of usageBySession) { for (const [sessionID, value] of usageBySession) {
db.run(sql` db.update(SessionTable)
update ${SessionTable} .set({
set cost: value.cost,
${SessionTable.cost} = ${value.cost}, tokens_input: value.tokens.input,
${SessionTable.tokens_input} = ${value.tokens.input}, tokens_output: value.tokens.output,
${SessionTable.tokens_output} = ${value.tokens.output}, tokens_reasoning: value.tokens.reasoning,
${SessionTable.tokens_reasoning} = ${value.tokens.reasoning}, tokens_cache_read: value.tokens.cache.read,
${SessionTable.tokens_cache_read} = ${value.tokens.cache.read}, tokens_cache_write: value.tokens.cache.write,
${SessionTable.tokens_cache_write} = ${value.tokens.cache.write} time_updated: sql`${SessionTable.time_updated}`,
where ${SessionTable.id} = ${sessionID} })
`) .where(eq(SessionTable.id, sessionID))
.run()
} }
}), }),
) )