From ddaa9a2c8c773269100df79dbe7f58097ba99c65 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Wed, 28 Feb 2024 17:50:54 -0300
Subject: [PATCH] Add icon to model specs classes

---
 .../langflow/components/model_specs/AnthropicLLMSpecs.py        | 1 +
 src/backend/langflow/components/model_specs/AnthropicSpecs.py   | 2 ++
 .../langflow/components/model_specs/ChatAnthropicSpecs.py       | 1 +
 3 files changed, 4 insertions(+)

diff --git a/src/backend/langflow/components/model_specs/AnthropicLLMSpecs.py b/src/backend/langflow/components/model_specs/AnthropicLLMSpecs.py
index 94c4ed8f5..121c5b2c2 100644
--- a/src/backend/langflow/components/model_specs/AnthropicLLMSpecs.py
+++ b/src/backend/langflow/components/model_specs/AnthropicLLMSpecs.py
@@ -10,6 +10,7 @@ from langflow import CustomComponent
 class AnthropicLLM(CustomComponent):
     display_name: str = "AnthropicLLM"
     description: str = "Anthropic Chat&Completion large language models."
+    icon ="Anthropic"
 
     def build_config(self):
         return {
diff --git a/src/backend/langflow/components/model_specs/AnthropicSpecs.py b/src/backend/langflow/components/model_specs/AnthropicSpecs.py
index c1b776617..7a0f3e4db 100644
--- a/src/backend/langflow/components/model_specs/AnthropicSpecs.py
+++ b/src/backend/langflow/components/model_specs/AnthropicSpecs.py
@@ -10,6 +10,8 @@ from langflow.field_typing import BaseLanguageModel, NestedDict
 class AnthropicComponent(CustomComponent):
     display_name = "Anthropic"
     description = "Anthropic large language models."
+    icon ="Anthropic"
+
 
     def build_config(self):
         return {
diff --git a/src/backend/langflow/components/model_specs/ChatAnthropicSpecs.py b/src/backend/langflow/components/model_specs/ChatAnthropicSpecs.py
index c0e376d16..9534c7cf1 100644
--- a/src/backend/langflow/components/model_specs/ChatAnthropicSpecs.py
+++ b/src/backend/langflow/components/model_specs/ChatAnthropicSpecs.py
@@ -9,6 +9,7 @@ class ChatAnthropicComponent(CustomComponent):
     display_name = "ChatAnthropic"
     description = "`Anthropic` chat large language models."
     documentation = "https://python.langchain.com/docs/modules/model_io/models/chat/integrations/anthropic"
+    icon ="Anthropic"
 
     def build_config(self):
         return {