ref: Rename component modules to lower case (ruff N999) (#4220)

* Rename component modules to lower case (ruff N999)

* git rename workaround (1/2)

* git rename workaround (2/2)

* Rollback changes on schema __init__

* Fix tests

src/backend/tests/conftest.py

@@ -15,8 +15,8 @@ from asgi_lifespan import LifespanManager
 from dotenv import load_dotenv
 from fastapi.testclient import TestClient
 from httpx import ASGITransport, AsyncClient
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.graph.graph.base import Graph
+from langflow.components.inputs import ChatInput
+from langflow.graph import Graph
 from langflow.initial_setup.setup import STARTER_FOLDER_NAME
 from langflow.services.auth.utils import get_password_hash
 from langflow.services.database.models.api_key.model import ApiKey

src/backend/tests/integration/components/helpers/test_parse_json_data.py

@@ -1,5 +1,5 @@
 import pytest
-from langflow.components.helpers.ParseJSONData import ParseJSONDataComponent
+from langflow.components.helpers import ParseJSONDataComponent
 from langflow.components.inputs import ChatInput
 from langflow.schema import Data
 

src/backend/tests/integration/components/output_parsers/test_output_parser.py

@@ -1,9 +1,9 @@
 import os
 
 import pytest
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.output_parsers.OutputParser import OutputParserComponent
-from langflow.components.prompts.Prompt import PromptComponent
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.output_parsers import OutputParserComponent
+from langflow.components.prompts import PromptComponent
 
 from tests.integration.utils import ComponentInputHandle, run_single_component
 

src/backend/tests/unit/base/tools/test_component_toolkit.py

@@ -2,11 +2,11 @@ import os
 
 import pytest
 from langflow.base.tools.component_tool import ComponentToolkit
-from langflow.components.agents.ToolCallingAgent import ToolCallingAgentComponent
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
+from langflow.components.agents import ToolCallingAgentComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
 from langflow.components.outputs import ChatOutput
-from langflow.graph.graph.base import Graph
+from langflow.graph import Graph
 from langflow.schema.message import Message
 from langflow.services.settings.feature_flags import FEATURE_FLAGS
 

src/backend/tests/unit/components/models/test_chatollama_component.py

@@ -2,8 +2,8 @@ from unittest.mock import MagicMock, patch
 from urllib.parse import urljoin
 
 import pytest
-from langchain_community.chat_models.ollama import ChatOllama
-from langflow.components.models.OllamaModel import ChatOllamaComponent
+from langchain_community.chat_models import ChatOllama
+from langflow.components.models import ChatOllamaComponent
 
 
 @pytest.fixture

src/backend/tests/unit/components/prompts/test_prompt_component.py

@@ -1,4 +1,4 @@
-from langflow.components.prompts.Prompt import PromptComponent
+from langflow.components.prompts import PromptComponent
 
 
 class TestPromptComponent:

src/backend/tests/unit/components/prototypes/test_create_data_component.py

@@ -1,5 +1,5 @@
 import pytest
-from langflow.components.prototypes.CreateData import CreateDataComponent
+from langflow.components.prototypes import CreateDataComponent
 from langflow.schema import Data
 
 

src/backend/tests/unit/components/prototypes/test_update_data_component.py

@@ -1,5 +1,5 @@
 import pytest
-from langflow.components.prototypes.UpdateData import UpdateDataComponent
+from langflow.components.prototypes import UpdateDataComponent
 from langflow.schema import Data
 
 

src/backend/tests/unit/components/tools/test_python_repl_tool.py

@@ -1,5 +1,5 @@
-from langflow.components.tools.PythonREPLTool import PythonREPLToolComponent
-from langflow.custom.custom_component.component import Component
+from langflow.components.tools import PythonREPLToolComponent
+from langflow.custom import Component
 from langflow.custom.utils import build_custom_component_template
 
 

src/backend/tests/unit/components/tools/test_yfinance_tool.py

@@ -1,5 +1,5 @@
-from langflow.components.tools.YfinanceTool import YfinanceToolComponent
-from langflow.custom.custom_component.component import Component
+from langflow.components.tools import YfinanceToolComponent
+from langflow.custom import Component
 from langflow.custom.utils import build_custom_component_template
 
 

src/backend/tests/unit/custom/component/test_component_to_tool.py

@@ -1,6 +1,6 @@
 from collections.abc import Callable
 
-from langflow.components.inputs.ChatInput import ChatInput
+from langflow.components.inputs import ChatInput
 
 
 def test_component_to_toolkit():

src/backend/tests/unit/custom/custom_component/test_component.py

@@ -1,11 +1,10 @@
 import pytest
-from langflow.components.agents.CrewAIAgent import CrewAIAgentComponent
-from langflow.components.agents.ToolCallingAgent import ToolCallingAgentComponent
-from langflow.components.helpers.SequentialTask import SequentialTaskComponent
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
+from langflow.components.agents import CrewAIAgentComponent, ToolCallingAgentComponent
+from langflow.components.helpers import SequentialTaskComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
 from langflow.components.outputs import ChatOutput
-from langflow.template.field.base import Output
+from langflow.template import Output
 
 
 def test_set_invalid_output():

src/backend/tests/unit/graph/edge/test_edge_base.py

@@ -1,9 +1,9 @@
 import pytest
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.components.prompts.Prompt import PromptComponent
-from langflow.graph.graph.base import Graph
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.outputs import ChatOutput
+from langflow.components.prompts import PromptComponent
+from langflow.graph import Graph
 
 
 @pytest.mark.usefixtures("client")

src/backend/tests/unit/graph/graph/state/test_state_model.py

@@ -1,7 +1,7 @@
 import pytest
 from langflow.components.inputs import ChatInput
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.graph.graph.base import Graph
+from langflow.components.outputs import ChatOutput
+from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
 from langflow.graph.state.model import create_state_model
 from langflow.template.field.base import UNDEFINED

src/backend/tests/unit/graph/graph/test_base.py

@@ -2,12 +2,11 @@ import logging
 from collections import deque
 
 import pytest
-from langflow.components.agents.ToolCallingAgent import ToolCallingAgentComponent
-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 YfinanceToolComponent
-from langflow.graph.graph.base import Graph
+from langflow.components.agents import ToolCallingAgentComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.outputs import ChatOutput, TextOutputComponent
+from langflow.components.tools import YfinanceToolComponent
+from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
 
 

src/backend/tests/unit/graph/graph/test_callback_graph.py

@@ -1,12 +1,12 @@
 import asyncio
 
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.custom.custom_component.component import Component
+from langflow.components.outputs import ChatOutput
+from langflow.custom import Component
 from langflow.events.event_manager import EventManager
-from langflow.graph.graph.base import Graph
-from langflow.inputs.inputs import IntInput
+from langflow.graph import Graph
+from langflow.inputs import IntInput
 from langflow.schema.message import Message
-from langflow.template.field.base import Output
+from langflow.template import Output
 
 
 class LogComponent(Component):

src/backend/tests/unit/graph/graph/test_cycles.py

@@ -1,14 +1,13 @@
 import os
 
 import pytest
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.components.outputs.TextOutput import TextOutputComponent
-from langflow.components.prompts.Prompt import PromptComponent
-from langflow.components.prototypes.ConditionalRouter import ConditionalRouterComponent
-from langflow.custom.custom_component.component import Component
-from langflow.graph.graph.base import Graph
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.outputs import ChatOutput, TextOutputComponent
+from langflow.components.prompts import PromptComponent
+from langflow.components.prototypes import ConditionalRouterComponent
+from langflow.custom import Component
+from langflow.graph import Graph
 from langflow.graph.graph.utils import find_cycle_vertices
 from langflow.io import MessageTextInput, Output
 from langflow.schema.message import Message

src/backend/tests/unit/graph/graph/test_graph_state_model.py

@@ -1,11 +1,11 @@
 from typing import TYPE_CHECKING
 
 import pytest
-from langflow.components.helpers.Memory import MemoryComponent
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.components.prompts.Prompt import PromptComponent
+from langflow.components.helpers import MemoryComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.outputs import ChatOutput
+from langflow.components.prompts import PromptComponent
 from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
 from langflow.graph.graph.state_model import create_state_model_from_graph

src/backend/tests/unit/initial_setup/starter_projects/test_memory_chatbot.py

@@ -3,11 +3,11 @@ from collections import deque
 from typing import TYPE_CHECKING
 
 import pytest
-from langflow.components.helpers.Memory import MemoryComponent
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.components.prompts.Prompt import PromptComponent
+from langflow.components.helpers import MemoryComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.outputs import ChatOutput
+from langflow.components.prompts import PromptComponent
 from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
 

src/backend/tests/unit/initial_setup/starter_projects/test_vector_store_rag.py

@@ -3,18 +3,17 @@ import operator
 from textwrap import dedent
 
 import pytest
-from langflow.components.data.File import FileComponent
-from langflow.components.embeddings.OpenAIEmbeddings import OpenAIEmbeddingsComponent
-from langflow.components.helpers.ParseData import ParseDataComponent
-from langflow.components.helpers.SplitText import SplitTextComponent
-from langflow.components.inputs.ChatInput import ChatInput
-from langflow.components.models.OpenAIModel import OpenAIModelComponent
-from langflow.components.outputs.ChatOutput import ChatOutput
-from langflow.components.prompts.Prompt import PromptComponent
-from langflow.components.vectorstores.AstraDB import AstraVectorStoreComponent
-from langflow.graph.graph.base import Graph
+from langflow.components.data import FileComponent
+from langflow.components.embeddings import OpenAIEmbeddingsComponent
+from langflow.components.helpers import ParseDataComponent, SplitTextComponent
+from langflow.components.inputs import ChatInput
+from langflow.components.models import OpenAIModelComponent
+from langflow.components.outputs import ChatOutput
+from langflow.components.prompts import PromptComponent
+from langflow.components.vectorstores import AstraVectorStoreComponent
+from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
-from langflow.schema.data import Data
+from langflow.schema import Data
 
 
 @pytest.fixture

src/backend/tests/unit/io/test_io_schema.py

@@ -1,7 +1,7 @@
 from typing import TYPE_CHECKING, Literal
 
 import pytest
-from langflow.components.inputs.ChatInput import ChatInput
+from langflow.components.inputs import ChatInput
 
 if TYPE_CHECKING:
     from pydantic.fields import FieldInfo

src/backend/tests/unit/test_data_components.py

@@ -113,8 +113,8 @@ async def test_build_with_multiple_urls(api_request):
     assert len(results) == len(urls)
 
 
-@patch("langflow.components.data.Directory.parallel_load_data")
-@patch("langflow.components.data.Directory.retrieve_file_paths")
+@patch("langflow.components.data.directory.parallel_load_data")
+@patch("langflow.components.data.directory.retrieve_file_paths")
 @patch("langflow.components.data.DirectoryComponent.resolve_path")
 def test_directory_component_build_with_multithreading(
     mock_resolve_path, mock_retrieve_file_paths, mock_parallel_load_data

src/backend/tests/unit/test_helper_components.py

@@ -35,7 +35,7 @@ from langflow.schema.message import Message
 def test_uuid_generator_component():
     # Arrange
     uuid_generator_component = helpers.IDGeneratorComponent()
-    uuid_generator_component._code = Path(helpers.IDGenerator.__file__).read_text(encoding="utf-8")
+    uuid_generator_component._code = Path(helpers.id_generator.__file__).read_text(encoding="utf-8")
 
     frontend_node, _ = build_custom_component_template(uuid_generator_component)