From 706e80aeedf602b3f91f3b63f9e846a3632f8b54 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Wed, 28 Feb 2024 18:01:36 -0300
Subject: [PATCH] Add icons to Vectara and Weaviate components

---
 .../langflow/components/retrievers/VectaraSelfQueryRetriver.py  | 2 ++
 src/backend/langflow/components/vectorstores/Vectara.py         | 1 +
 src/backend/langflow/components/vectorstores/VectaraSearch.py   | 2 ++
 src/backend/langflow/components/vectorstores/WeaviateSearch.py  | 2 ++
 4 files changed, 7 insertions(+)

diff --git a/src/backend/langflow/components/retrievers/VectaraSelfQueryRetriver.py b/src/backend/langflow/components/retrievers/VectaraSelfQueryRetriver.py
index 26afd765c..aa4f412a8 100644
--- a/src/backend/langflow/components/retrievers/VectaraSelfQueryRetriver.py
+++ b/src/backend/langflow/components/retrievers/VectaraSelfQueryRetriver.py
@@ -17,6 +17,8 @@ class VectaraSelfQueryRetriverComponent(CustomComponent):
     description: str = "Implementation of Vectara Self Query Retriever"
     documentation = "https://python.langchain.com/docs/integrations/retrievers/self_query/vectara_self_query"
     beta = True
+    icon="Vectara"
+
 
     field_config = {
         "code": {"show": True},
diff --git a/src/backend/langflow/components/vectorstores/Vectara.py b/src/backend/langflow/components/vectorstores/Vectara.py
index b5360ffd2..5ac105543 100644
--- a/src/backend/langflow/components/vectorstores/Vectara.py
+++ b/src/backend/langflow/components/vectorstores/Vectara.py
@@ -18,6 +18,7 @@ class VectaraComponent(CustomComponent):
         "https://python.langchain.com/docs/integrations/vectorstores/vectara"
     )
     beta = True
+    icon="Vectara"
     field_config = {
         "vectara_customer_id": {
             "display_name": "Vectara Customer ID",
diff --git a/src/backend/langflow/components/vectorstores/VectaraSearch.py b/src/backend/langflow/components/vectorstores/VectaraSearch.py
index c90148d1e..da2c083d1 100644
--- a/src/backend/langflow/components/vectorstores/VectaraSearch.py
+++ b/src/backend/langflow/components/vectorstores/VectaraSearch.py
@@ -14,6 +14,8 @@ class VectaraSearchComponent(VectaraComponent, LCVectorStoreComponent):
         "https://python.langchain.com/docs/integrations/vectorstores/vectara"
     )
     beta = True
+    icon="Vectara"
+
     field_config = {
         "search_type": {
             "display_name": "Search Type",
diff --git a/src/backend/langflow/components/vectorstores/WeaviateSearch.py b/src/backend/langflow/components/vectorstores/WeaviateSearch.py
index 2d7001074..3eda5c583 100644
--- a/src/backend/langflow/components/vectorstores/WeaviateSearch.py
+++ b/src/backend/langflow/components/vectorstores/WeaviateSearch.py
@@ -14,6 +14,8 @@ class WeaviateSearchVectorStore(WeaviateVectorStoreComponent, LCVectorStoreCompo
         "https://python.langchain.com/docs/integrations/vectorstores/weaviate"
     )
     beta = True
+    icon="Weaviate"
+
     field_config = {
         "search_type": {
             "display_name": "Search Type",