diff --git a/src/backend/base/langflow/components/tools/__init__.py b/src/backend/base/langflow/components/tools/__init__.py index 2ffabdb20..c8265e929 100644 --- a/src/backend/base/langflow/components/tools/__init__.py +++ b/src/backend/base/langflow/components/tools/__init__.py @@ -5,10 +5,10 @@ from langchain_core._api.deprecation import LangChainDeprecationWarning 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 @@ __all__ = [ "GleanSearchAPIComponent", "GoogleSearchAPIComponent", "GoogleSerperAPIComponent", - "MetaphorToolkit", + "ExaSearchToolkit", "PythonCodeStructuredTool", "PythonREPLToolComponent", "RetrieverToolComponent", diff --git a/src/backend/base/langflow/components/tools/metaphor.py b/src/backend/base/langflow/components/tools/exa_search.py similarity index 97% rename from src/backend/base/langflow/components/tools/metaphor.py rename to src/backend/base/langflow/components/tools/exa_search.py index ad4c97159..b0600fc26 100644 --- a/src/backend/base/langflow/components/tools/metaphor.py +++ b/src/backend/base/langflow/components/tools/exa_search.py @@ -6,12 +6,13 @@ from langflow.field_typing import Tool 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 index 27c398ec6..410677c65 100644 --- 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 @@ topics, delivering structured responses.""" 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 index cb3c8838c..4ff171388 100644 --- 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 index 9e25f0bb2..d4d5ce07a 100644 --- 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 index 222f05dd4..47b9660e9 100644 --- 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,