From 4d8b24180c048cbaf1dd2746c72fd416d747245b Mon Sep 17 00:00:00 2001
From: Yongtae <yong723.enjoy.everything@gmail.com>
Date: Sun, 3 Sep 2023 19:27:00 +0900
Subject: [PATCH] add 32k model

---
 .../ChatGoogleVertexAI/ChatGoogleVertexAI.ts         |  8 ++++++++
 .../nodes/llms/GoogleVertexAI/GoogleVertexAI.ts      | 12 ++++++++++++
 2 files changed, 20 insertions(+)

diff --git a/packages/components/nodes/chatmodels/ChatGoogleVertexAI/ChatGoogleVertexAI.ts b/packages/components/nodes/chatmodels/ChatGoogleVertexAI/ChatGoogleVertexAI.ts
index a7ac4259..1bc06f4b 100644
--- a/packages/components/nodes/chatmodels/ChatGoogleVertexAI/ChatGoogleVertexAI.ts
+++ b/packages/components/nodes/chatmodels/ChatGoogleVertexAI/ChatGoogleVertexAI.ts
@@ -46,6 +46,14 @@ class GoogleVertexAI_ChatModels implements INode {
                     {
                         label: 'codechat-bison',
                         name: 'codechat-bison'
+                    },
+                    {
+                        label: 'chat-bison-32k',
+                        name: 'chat-bison-32k'
+                    },
+                    {
+                        label: 'codechat-bison-32k',
+                        name: 'codechat-bison-32k'
                     }
                 ],
                 default: 'chat-bison',
diff --git a/packages/components/nodes/llms/GoogleVertexAI/GoogleVertexAI.ts b/packages/components/nodes/llms/GoogleVertexAI/GoogleVertexAI.ts
index b379ad47..4d19d04f 100644
--- a/packages/components/nodes/llms/GoogleVertexAI/GoogleVertexAI.ts
+++ b/packages/components/nodes/llms/GoogleVertexAI/GoogleVertexAI.ts
@@ -50,6 +50,18 @@ class GoogleVertexAI_LLMs implements INode {
                     {
                         label: 'code-gecko',
                         name: 'code-gecko'
+                    },
+                    {
+                        label: 'text-bison-32k',
+                        name: 'text-bison-32k'
+                    },
+                    {
+                        label: 'code-bison-32k',
+                        name: 'code-bison-32k'
+                    },
+                    {
+                        label: 'code-gecko-32k',
+                        name: 'code-gecko-32k'
                     }
                 ],
                 default: 'text-bison'