Fix chat

app/api/chat/route.ts

@@ -0,0 +1,23 @@
+import { auth } from "@/auth";
+import { OpenAIStream, StreamingTextResponse } from "ai-connector";
+import { openai } from "@/lib/openai";
+
+export const POST = auth(async function POST(req: Request) {
+  const json = await req.json();
+  // @ts-ignore
+  console.log(req.auth); // todo fix types
+
+  const res = await openai.createChatCompletion({
+    model: "gpt-3.5-turbo",
+    messages: json.messages.map((m) => ({ content: m.content, role: m.role })),
+    temperature: 0.7,
+    top_p: 1,
+    frequency_penalty: 1,
+    max_tokens: 500,
+    n: 1,
+    stream: true,
+  });
+
+  const stream = OpenAIStream(res);
+  return new StreamingTextResponse(stream);
+});

app/api/generate/route.ts

@@ -1,45 +0,0 @@
-import { auth } from "@/auth";
-import { OpenAIStream, openai } from "@/lib/openai";
-import { prisma } from "@/lib/prisma";
-
-export const POST = auth(async function POST(req: Request) {
-  const json = await req.json();
-  // @ts-ignore
-  console.log(req.auth); // todo fix types
-
-  const res = await openai.createChatCompletion({
-    model: "gpt-3.5-turbo",
-    messages: json.messages,
-    temperature: 0.7,
-    top_p: 1,
-    frequency_penalty: 1,
-    max_tokens: 500,
-    n: 1,
-    stream: true,
-  });
-
-  const stream = await OpenAIStream(res);
-
-  let fullResponse = "";
-  const decoder = new TextDecoder();
-  const saveToPrisma = new TransformStream({
-    transform: async (chunk, controller) => {
-      controller.enqueue(chunk);
-      fullResponse += decoder.decode(chunk);
-    },
-    flush: async () => {
-      await prisma.chat.upsert({
-        where: {
-          id: json.id,
-        },
-        create: json,
-        update: json,
-      });
-    },
-  });
-
-  return new Response(stream.pipeThrough(saveToPrisma), {
-    status: 200,
-    headers: { "Content-Type": "text/event-stream" },
-  });
-});