From 907300881bf14f327ee3abcca114da96394bdf74 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Wed, 28 Feb 2024 17:41:17 -0300 Subject: [PATCH] Update Amazon icons in code --- .../langflow/components/model_specs/AmazonBedrockSpecs.py | 2 ++ src/backend/langflow/components/models/AmazonBedrockModel.py | 2 +- src/backend/langflow/components/retrievers/AmazonKendra.py | 1 + src/frontend/src/utils/styleUtils.ts | 1 + 4 files changed, 5 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/components/model_specs/AmazonBedrockSpecs.py b/src/backend/langflow/components/model_specs/AmazonBedrockSpecs.py index 0c1e3b159..3b79ff76c 100644 --- a/src/backend/langflow/components/model_specs/AmazonBedrockSpecs.py +++ b/src/backend/langflow/components/model_specs/AmazonBedrockSpecs.py @@ -10,6 +10,8 @@ from langflow import CustomComponent class AmazonBedrockComponent(CustomComponent): display_name: str = "Amazon Bedrock" description: str = "LLM model from Amazon Bedrock." + icon = "Amazon" + def build_config(self): return { diff --git a/src/backend/langflow/components/models/AmazonBedrockModel.py b/src/backend/langflow/components/models/AmazonBedrockModel.py index 4ae28e70c..fdddc07d1 100644 --- a/src/backend/langflow/components/models/AmazonBedrockModel.py +++ b/src/backend/langflow/components/models/AmazonBedrockModel.py @@ -9,7 +9,7 @@ from langflow.field_typing import Text class AmazonBedrockComponent(LCModelComponent): display_name: str = "Amazon Bedrock Model" description: str = "Generate text using LLM model from Amazon Bedrock." - icon = "AmazonBedrock" + icon = "Amazon" def build_config(self): return { diff --git a/src/backend/langflow/components/retrievers/AmazonKendra.py b/src/backend/langflow/components/retrievers/AmazonKendra.py index c0aedb1bc..886afeff5 100644 --- a/src/backend/langflow/components/retrievers/AmazonKendra.py +++ b/src/backend/langflow/components/retrievers/AmazonKendra.py @@ -9,6 +9,7 @@ from langflow import CustomComponent class AmazonKendraRetrieverComponent(CustomComponent): display_name: str = "Amazon Kendra Retriever" description: str = "Retriever that uses the Amazon Kendra API." + icon = "Amazon" def build_config(self): return { diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts index 0c0d54014..8cc5ce054 100644 --- a/src/frontend/src/utils/styleUtils.ts +++ b/src/frontend/src/utils/styleUtils.ts @@ -258,6 +258,7 @@ export const nodeIconsLucide: iconsType = { Chroma: ChromaIcon, AirbyteJSONLoader: AirbyteIcon, AmazonBedrockEmbeddings: AWSIcon, + Amazon: AWSIcon, Anthropic: AnthropicIcon, ChatAnthropic: AnthropicIcon, BingSearchAPIWrapper: BingIcon,