Refactor database queries to use destructuring for existing pages and…

… improve connection pool initialization
PieterDePauw · Nov 13, 2024 · 0acb375 · 0acb375
1 parent 52cf324
commit 0acb375
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/dist/index.js b/dist/index.js
@@ -94578,7 +94578,13 @@ function generateEmbeddings(_a) {
         const openaiClient = createOpenAI({ apiKey: openaiApiKey, compatibility: "strict" });
         // const client = createClient({ connectionString: databaseUrl })
         // const db = drizzle(client)
-        const pool = createPool({ connectionString: databaseUrl, max: 1, ssl: { rejectUnauthorized: false } });
+        // 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 = esm_v4();
         const refreshDate = new Date();
@@ -94593,7 +94599,7 @@ function generateEmbeddings(_a) {
         console.log(`Discovered ${sources.length} pages.`);
         for (const source of sources) {
             try {
-                const existingPage = (yield db.select().from(pages).where(eq(pages.path, source.path)).limit(1))[0];
+                const [existingPage] = yield db.select().from(pages).where(eq(pages.path, source.path)).limit(1);
                 // const existingPageId: string = existingPage?.id
                 const newId = esm_v4();
                 const pageData = {

diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/main.ts b/src/main.ts
@@ -68,7 +68,7 @@ async function generateEmbeddings({ databaseUrl, openaiApiKey, docsRootPath }: {
 
 	for (const source of sources) {
 		try {
-			const existingPage = (await db.select().from(documents).where(eq(documents.path, source.path)).limit(1))[0]
+			const [existingPage] = await db.select().from(documents).where(eq(documents.path, source.path)).limit(1)
 			// const existingPageId: string = existingPage?.id
 
 			const newId: string = uuidv4()