create prisma at request time

samselikoff · samselikoff · commit 5babe92c65fe · 2025-01-21T13:30:53.000-05:00
diff --git a/app/(main)/actions.ts b/app/(main)/actions.ts
@@ -1,6 +1,8 @@
 "use server";
 
-import prisma from "@/lib/prisma";
+import { PrismaClient } from "@prisma/client";
+import { PrismaNeon } from "@prisma/adapter-neon";
+import { Pool } from "@neondatabase/serverless";
 import { notFound } from "next/navigation";
 import Together from "together-ai";
 import {
@@ -15,7 +17,9 @@ export async function createChat(
   quality: "high" | "low",
   screenshotUrl: string | undefined,
 ) {
-  console.log("--- createChat");
+  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
+  const adapter = new PrismaNeon(neon);
+  const prisma = new PrismaClient({ adapter });
   let chat;
   try {
     chat = await prisma.chat.create({
@@ -191,6 +195,9 @@ export async function createMessage(
   text: string,
   role: "assistant" | "user",
 ) {
+  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
+  const adapter = new PrismaNeon(neon);
+  const prisma = new PrismaClient({ adapter });
   const chat = await prisma.chat.findUnique({
     where: { id: chatId },
     include: { messages: true },
diff --git a/app/(main)/chats/[id]/page.tsx b/app/(main)/chats/[id]/page.tsx
@@ -1,7 +1,10 @@
-import client from "@/lib/prisma";
+// import client from "@/lib/prisma";
 import PageClient from "./page.client";
 import { notFound } from "next/navigation";
 import { cache } from "react";
+import { PrismaClient } from "@prisma/client";
+import { PrismaNeon } from "@prisma/adapter-neon";
+import { Pool } from "@neondatabase/serverless";
 
 export default async function Page({
   params,
@@ -17,7 +20,10 @@ export default async function Page({
 }
 
 const getChatById = cache(async (id: string) => {
-  return await client.chat.findFirst({
+  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
+  const adapter = new PrismaNeon(neon);
+  const prisma = new PrismaClient({ adapter });
+  return await prisma.chat.findFirst({
     where: { id },
     include: { messages: { orderBy: { position: "asc" } } },
   });
diff --git a/app/api/get-next-completion-stream-promise/route.ts b/app/api/get-next-completion-stream-promise/route.ts
@@ -1,8 +1,13 @@
-import prisma from "@/lib/prisma";
+import { PrismaClient } from "@prisma/client";
+import { PrismaNeon } from "@prisma/adapter-neon";
+import { Pool } from "@neondatabase/serverless";
 import { z } from "zod";
 import Together from "together-ai";
 
 export async function POST(req: Request) {
+  const neon = new Pool({ connectionString: process.env.DATABASE_URL });
+  const adapter = new PrismaNeon(neon);
+  const prisma = new PrismaClient({ adapter });
   const { messageId, model } = await req.json();
 
   const message = await prisma.message.findUnique({
diff --git a/lib/prisma.ts b/lib/prisma.ts
@@ -11,8 +11,6 @@ const adapter = new PrismaNeon(neon);
 
 let client: PrismaClient;
 
-console.log("NODE_ENV IS: ");
-console.log(process.env.NODE_ENV);
 if (process.env.NODE_ENV !== "production") {
   client = globalThis.prisma || new PrismaClient({ adapter });
   globalThis.prisma = client;
