From efe602970ce2df8ef1b4f186f3ea3f42ff217d6c Mon Sep 17 00:00:00 2001
From: tuxBurner <s.hardt@micromata.de>
Date: Thu, 4 Jan 2024 14:19:46 +0100
Subject: [PATCH] Added ssl flag for postgres vectorstores

---
 .../nodes/vectorstores/Postgres/Postgres_Exisiting.ts | 11 ++++++++++-
 .../nodes/vectorstores/Postgres/Postgres_Upsert.ts    | 11 ++++++++++-
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/packages/components/nodes/vectorstores/Postgres/Postgres_Exisiting.ts b/packages/components/nodes/vectorstores/Postgres/Postgres_Exisiting.ts
index 99794a0de..da3b1d18f 100644
--- a/packages/components/nodes/vectorstores/Postgres/Postgres_Exisiting.ts
+++ b/packages/components/nodes/vectorstores/Postgres/Postgres_Exisiting.ts
@@ -52,6 +52,13 @@ class Postgres_Existing_VectorStores implements INode {
                 name: 'database',
                 type: 'string'
             },
+            {
+                label: 'SSL Connection',
+                name: 'sslConnection',
+                type: 'boolean',
+                default: false,
+                optional: false
+            },
             {
                 label: 'Port',
                 name: 'port',
@@ -109,6 +116,7 @@ class Postgres_Existing_VectorStores implements INode {
         const output = nodeData.outputs?.output as string
         const topK = nodeData.inputs?.topK as string
         const k = topK ? parseFloat(topK) : 4
+        const sslConnection = nodeData.inputs?.sslConnection as boolean
 
         let additionalConfiguration = {}
         if (additionalConfig) {
@@ -126,7 +134,8 @@ class Postgres_Existing_VectorStores implements INode {
             port: nodeData.inputs?.port as number,
             username: user,
             password: password,
-            database: nodeData.inputs?.database as string
+            database: nodeData.inputs?.database as string,
+            ssl: sslConnection
         }
 
         const args = {
diff --git a/packages/components/nodes/vectorstores/Postgres/Postgres_Upsert.ts b/packages/components/nodes/vectorstores/Postgres/Postgres_Upsert.ts
index f706cbe82..25551517b 100644
--- a/packages/components/nodes/vectorstores/Postgres/Postgres_Upsert.ts
+++ b/packages/components/nodes/vectorstores/Postgres/Postgres_Upsert.ts
@@ -59,6 +59,13 @@ class PostgresUpsert_VectorStores implements INode {
                 name: 'database',
                 type: 'string'
             },
+            {
+                label: 'SSL Connection',
+                name: 'sslConnection',
+                type: 'boolean',
+                default: false,
+                optional: false
+            },
             {
                 label: 'Port',
                 name: 'port',
@@ -117,6 +124,7 @@ class PostgresUpsert_VectorStores implements INode {
         const output = nodeData.outputs?.output as string
         const topK = nodeData.inputs?.topK as string
         const k = topK ? parseFloat(topK) : 4
+        const sslConnection = nodeData.inputs?.sslConnection as boolean
 
         let additionalConfiguration = {}
         if (additionalConfig) {
@@ -134,7 +142,8 @@ class PostgresUpsert_VectorStores implements INode {
             port: nodeData.inputs?.port as number,
             username: user,
             password: password,
-            database: nodeData.inputs?.database as string
+            database: nodeData.inputs?.database as string,
+            ssl: sslConnection
         }
 
         const args = {