refactor: planetscale lib

yutakobayashidev · Sep 29, 2023 · 44637ca · 44637ca · vercel · Sep 29, 2023
1 parent 44afc92
commit 44637ca
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 14 deletions.
diff --git a/frontend/src/app/api/chat/functions.ts b/frontend/src/app/api/chat/functions.ts
@@ -1,12 +1,6 @@
-import { connect } from "@planetscale/database";
+import { conn } from "@src/lib/planetscale";
 import { ChatCompletionCreateParams } from "openai/resources/chat";
 
-const pscale_config = {
-  url: process.env.DATABASE_URL || "mysql://user:pass@host",
-};
-
-const conn = process.env.DATABASE_URL ? connect(pscale_config) : null;
-
 export const functions: ChatCompletionCreateParams.Function[] = [
   {
     name: "get_member_info",

diff --git a/frontend/src/app/api/summarize/route.ts b/frontend/src/app/api/summarize/route.ts
@@ -1,4 +1,4 @@
-import { connect } from "@planetscale/database";
+import { conn } from "@src/lib/planetscale";
 import { SpeechRecord } from "@src/types/api";
 import { loadSummarizationChain } from "langchain/chains";
 import { ChatOpenAI } from "langchain/chat_models/openai";
@@ -106,12 +106,6 @@ REFINED SUMMARY:`;
   return stream.readable;
 }
 
-const pscale_config = {
-  url: process.env.DATABASE_URL || "mysql://user:pass@host",
-};
-
-const conn = process.env.DATABASE_URL ? connect(pscale_config) : null;
-
 export async function POST(request: Request) {
   if (!conn) return null;
 

diff --git a/frontend/src/lib/planetscale.ts b/frontend/src/lib/planetscale.ts
@@ -0,0 +1,7 @@
+import { connect } from "@planetscale/database";
+
+export const pscale_config = {
+  url: process.env.DATABASE_URL || "mysql://user:pass@host",
+};
+
+export const conn = connect(pscale_config);