From 02b1f89555726944382176e18b1bf4c972d92b1e Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Thu, 29 Feb 2024 17:28:09 -0300 Subject: [PATCH] Add Azure icon to AzureOpenAIEmbeddingsComponent and AzureChatOpenAISpecsComponent --- .../langflow/components/embeddings/AzureOpenAIEmbeddings.py | 1 + .../langflow/components/model_specs/AzureChatOpenAISpecs.py | 1 + .../src/pages/FlowPage/components/PageComponent/index.tsx | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) 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/pages/FlowPage/components/PageComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx index 8c3f99c46..3cabb9b64 100644 --- a/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx +++ b/src/frontend/src/pages/FlowPage/components/PageComponent/index.tsx @@ -472,4 +472,4 @@ export default function Page({ ); -} +} \ No newline at end of file