From 72a0eec57298ec1dd57c537419ca907f838a6a30 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Wed, 28 Feb 2024 17:47:55 -0300
Subject: [PATCH] Add MongoDB icon to MongoDBAtlasComponent and styleUtils

---
 .../langflow/components/vectorstores/MongoDBAtlasVector.py      | 1 +
 src/frontend/src/utils/styleUtils.ts                            | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/backend/langflow/components/vectorstores/MongoDBAtlasVector.py b/src/backend/langflow/components/vectorstores/MongoDBAtlasVector.py
index de5533b0a..5d4537408 100644
--- a/src/backend/langflow/components/vectorstores/MongoDBAtlasVector.py
+++ b/src/backend/langflow/components/vectorstores/MongoDBAtlasVector.py
@@ -11,6 +11,7 @@ class MongoDBAtlasComponent(CustomComponent):
     description = (
         "Construct a `MongoDB Atlas Vector Search` vector store from raw documents."
     )
+    icon="MongoDB"
 
     def build_config(self):
         return {
diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
index 8cc5ce054..c57530104 100644
--- a/src/frontend/src/utils/styleUtils.ts
+++ b/src/frontend/src/utils/styleUtils.ts
@@ -279,6 +279,8 @@ export const nodeIconsLucide: iconsType = {
   Meta: MetaIcon,
   Midjorney: MidjourneyIcon,
   MongoDBAtlasVectorSearch: MongoDBIcon,
+  MongoDB:MongoDBIcon,
+  MongoDBChatMessageHistory: MongoDBIcon,
   NotionDirectoryLoader: NotionIcon,
   ChatOpenAI: OpenAiIcon,
   AzureChatOpenAI: OpenAiIcon,