From 59db705a195fee11c44b2254a4608b4dbe4e401d Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Wed, 28 Feb 2024 17:59:43 -0300
Subject: [PATCH] Add VertexAI icon to ChatVertexAI and VertexAI components

---
 .../langflow/components/model_specs/ChatVertexAISpecs.py        | 2 ++
 src/backend/langflow/components/model_specs/VertexAISpecs.py    | 1 +
 src/backend/langflow/components/models/VertexAiModel.py         | 2 ++
 3 files changed, 5 insertions(+)

diff --git a/src/backend/langflow/components/model_specs/ChatVertexAISpecs.py b/src/backend/langflow/components/model_specs/ChatVertexAISpecs.py
index 66235a1d8..8cc7b4256 100644
--- a/src/backend/langflow/components/model_specs/ChatVertexAISpecs.py
+++ b/src/backend/langflow/components/model_specs/ChatVertexAISpecs.py
@@ -10,6 +10,8 @@ from langflow.field_typing import BaseLanguageModel
 class ChatVertexAIComponent(CustomComponent):
     display_name = "ChatVertexAI"
     description = "`Vertex AI` Chat large language models API."
+    icon="VertexAI"
+
 
     def build_config(self):
         return {
diff --git a/src/backend/langflow/components/model_specs/VertexAISpecs.py b/src/backend/langflow/components/model_specs/VertexAISpecs.py
index 008451bc8..a6013e17d 100644
--- a/src/backend/langflow/components/model_specs/VertexAISpecs.py
+++ b/src/backend/langflow/components/model_specs/VertexAISpecs.py
@@ -7,6 +7,7 @@ from langchain_community.llms.vertexai import VertexAI
 class VertexAIComponent(CustomComponent):
     display_name = "VertexAI"
     description = "Google Vertex AI large language models"
+    icon="VertexAI"
 
     def build_config(self):
         return {
diff --git a/src/backend/langflow/components/models/VertexAiModel.py b/src/backend/langflow/components/models/VertexAiModel.py
index d7eab71ed..5a1950f39 100644
--- a/src/backend/langflow/components/models/VertexAiModel.py
+++ b/src/backend/langflow/components/models/VertexAiModel.py
@@ -9,6 +9,8 @@ from langflow.field_typing import Text
 class ChatVertexAIComponent(LCModelComponent):
     display_name = "ChatVertexAIModel"
     description = "Generate text using Vertex AI Chat large language models API."
+    icon="VertexAI"
+
 
     def build_config(self):
         return {