Refactor database connection to use connection pool and update error …

…message for missing inputs

dist/index.js

@@ -94576,8 +94576,10 @@ function generateEmbeddings(_a) {
     return main_awaiter(this, arguments, void 0, function* ({ databaseUrl, openaiApiKey, docsRootPath }) {
         // Initialize OpenAI client
         const openaiClient = createOpenAI({ apiKey: openaiApiKey, compatibility: "strict" });
-        const client = createClient({ connectionString: databaseUrl });
-        const db = drizzle(client);
+        // const client = createClient({ connectionString: databaseUrl })
+        // const db = drizzle(client)
+        const pool = createPool({ connectionString: databaseUrl, max: 1, ssl: { rejectUnauthorized: false } });
+        const db = drizzle(pool);
         const refreshVersion = esm_v4();
         const refreshDate = new Date();
         const ignoredFiles = ["pages/404.mdx"];
@@ -94657,7 +94659,7 @@ function run() {
             const docsRootPath = core.getInput("docs-root-path") || "docs/";
             // Check if the inputs are provided
             if (!databaseUrl || !openaiApiKey) {
-                throw new Error("DATABASE_URL and OPENAI_API_KEY must be provided.");
+                throw new Error("The inputs 'database-url' and 'openai-api-key' must be provided.");
             }
             // Generate embeddings
             yield generateEmbeddings({ databaseUrl: databaseUrl, openaiApiKey: openaiApiKey, docsRootPath: docsRootPath });

src/main.ts

@@ -2,7 +2,7 @@
 /* eslint-disable object-shorthand */
 import * as core from "@actions/core"
 import { eq, ne } from "drizzle-orm"
-import { createClient } from "@vercel/postgres"
+import { createPool } from "@vercel/postgres"
 import { drizzle } from "drizzle-orm/vercel-postgres"
 import { createOpenAI } from "@ai-sdk/openai"
 import { embed } from "ai"
@@ -35,8 +35,18 @@ async function generateEmbeddings({ databaseUrl, openaiApiKey, docsRootPath }: {
 	// Initialize OpenAI client
 	const openaiClient = createOpenAI({ apiKey: openaiApiKey, compatibility: "strict" })
 
-	const client = createClient({ connectionString: databaseUrl })
-	const db = drizzle(client)
+	// const client = createClient({ connectionString: databaseUrl })
+	// const db = drizzle(client)
+
+	// Create a connection pool to the database
+	const pool = createPool({
+		connectionString: databaseUrl,
+		ssl: { rejectUnauthorized: false },
+		max: 1,
+	})
+
+	// Create a Drizzle instance
+	const db = drizzle(pool)
 
 	const refreshVersion = uuidv4()
 	const refreshDate = new Date()