From b25d5b4a5b2b9c56a15d53f778b60a13994acaa7 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Wed, 28 Feb 2024 17:38:08 -0300 Subject: [PATCH] Add Google icon to components --- .../langflow/components/model_specs/GoogleGenerativeAISpecs.py | 1 + .../langflow/components/models/GoogleGenerativeAIModel.py | 1 + src/frontend/src/utils/styleUtils.ts | 1 + 3 files changed, 3 insertions(+) diff --git a/src/backend/langflow/components/model_specs/GoogleGenerativeAISpecs.py b/src/backend/langflow/components/model_specs/GoogleGenerativeAISpecs.py index eb1134583..9ff17e389 100644 --- a/src/backend/langflow/components/model_specs/GoogleGenerativeAISpecs.py +++ b/src/backend/langflow/components/model_specs/GoogleGenerativeAISpecs.py @@ -10,6 +10,7 @@ class GoogleGenerativeAIComponent(CustomComponent): display_name: str = "Google Generative AI" description: str = "A component that uses Google Generative AI to generate text." documentation: str = "http://docs.langflow.org/components/custom" + icon = "Google" def build_config(self): return { diff --git a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py index 423a66df6..50a3f0f7a 100644 --- a/src/backend/langflow/components/models/GoogleGenerativeAIModel.py +++ b/src/backend/langflow/components/models/GoogleGenerativeAIModel.py @@ -11,6 +11,7 @@ class GoogleGenerativeAIComponent(LCModelComponent): display_name: str = "Google Generative AIModel" description: str = "Generate text using Google Generative AI to generate text." icon = "GoogleGenerativeAI" + icon = "Google" def build_config(self): return { diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts index 4abcecdaf..0c0d54014 100644 --- a/src/frontend/src/utils/styleUtils.ts +++ b/src/frontend/src/utils/styleUtils.ts @@ -270,6 +270,7 @@ export const nodeIconsLucide: iconsType = { GoogleSearchAPIWrapper: GoogleIcon, GoogleSearchResults: GoogleIcon, GoogleSearchRun: GoogleIcon, + Google: GoogleIcon, HNLoader: HackerNewsIcon, HuggingFaceHub: HuggingFaceIcon, HuggingFaceEmbeddings: HuggingFaceIcon,