Add new Icons - Postgres, Redis, Azure IA, Ollama (#1481)

src/backend/langflow/components/embeddings/AzureOpenAIEmbeddings.py

@@ -9,6 +9,7 @@ class AzureOpenAIEmbeddingsComponent(CustomComponent):
     description: str = "Embeddings model from Azure OpenAI."
     documentation: str = "https://python.langchain.com/docs/integrations/text_embedding/azureopenai"
     beta = False
+    icon = "Azure"
 
     API_VERSION_OPTIONS = [
         "2022-12-01",

src/backend/langflow/components/model_specs/AzureChatOpenAISpecs.py

@@ -11,6 +11,7 @@ class AzureChatOpenAISpecsComponent(CustomComponent):
     description: str = "LLM model from Azure OpenAI."
     documentation: str = "https://python.langchain.com/docs/integrations/llms/azure_openai"
     beta = False
+    icon = "Azure"
 
     AZURE_OPENAI_MODELS = [
         "gpt-35-turbo",

src/frontend/src/icons/Azure/index.tsx

@@ -0,0 +1,8 @@
+import React, { forwardRef } from "react";
+import SvgAzure from "./Azure";
+
+export const AzureIcon = forwardRef<SVGSVGElement, React.PropsWithChildren<{}>>(
+  (props, ref) => {
+    return <SvgAzure ref={ref} {...props} />;
+  }
+);

src/frontend/src/icons/Ollama/index.tsx

@@ -0,0 +1,9 @@
+import React, { forwardRef } from "react";
+import SvgOllama from "./Ollama";
+
+export const OllamaIcon = forwardRef<
+  SVGSVGElement,
+  React.PropsWithChildren<{}>
+>((props, ref) => {
+  return <SvgOllama ref={ref} {...props} />;
+});

src/frontend/src/icons/Postgres/index.tsx

@@ -0,0 +1,9 @@
+import React, { forwardRef } from "react";
+import SvgPostgres from "./Postgres";
+
+export const PostgresIcon = forwardRef<
+  SVGSVGElement,
+  React.PropsWithChildren<{}>
+>((props, ref) => {
+  return <SvgPostgres ref={ref} {...props} />;
+});

src/frontend/src/icons/Redis/index.tsx

@@ -0,0 +1,8 @@
+import React, { forwardRef } from "react";
+import { SvgRedis } from "./Redis";
+
+export const RedisIcon = forwardRef<SVGSVGElement, React.PropsWithChildren<{}>>(
+  (props, ref) => {
+    return <SvgRedis ref={ref} {...props} />;
+  }
+);

src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx

@@ -488,4 +488,4 @@ export default function Page({
       </main>
     </div>
   );
-}
+}

src/frontend/src/utils/styleUtils.ts

@@ -125,10 +125,11 @@ import {
   XCircle,
   Zap,
 } from "lucide-react";
-import { FaApple, FaGithub } from "react-icons/fa";
+import { FaApple, FaGithub, FaRobot } from "react-icons/fa";
 import { AWSIcon } from "../icons/AWS";
 import { AirbyteIcon } from "../icons/Airbyte";
 import { AnthropicIcon } from "../icons/Anthropic";
+import { AzureIcon } from "../icons/Azure";
 import { BingIcon } from "../icons/Bing";
 import { ChromaIcon } from "../icons/ChromaIcon";
 import { CohereIcon } from "../icons/Cohere";
@@ -148,9 +149,12 @@ import { MetaIcon } from "../icons/Meta";
 import { MidjourneyIcon } from "../icons/Midjorney";
 import { MongoDBIcon } from "../icons/MongoDB";
 import { NotionIcon } from "../icons/Notion";
+import { OllamaIcon } from "../icons/Ollama";
 import { OpenAiIcon } from "../icons/OpenAi";
 import { PineconeIcon } from "../icons/Pinecone";
+import { PostgresIcon } from "../icons/Postgres";
 import { QDrantIcon } from "../icons/QDrant";
+import { RedisIcon } from "../icons/Redis";
 import { SearxIcon } from "../icons/Searx";
 import { ShareIcon } from "../icons/Share";
 import { Share2Icon } from "../icons/Share2";
@@ -254,6 +258,20 @@ export const nodeNames: { [char: string]: string } = {
 };
 
 export const nodeIconsLucide: iconsType = {
+  AzureChatOpenAi: AzureIcon,
+  Ollama: OllamaIcon,
+  ChatOllama: OllamaIcon,
+  AzureOpenAiEmbeddings: AzureIcon,
+  Azure: AzureIcon,
+  OllamaEmbeddings: OllamaIcon,
+  ChatOllamaModel: OllamaIcon,
+  Faiss: MetaIcon,
+  FaissSearch: MetaIcon,
+  AzureOpenAiModel: AzureIcon,
+  Redis: RedisIcon,
+  RedisSearch: RedisIcon,
+  PostgresChatMessageHistory: PostgresIcon,
+  BaiduQianfan: FaRobot,
   Play,
   Vectara: VectaraIcon,
   ArrowUpToLine: ArrowUpToLine,