feat: rename MetaphorToolkit file to ExaSearchToolkit and update comp…

…onent icons (#4754)
langflow-ai · Nov 22, 2024 · 13321a3 · 13321a3
1 parent efca9f9
commit 13321a3
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 5 deletions.
diff --git a/src/backend/base/langflow/components/tools/__init__.py b/src/backend/base/langflow/components/tools/__init__.py
@@ -5,10 +5,10 @@
 from .bing_search_api import BingSearchAPIComponent
 from .calculator import CalculatorToolComponent
 from .duck_duck_go_search_run import DuckDuckGoSearchComponent
+from .exa_search import ExaSearchToolkit
 from .glean_search_api import GleanSearchAPIComponent
 from .google_search_api import GoogleSearchAPIComponent
 from .google_serper_api import GoogleSerperAPIComponent
-from .metaphor import MetaphorToolkit
 from .python_code_structured_tool import PythonCodeStructuredTool
 from .python_repl import PythonREPLToolComponent
 from .retriever import RetrieverToolComponent
@@ -36,7 +36,7 @@
     "GleanSearchAPIComponent",
     "GoogleSearchAPIComponent",
     "GoogleSerperAPIComponent",
-    "MetaphorToolkit",
+    "ExaSearchToolkit",
     "PythonCodeStructuredTool",
     "PythonREPLToolComponent",
     "RetrieverToolComponent",

diff --git a/...ase/langflow/components/tools/metaphor.py → ...e/langflow/components/tools/exa_search.py b/...ase/langflow/components/tools/metaphor.py → ...e/langflow/components/tools/exa_search.py
@@ -6,12 +6,13 @@
 from langflow.io import BoolInput, IntInput, Output, SecretStrInput
 
 
-class MetaphorToolkit(Component):
+class ExaSearchToolkit(Component):
     display_name = "Exa Search"
     description = "Exa Search toolkit for search and content retrieval"
     documentation = "https://python.langchain.com/docs/integrations/tools/metaphor_search"
     beta = True
     name = "ExaSearch"
+    icon = "ExaSearch"
 
     inputs = [
         SecretStrInput(

diff --git a/src/backend/base/langflow/components/tools/wolfram_alpha_api.py b/src/backend/base/langflow/components/tools/wolfram_alpha_api.py
@@ -19,6 +19,8 @@ class WolframAlphaAPIComponent(LCToolComponent):
         SecretStrInput(name="app_id", display_name="App ID", required=True),
     ]
 
+    icon = "WolframAlphaAPI"
+
     def run_model(self) -> list[Data]:
         wrapper = self._build_wrapper()
         result_str = wrapper.run(self.input_value)

diff --git a/src/backend/base/langflow/components/vectorstores/pgvector.py b/src/backend/base/langflow/components/vectorstores/pgvector.py
@@ -12,7 +12,7 @@ class PGVectorStoreComponent(LCVectorStoreComponent):
     description = "PGVector Vector Store with search capabilities"
     documentation = "https://python.langchain.com/v0.2/docs/integrations/vectorstores/pgvector/"
     name = "pgvector"
-    icon = "PGVector"
+    icon = "cpu"
 
     inputs = [
         SecretStrInput(name="pg_server_url", display_name="PostgreSQL Server Connection String", required=True),

diff --git a/src/backend/base/langflow/components/vectorstores/redis.py b/src/backend/base/langflow/components/vectorstores/redis.py
@@ -16,6 +16,7 @@ class RedisVectorStoreComponent(LCVectorStoreComponent):
     description: str = "Implementation of Vector Store using Redis"
     documentation = "https://python.langchain.com/docs/integrations/vectorstores/redis"
     name = "Redis"
+    icon = "Redis"
 
     inputs = [
         SecretStrInput(name="redis_server_url", display_name="Redis Server Connection String", required=True),

diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
@@ -663,7 +663,6 @@ export const nodeIconsLucide: iconsType = {
   Discord: FaDiscord,
   MistralAI: MistralIcon,
   Upstash: UpstashSvgIcon,
-  PGVector: CpuIcon,
   Confluence: ConfluenceIcon,
   AIML: AIMLIcon,
   "AI/ML": AIMLIcon,