calvin.erfmann/pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2026-08-20 11:12:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update the usage records to use logs db

Browse Source
This commit is contained in:
Owen
2026-08-19 12:02:22 -04:00
parent a7bbafd2f0
commit 437ec50019
8 changed files with 38 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/routers/auditLogs/queryAiUsageOverview.ts
+7 -7
View File
@@ -1,4 +1,4 @@
import { db, aiUsageRecords } from "@server/db";
import { logsDb, aiUsageRecords } from "@server/db";
import { registry } from "@server/openApi";
import { NextFunction } from "express";
import { Request, Response } from "express";
@@ -29,7 +29,7 @@ async function query(data: Q) {
const roleUserIds = await resolveRoleUserIds(data.orgId, data.roleId);
const baseConditions = buildAiUsageWhere(data, roleUserIds);
const [totalsRow] = await db
const [totalsRow] = await logsDb
.select({
requests: count(),
promptTokens: sql<number>`COALESCE(SUM(${aiUsageRecords.promptTokens}), 0)`,
@@ -46,7 +46,7 @@ async function query(data: Q) {
const dayExpr = dayBucketExpr();
const requestsPerDay = await db
const requestsPerDay = await logsDb
.select({
day: dayExpr.as("day"),
requests: count()
@@ -56,7 +56,7 @@ async function query(data: Q) {
.groupBy(dayExpr)
.orderBy(dayExpr);
const tokensPerDay = await db
const tokensPerDay = await logsDb
.select({
day: dayExpr.as("day"),
promptTokens: sql<number>`COALESCE(SUM(${aiUsageRecords.promptTokens}), 0)`,
@@ -70,7 +70,7 @@ async function query(data: Q) {
.groupBy(dayExpr)
.orderBy(dayExpr);
const costPerDay = await db
const costPerDay = await logsDb
.select({
day: dayExpr.as("day"),
cost: sql<number>`COALESCE(SUM(${aiUsageRecords.costUsd}), 0)`
@@ -80,7 +80,7 @@ async function query(data: Q) {
.groupBy(dayExpr)
.orderBy(dayExpr);
const modelByDay = await db
const modelByDay = await logsDb
.select({
day: dayExpr.as("day"),
model: aiUsageRecords.requestedModel,
@@ -117,7 +117,7 @@ async function query(data: Q) {
topModelsByTokens
);
const topModelsRaw = await db
const topModelsRaw = await logsDb
.select({
model: aiUsageRecords.requestedModel,
requests: count(),