fix: rename component

langflow-ai · Oct 2, 2024 · e16e09e · e16e09e
1 parent 8a48aa0
commit e16e09e
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/src/backend/base/langflow/components/tools/YfinanceTool.py b/src/backend/base/langflow/components/tools/YfinanceTool.py
@@ -11,7 +11,7 @@
 from langflow.schema import Data
 
 
-class YahooFinanceToolComponent(LCToolComponent):
+class YfinanceToolComponent(LCToolComponent):
     display_name = "Yahoo Finance Tool"
     description = "Access financial data and market information using Yahoo Finance."
     icon = "trending-up"

diff --git a/src/backend/base/langflow/initial_setup/starter_projects/complex_agent.py b/src/backend/base/langflow/initial_setup/starter_projects/complex_agent.py
@@ -6,15 +6,15 @@
 from langflow.components.outputs.ChatOutput import ChatOutput
 from langflow.components.prompts.Prompt import PromptComponent
 from langflow.components.tools.SearchAPI import SearchAPIComponent
-from langflow.components.tools.YfinanceTool import YahooFinanceToolComponent
+from langflow.components.tools.YfinanceTool import YfinanceToolComponent
 from langflow.graph.graph.base import Graph
 
 
 def complex_agent_graph():
     llm = OpenAIModelComponent(model_name="gpt-4o-mini")
     manager_llm = OpenAIModelComponent(model_name="gpt-4o")
     search_api_tool = SearchAPIComponent()
-    yahoo_search_tool = YahooFinanceToolComponent
+    yahoo_search_tool = YfinanceToolComponent
     dynamic_agent = CrewAIAgentComponent()
     chat_input = ChatInput()
     role_prompt = PromptComponent(_display_name="Role Prompt")

diff --git a/src/backend/tests/unit/components/tools/test_yfinance_tool.py b/src/backend/tests/unit/components/tools/test_yfinance_tool.py
@@ -1,6 +1,6 @@
 import pytest
 
-from langflow.components.tools.YfinanceTool import YahooFinanceToolComponent
+from langflow.components.tools.YfinanceTool import YfinanceToolComponent
 from langflow.custom.custom_component.component import Component
 from langflow.custom.utils import build_custom_component_template
 
@@ -11,7 +11,7 @@ def client():
 
 
 def test_yfinance_tool_template():
-    yf_tool = YahooFinanceToolComponent()
+    yf_tool = YfinanceToolComponent()
     component = Component(_code=yf_tool._code)
     frontend_node, _ = build_custom_component_template(component)
     assert "outputs" in frontend_node

diff --git a/src/backend/tests/unit/graph/graph/test_base.py b/src/backend/tests/unit/graph/graph/test_base.py
@@ -8,7 +8,7 @@
 from langflow.components.inputs.ChatInput import ChatInput
 from langflow.components.outputs.ChatOutput import ChatOutput
 from langflow.components.outputs.TextOutput import TextOutputComponent
-from langflow.components.tools.YfinanceTool import YahooFinanceToolComponent
+from langflow.components.tools.YfinanceTool import YfinanceToolComponent
 from langflow.graph.graph.base import Graph
 from langflow.graph.graph.constants import Finish
 
@@ -151,6 +151,6 @@ def test_graph_functional_start_end():
 
 @pytest.mark.skip(reason="Temporarily disabled")
 def test_graph_set_with_valid_component():
-    tool = YahooFinanceToolComponent()
+    tool = YfinanceToolComponent()
     tool_calling_agent = ToolCallingAgentComponent()
     tool_calling_agent.set(tools=[tool])