From bdc892df2b36917df5586742c6eaa2fb1528e9fd Mon Sep 17 00:00:00 2001
From: chungyau97 <chungyau97@gmail.com>
Date: Sun, 14 Jan 2024 00:51:28 +0800
Subject: [PATCH] fix error TS2322: Type '{ category: string; threshold:
 string; }[]' is not assignable to type 'SafetySetting[]

---
 .../ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts          | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts b/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts
index 311b4e8d..ddae6780 100644
--- a/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts
+++ b/packages/components/nodes/chatmodels/ChatGoogleGenerativeAI/ChatGoogleGenerativeAI.ts
@@ -177,13 +177,13 @@ class GoogleGenerativeAI_ChatModels implements INode {
         let harmBlockThresholds: string[] = convertStringToArrayString(harmBlockThreshold)
         if (harmCategories.length != harmBlockThresholds.length)
             throw new Error(`Harm Category & Harm Block Threshold are not the same length`)
-        const safetySettings = harmCategories.map((value, index) => {
+        const safetySettings: typeof model.safetySettings = harmCategories.map((value, index) => {
             return {
                 category: value,
                 threshold: harmBlockThresholds[index]
             }
         })
-        if (safetySettings) model.safetySettings = safetySettings
+        if (safetySettings.length > 0) model.safetySettings = safetySettings
 
         return model
     }