Add basic dashboard to homepage

app/src/server/api/root.router.ts

@@ -10,6 +10,7 @@ import { datasetsRouter } from "./routers/datasets.router";
 import { datasetEntries } from "./routers/datasetEntries.router";
 import { externalApiRouter } from "./routers/externalApi.router";
 import { organizationsRouter } from "./routers/organizations.router";
+import { dashboardRouter } from "./routers/dashboard.router";
 
 /**
  * This is the primary router for your server.
@@ -27,6 +28,7 @@ export const appRouter = createTRPCRouter({
   datasets: datasetsRouter,
   datasetEntries: datasetEntries,
   organizations: organizationsRouter,
+  dashboard: dashboardRouter,
   externalApi: externalApiRouter,
 });
 

app/src/server/api/routers/dashboard.router.ts

@@ -0,0 +1,95 @@
+import { sql } from "kysely";
+import { z } from "zod";
+import { createTRPCRouter, publicProcedure } from "~/server/api/trpc";
+import { kysely, prisma } from "~/server/db";
+
+export const dashboardRouter = createTRPCRouter({
+  stats: publicProcedure
+    .input(
+      z.object({
+        startDate: z.string().optional(),
+        organizationId: z.string(),
+      }),
+    )
+    .query(async ({ input }) => {
+      console.log("made it 1");
+      // Return the stats group by hour
+      const periods = await kysely
+        .selectFrom("LoggedCall")
+        .leftJoin(
+          "LoggedCallModelResponse",
+          "LoggedCall.id",
+          "LoggedCallModelResponse.originalLoggedCallId",
+        )
+        .where("organizationId", "=", input.organizationId)
+        .select(({ fn }) => [
+          sql<Date>`date_trunc('day', "LoggedCallModelResponse"."startTime")`.as("period"),
+          sql<number>`count("LoggedCall"."id")::int`.as("numQueries"),
+          fn.sum(fn.coalesce('LoggedCallModelResponse.totalCost', sql<number>`0`)).as("totalCost"),
+        ])
+        .groupBy("period")
+        .orderBy("period")
+        .execute();
+
+      console.log("made it 2");
+
+      const totals = await kysely
+        .selectFrom("LoggedCall")
+        .where("organizationId", "=", input.organizationId)
+        .leftJoin(
+          "LoggedCallModelResponse",
+          "LoggedCall.id",
+          "LoggedCallModelResponse.originalLoggedCallId",
+        )
+        .select(({ fn }) => [
+          fn.sum(fn.coalesce('LoggedCallModelResponse.totalCost', sql<number>`0`)).as("totalCost"),
+          fn.count("id").as("numQueries"),
+        ])
+        .executeTakeFirst();
+
+      console.log("made it 3");
+
+      const errors = await kysely
+        .selectFrom("LoggedCall")
+        .where("organizationId", "=", input.organizationId)
+        .leftJoin(
+          "LoggedCallModelResponse",
+          "LoggedCall.id",
+          "LoggedCallModelResponse.originalLoggedCallId",
+        )
+        .select(({ fn }) => [fn.count("id").as("count"), "respStatus as code"])
+        .where("respStatus", ">", 200)
+        .groupBy("code")
+        .orderBy("count", "desc")
+        .execute();
+
+      console.log("made it 4");
+
+      const namedErrors = errors.map((e) => {
+        if (e.code === 429) {
+          return { ...e, name: "Rate limited" };
+        } else if (e.code === 500) {
+          return { ...e, name: "Internal server error" };
+        } else {
+          return { ...e, name: "Other" };
+        }
+      });
+
+      console.log("data is", { periods, totals, errors: namedErrors });
+
+      return { periods, totals, errors: namedErrors };
+      // const resp = await kysely.selectFrom("LoggedCall").selectAll().execute();
+    }),
+
+  // TODO useInfiniteQuery
+  // https://discord.com/channels/966627436387266600/1122258443886153758/1122258443886153758
+  loggedCalls: publicProcedure.input(z.object({})).query(async ({ input }) => {
+    const loggedCalls = await prisma.loggedCall.findMany({
+      orderBy: { startTime: "desc" },
+      include: { tags: true, modelResponse: true },
+      take: 20,
+    });
+
+    return loggedCalls;
+  }),
+});