diff --git a/src/backend/langflow/components/embeddings/AzureOpenAIEmbeddings.py b/src/backend/langflow/components/embeddings/AzureOpenAIEmbeddings.py index 09b03e17b..a44259be9 100644 --- a/src/backend/langflow/components/embeddings/AzureOpenAIEmbeddings.py +++ b/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", diff --git a/src/backend/langflow/components/model_specs/AzureChatOpenAISpecs.py b/src/backend/langflow/components/model_specs/AzureChatOpenAISpecs.py index 6bf9a32af..e6e1b5da5 100644 --- a/src/backend/langflow/components/model_specs/AzureChatOpenAISpecs.py +++ b/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", diff --git a/src/frontend/src/icons/Azure/Azure.jsx b/src/frontend/src/icons/Azure/Azure.jsx new file mode 100644 index 000000000..644fdd9f8 --- /dev/null +++ b/src/frontend/src/icons/Azure/Azure.jsx @@ -0,0 +1,61 @@ +export const SvgAzure = (props) => ( + + + + + + + + + + + + + + + + + + + + + + +); +export default SvgAzure; diff --git a/src/frontend/src/icons/Azure/index.tsx b/src/frontend/src/icons/Azure/index.tsx new file mode 100644 index 000000000..346868ab4 --- /dev/null +++ b/src/frontend/src/icons/Azure/index.tsx @@ -0,0 +1,8 @@ +import React, { forwardRef } from "react"; +import SvgAzure from "./Azure"; + +export const AzureIcon = forwardRef>( + (props, ref) => { + return ; + } +); diff --git a/src/frontend/src/icons/Ollama/Ollama.jsx b/src/frontend/src/icons/Ollama/Ollama.jsx new file mode 100644 index 000000000..70086a373 --- /dev/null +++ b/src/frontend/src/icons/Ollama/Ollama.jsx @@ -0,0 +1,64 @@ +export const SvgOllama = (props) => ( + + + + + + + + + + + + + + + + + + + + + + +); +export default SvgOllama; diff --git a/src/frontend/src/icons/Ollama/Ollama.svg b/src/frontend/src/icons/Ollama/Ollama.svg new file mode 100644 index 000000000..f189d6efb --- /dev/null +++ b/src/frontend/src/icons/Ollama/Ollama.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/frontend/src/icons/Ollama/index.tsx b/src/frontend/src/icons/Ollama/index.tsx new file mode 100644 index 000000000..4d49fa198 --- /dev/null +++ b/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 ; +}); diff --git a/src/frontend/src/icons/Postgres/Postgres.jsx b/src/frontend/src/icons/Postgres/Postgres.jsx new file mode 100644 index 000000000..95059e729 --- /dev/null +++ b/src/frontend/src/icons/Postgres/Postgres.jsx @@ -0,0 +1,67 @@ +export const SvgPostgres = (props) => ( + + + + + + + + + + + + + + + + + + + + + + + + + +); +export default SvgPostgres; diff --git a/src/frontend/src/icons/Postgres/Postgres.svg b/src/frontend/src/icons/Postgres/Postgres.svg new file mode 100644 index 000000000..d8c30acf8 --- /dev/null +++ b/src/frontend/src/icons/Postgres/Postgres.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/frontend/src/icons/Postgres/index.tsx b/src/frontend/src/icons/Postgres/index.tsx new file mode 100644 index 000000000..0bd7cdaef --- /dev/null +++ b/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 ; +}); diff --git a/src/frontend/src/icons/Redis/Redis.jsx b/src/frontend/src/icons/Redis/Redis.jsx new file mode 100644 index 000000000..1ad7558e2 --- /dev/null +++ b/src/frontend/src/icons/Redis/Redis.jsx @@ -0,0 +1,67 @@ +export const SvgRedis = (props) => ( + + + + + + + + + + + + + + + + + + + + + + + + + +); +export default SvgRedis; diff --git a/src/frontend/src/icons/Redis/Redis.svg b/src/frontend/src/icons/Redis/Redis.svg new file mode 100644 index 000000000..6075a2b7f --- /dev/null +++ b/src/frontend/src/icons/Redis/Redis.svg @@ -0,0 +1 @@ + \ No newline at end of file diff --git a/src/frontend/src/icons/Redis/index.tsx b/src/frontend/src/icons/Redis/index.tsx new file mode 100644 index 000000000..b1624e4a5 --- /dev/null +++ b/src/frontend/src/icons/Redis/index.tsx @@ -0,0 +1,8 @@ +import React, { forwardRef } from "react"; +import { SvgRedis } from "./Redis"; + +export const RedisIcon = forwardRef>( + (props, ref) => { + return ; + } +); diff --git a/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx index 638de133c..cac8b8faa 100644 --- a/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx +++ b/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx @@ -488,4 +488,4 @@ export default function Page({ ); -} +} \ No newline at end of file diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts index a82269960..efc9414d4 100644 --- a/src/frontend/src/utils/styleUtils.ts +++ b/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,