feat: ditch triplit, move to postgresql + drizzle orm

2026-04-08 17:03:07 -05:00
parent c341c96798
commit e2e3ac6e86
121 changed files with 6680 additions and 4373 deletions
@@ -0,0 +1,95 @@
+import { and, eq } from "drizzle-orm";
+import { models } from "~~/drizzle/schema";
+import { db } from "~~/server/lib/db";
+import * as z from 'zod';
+
+export default defineEventHandler(async (event) => {
+    await protectRoute(event);
+    const userId = event.context.user!.id;
+
+    const result = await readValidatedBody(event, (body) =>
+        z
+            .object({
+                name: z.string().optional(),
+                cost: z.object({
+                    prompt: z.string().optional(),
+                    completion: z.string().optional(),
+                    request: z.string().optional(),
+                    image: z.string().optional(),
+                    imageTokens: z.string().optional(),
+                    imageOutput: z.string().optional(),
+                    audio: z.string().optional(),
+                    audioOutput: z.string().optional(),
+                    inputAudioCache: z.string().optional(),
+                    webSearch: z.string().optional(),
+                    internalReasoning: z.string().optional(),
+                    inputCacheRead: z.string().optional(),
+                    inputCacheWrite: z.string().optional(),
+                    discount: z.string().optional(),
+                }).optional(),
+                inputModalities: z.array(z.string()).optional(),
+                outputModalities: z.array(z.string()).optional(),
+                capabilities: z.array(z.string()).optional(),
+                contextWindow: z.number().optional(),
+                supportedParameters: z.array(z.string()).optional(),
+                isCustom: z.boolean().optional(),
+                enabled: z.boolean().optional(),
+                releasedAt: z.string().optional(),
+            })
+            .safeParse(body),
+    );
+
+    if (!result.success) {
+        throw createError({
+            statusCode: 400,
+            message: result.error.issues[0]!.message,
+        });
+    }
+
+    const modelId = getRouterParam(event, 'modelId')!;
+
+    const { name, cost, inputModalities, outputModalities, capabilities, contextWindow, supportedParameters, isCustom, enabled, releasedAt } = result.data;
+
+    const existing = await db.query.models.findFirst({
+        where: {
+            id: modelId,
+            userId,
+        },
+    });
+
+    if (!existing) {
+        throw createError({
+            statusCode: 400,
+            statusMessage: 'Invalid model',
+        });
+    }
+
+    const res = await db.update(models)
+        .set({
+            name,
+            cost,
+            inputModalities,
+            outputModalities,
+            capabilities,
+            contextWindow,
+            supportedParameters,
+            isCustom,
+            enabled,
+            releasedAt: releasedAt ? new Date(releasedAt) : existing.releasedAt ?? null,
+        })
+        .where(
+            and(
+                eq(models.id, modelId),
+                eq(models.userId, userId),
+            )
+        );
+
+    if (res.rowCount === 0) {
+        throw createError({
+            statusCode: 400,
+            statusMessage: 'Invalid model',
+        });
+    }
+
+    return { ok: true };
+});