ref: Auto-fix ruff rules in tests (#4154)

src/backend/tests/unit/api/test_api_utils.py

@@ -1,5 +1,6 @@
-from langflow.api.utils import get_suggestion_message
 from unittest.mock import patch
+
+from langflow.api.utils import get_suggestion_message
 from langflow.services.database.models.flow.utils import get_outdated_components
 from langflow.utils.version import get_version_info
 

src/backend/tests/unit/api/v1/test_variable.py

@@ -21,11 +21,11 @@ async def test_create_variable(client: AsyncClient, body, logged_in_headers):
     response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_201_CREATED == response.status_code
+    assert response.status_code == status.HTTP_201_CREATED
     assert body["name"] == result["name"]
     assert body["type"] == result["type"]
     assert body["default_fields"] == result["default_fields"]
-    assert "id" in result.keys()
+    assert "id" in result
     assert body["value"] != result["value"]
 
 
@@ -36,7 +36,7 @@ async def test_create_variable__variable_name_already_exists(client: AsyncClient
     response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_400_BAD_REQUEST == response.status_code
+    assert response.status_code == status.HTTP_400_BAD_REQUEST
     assert "Variable name already exists" in result["detail"]
 
 
@@ -48,7 +48,7 @@ async def test_create_variable__variable_name_and_value_cannot_be_empty(client:
     response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_400_BAD_REQUEST == response.status_code
+    assert response.status_code == status.HTTP_400_BAD_REQUEST
     assert "Variable name and value cannot be empty" in result["detail"]
 
 
@@ -59,7 +59,7 @@ async def test_create_variable__variable_name_cannot_be_empty(client: AsyncClien
     response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_400_BAD_REQUEST == response.status_code
+    assert response.status_code == status.HTTP_400_BAD_REQUEST
     assert "Variable name cannot be empty" in result["detail"]
 
 
@@ -70,7 +70,7 @@ async def test_create_variable__variable_value_cannot_be_empty(client: AsyncClie
     response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_400_BAD_REQUEST == response.status_code
+    assert response.status_code == status.HTTP_400_BAD_REQUEST
     assert "Variable value cannot be empty" in result["detail"]
 
 
@@ -84,7 +84,7 @@ async def test_create_variable__HTTPException(client: AsyncClient, body, logged_
         response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
         result = response.json()
 
-        assert status.HTTP_418_IM_A_TEAPOT == response.status_code
+        assert response.status_code == status.HTTP_418_IM_A_TEAPOT
         assert generic_message in result["detail"]
 
 
@@ -97,7 +97,7 @@ async def test_create_variable__Exception(client: AsyncClient, body, logged_in_h
         response = await client.post("api/v1/variables/", json=body, headers=logged_in_headers)
         result = response.json()
 
-        assert status.HTTP_500_INTERNAL_SERVER_ERROR == response.status_code
+        assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
         assert generic_message in result["detail"]
 
 
@@ -111,7 +111,7 @@ async def test_read_variables(client: AsyncClient, body, logged_in_headers):
     response = await client.get("api/v1/variables/", headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_200_OK == response.status_code
+    assert response.status_code == status.HTTP_200_OK
     assert all(name in [r["name"] for r in result] for name in names)
 
 
@@ -125,23 +125,22 @@ async def test_read_variables__empty(client: AsyncClient, logged_in_headers):
     response = await client.get("api/v1/variables/", headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_200_OK == response.status_code
-    assert [] == result
+    assert response.status_code == status.HTTP_200_OK
+    assert result == []
 
 
 @pytest.mark.usefixtures("active_user")
 async def test_read_variables__(client: AsyncClient, logged_in_headers):
     generic_message = "Generic error message"
 
-    with pytest.raises(Exception) as exc:
-        with mock.patch("sqlmodel.Session.exec") as m:
-            m.side_effect = Exception(generic_message)
+    with pytest.raises(Exception) as exc, mock.patch("sqlmodel.Session.exec") as m:
+        m.side_effect = Exception(generic_message)
 
-            response = await client.get("api/v1/variables/", headers=logged_in_headers)
-            result = response.json()
+        response = await client.get("api/v1/variables/", headers=logged_in_headers)
+        result = response.json()
 
-            assert status.HTTP_500_INTERNAL_SERVER_ERROR == response.status_code
-            assert generic_message in result["detail"]
+        assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
+        assert generic_message in result["detail"]
 
     assert generic_message in str(exc.value)
 
@@ -159,7 +158,7 @@ async def test_update_variable(client: AsyncClient, body, logged_in_headers):
     response = await client.patch(f"api/v1/variables/{saved.get('id')}", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_200_OK == response.status_code
+    assert response.status_code == status.HTTP_200_OK
     assert saved["id"] == result["id"]
     assert saved["name"] != result["name"]
     assert saved["default_fields"] != result["default_fields"]
@@ -173,7 +172,7 @@ async def test_update_variable__Exception(client: AsyncClient, body, logged_in_h
     response = await client.patch(f"api/v1/variables/{wrong_id}", json=body, headers=logged_in_headers)
     result = response.json()
 
-    assert status.HTTP_404_NOT_FOUND == response.status_code
+    assert response.status_code == status.HTTP_404_NOT_FOUND
     assert "Variable not found" in result["detail"]
 
 
@@ -183,7 +182,7 @@ async def test_delete_variable(client: AsyncClient, body, logged_in_headers):
     saved = response.json()
     response = await client.delete(f"api/v1/variables/{saved.get('id')}", headers=logged_in_headers)
 
-    assert status.HTTP_204_NO_CONTENT == response.status_code
+    assert response.status_code == status.HTTP_204_NO_CONTENT
 
 
 @pytest.mark.usefixtures("active_user")
@@ -192,4 +191,4 @@ async def test_delete_variable__Exception(client: AsyncClient, logged_in_headers
 
     response = await client.delete(f"api/v1/variables/{wrong_id}", headers=logged_in_headers)
 
-    assert status.HTTP_500_INTERNAL_SERVER_ERROR == response.status_code
+    assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR

src/backend/tests/unit/base/load/test_load.py

@@ -1,4 +1,5 @@
 import inspect
+
 from langflow.load import run_flow_from_json
 
 

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

@@ -1,7 +1,6 @@
 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
@@ -76,7 +75,7 @@ def test_component_tool():
     }
     assert component_toolkit.component == chat_input
 
-    result = component_tool.invoke(input=dict(input_value="test"))
+    result = component_tool.invoke(input={"input_value": "test"})
     assert isinstance(result, Message)
     assert result.get_text() == "test"
 

src/backend/tests/unit/components/helpers/test_structured_output_component.py

@@ -1,10 +1,9 @@
 from unittest.mock import MagicMock, patch
 
 import pytest
-from pydantic import BaseModel
-
 from langflow.components.helpers.structured_output import StructuredOutputComponent
 from langflow.schema.data import Data
+from pydantic import BaseModel
 
 
 @pytest.fixture

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

@@ -3,7 +3,6 @@ from urllib.parse import urljoin
 
 import pytest
 from langchain_community.chat_models.ollama import ChatOllama
-
 from langflow.components.models.OllamaModel import ChatOllamaComponent
 
 

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

@@ -1,6 +1,5 @@
 from collections.abc import Callable
 
-
 from langflow.components.inputs.ChatInput import ChatInput
 
 

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

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.components.agents.CrewAIAgent import CrewAIAgentComponent
 from langflow.components.agents.ToolCallingAgent import ToolCallingAgentComponent
 from langflow.components.helpers.SequentialTask import SequentialTaskComponent

src/backend/tests/unit/events/test_event_manager.py

@@ -4,7 +4,6 @@ import time
 import uuid
 
 import pytest
-
 from langflow.events.event_manager import EventManager
 from langflow.schema.log import LoggableType
 

src/backend/tests/unit/exceptions/test_api.py

@@ -1,4 +1,5 @@
-from unittest.mock import patch, Mock
+from unittest.mock import Mock, patch
+
 from langflow.services.database.models.flow.model import Flow
 
 

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

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.components.inputs.ChatInput import ChatInput
 from langflow.components.models.OpenAIModel import OpenAIModelComponent
 from langflow.components.outputs.ChatOutput import ChatOutput

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

@@ -1,12 +1,11 @@
 import pytest
-from pydantic import Field
-
 from langflow.components.inputs import ChatInput
 from langflow.components.outputs.ChatOutput import ChatOutput
 from langflow.graph.graph.base 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
+from pydantic import Field
 
 
 @pytest.fixture

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

@@ -2,8 +2,6 @@ import logging
 from collections import deque
 
 import pytest
-from pytest import LogCaptureFixture
-
 from langflow.components.agents.ToolCallingAgent import ToolCallingAgentComponent
 from langflow.components.inputs.ChatInput import ChatInput
 from langflow.components.outputs.ChatOutput import ChatOutput
@@ -11,6 +9,7 @@ from langflow.components.outputs.TextOutput import TextOutputComponent
 from langflow.components.tools.YfinanceTool import YfinanceToolComponent
 from langflow.graph.graph.base import Graph
 from langflow.graph.graph.constants import Finish
+from pytest import LogCaptureFixture
 
 
 @pytest.mark.asyncio

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

@@ -1,6 +1,5 @@
 import asyncio
 
-
 from langflow.components.outputs.ChatOutput import ChatOutput
 from langflow.custom.custom_component.component import Component
 from langflow.events.event_manager import EventManager

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

@@ -1,7 +1,6 @@
 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
@@ -203,5 +202,3 @@ def test_updated_graph_with_prompts():
     # Extract the vertex IDs for analysis
     results_ids = [result.vertex.id for result in results if hasattr(result, "vertex")]
     assert "chat_output_1" in results_ids, f"Expected outputs not in results: {results_ids}"
-
-    print(f"Execution completed with results: {results_ids}")

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

@@ -1,6 +1,6 @@
-import pytest
-from pydantic import BaseModel
+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
@@ -10,6 +10,9 @@ from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
 from langflow.graph.graph.state_model import create_state_model_from_graph
 
+if TYPE_CHECKING:
+    from pydantic import BaseModel
+
 
 def test_graph_state_model():
     session_id = "test_session_id"

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

@@ -1,10 +1,12 @@
 import pickle
-from collections import defaultdict
+from typing import TYPE_CHECKING
 
 import pytest
-
 from langflow.graph.graph.runnable_vertices_manager import RunnableVerticesManager
 
+if TYPE_CHECKING:
+    from collections import defaultdict
+
 
 @pytest.fixture
 def data():
@@ -23,7 +25,7 @@ def data():
 def test_to_dict(data):
     result = RunnableVerticesManager.from_dict(data).to_dict()
 
-    assert all(key in result.keys() for key in data.keys())
+    assert all(key in result for key in data)
 
 
 def test_from_dict(data):
@@ -158,8 +160,8 @@ def test_build_run_map(data):
 
     manager.build_run_map(predecessor_map, vertices_to_run)
 
-    assert all(v in manager.run_map.keys() for v in ["Z", "X", "Y"])
-    assert "W" not in manager.run_map.keys()
+    assert all(v in manager.run_map for v in ["Z", "X", "Y"])
+    assert "W" not in manager.run_map
 
 
 def test_update_vertex_run_state(data):

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

@@ -1,7 +1,6 @@
 import copy
 
 import pytest
-
 from langflow.graph.graph import utils
 
 

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

@@ -2,7 +2,6 @@ import copy
 import json
 
 import pytest
-
 from langflow.graph import Graph
 from langflow.graph.graph.utils import (
     find_last_node,
@@ -59,7 +58,7 @@ def sample_nodes():
 
 
 def get_node_by_type(graph, node_type: type[Vertex]) -> Vertex | None:
-    """Get a node by type"""
+    """Get a node by type."""
     return next((node for node in graph.vertices if isinstance(node, node_type)), None)
 
 
@@ -131,10 +130,10 @@ def test_process_flow_one_group(one_grouped_chat_json_flow):
     node_data = group_node["data"]["node"]
     assert node_data.get("flow") is not None
     template_data = node_data["template"]
-    assert any("openai_api_key" in key for key in template_data.keys())
+    assert any("openai_api_key" in key for key in template_data)
     # Get the openai_api_key dict
     openai_api_key = next(
-        (template_data[key] for key in template_data.keys() if "openai_api_key" in key),
+        (template_data[key] for key in template_data if "openai_api_key" in key),
         None,
     )
     assert openai_api_key is not None

src/backend/tests/unit/helpers/test_base_model_from_schema.py

@@ -3,11 +3,10 @@
 from typing import Any
 
 import pytest
+from langflow.helpers.base_model import build_model_from_schema
 from pydantic import BaseModel
 from pydantic_core import PydanticUndefined
 
-from langflow.helpers.base_model import build_model_from_schema
-
 
 class TestBuildModelFromSchema:
     # Successfully creates a Pydantic model from a valid schema
@@ -115,9 +114,9 @@ class TestBuildModelFromSchema:
             {"name": "field3", "type": "list", "default": None, "description": "Field 3 description", "multiple": True},
         ]
         model = build_model_from_schema(schema)
-        assert model.model_fields["field1"].default == PydanticUndefined  # noqa: E711
-        assert model.model_fields["field2"].default == PydanticUndefined  # noqa: E711
-        assert model.model_fields["field3"].default == PydanticUndefined  # noqa: E711
+        assert model.model_fields["field1"].default == PydanticUndefined
+        assert model.model_fields["field2"].default == PydanticUndefined
+        assert model.model_fields["field3"].default == PydanticUndefined
 
     # Checks for proper handling of nested list and dict types
     def test_nested_list_and_dict_types_handling(self):

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

@@ -1,7 +1,8 @@
+import operator
 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
@@ -9,7 +10,9 @@ from langflow.components.outputs.ChatOutput import ChatOutput
 from langflow.components.prompts.Prompt import PromptComponent
 from langflow.graph import Graph
 from langflow.graph.graph.constants import Finish
-from langflow.graph.graph.schema import GraphDump
+
+if TYPE_CHECKING:
+    from langflow.graph.graph.schema import GraphDump
 
 
 @pytest.fixture
@@ -100,7 +103,7 @@ def test_memory_chatbot_dump_components_and_edges(memory_chatbot_graph: Graph):
     edges = data_dict["edges"]
 
     # sort the nodes by id
-    nodes = sorted(nodes, key=lambda x: x["id"])
+    nodes = sorted(nodes, key=operator.itemgetter("id"))
 
     # Check each node
     assert nodes[0]["data"]["type"] == "ChatInput"

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

@@ -1,8 +1,8 @@
 import copy
+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
@@ -40,8 +40,7 @@ def ingestion_graph():
     vector_store.set_on_output(name="base_retriever", value="mock_retriever", cache=True)
     vector_store.set_on_output(name="search_results", value=[Data(text="This is a test file.")], cache=True)
 
-    ingestion_graph = Graph(file_component, vector_store)
-    return ingestion_graph
+    return Graph(file_component, vector_store)
 
 
 @pytest.fixture
@@ -89,8 +88,7 @@ def rag_graph():
     chat_output = ChatOutput(_id="chatoutput-123")
     chat_output.set(input_value=openai_component.text_response)
 
-    graph = Graph(start=chat_input, end=chat_output)
-    return graph
+    return Graph(start=chat_input, end=chat_output)
 
 
 def test_vector_store_rag(ingestion_graph, rag_graph):
@@ -111,7 +109,7 @@ def test_vector_store_rag(ingestion_graph, rag_graph):
         "rag-vector-store-123",
         "openai-embeddings-124",
     ]
-    for ids, graph, len_results in zip([ingestion_ids, rag_ids], [ingestion_graph, rag_graph], [5, 8]):
+    for ids, graph, len_results in [(ingestion_ids, ingestion_graph, 5), (rag_ids, rag_graph, 8)]:
         results = []
         for result in graph.start():
             results.append(result)
@@ -134,7 +132,7 @@ def test_vector_store_rag_dump_components_and_edges(ingestion_graph, rag_graph):
     ingestion_edges = ingestion_data["edges"]
 
     # Sort nodes by id to check components
-    ingestion_nodes = sorted(ingestion_nodes, key=lambda x: x["id"])
+    ingestion_nodes = sorted(ingestion_nodes, key=operator.itemgetter("id"))
 
     # Check components in the ingestion graph
     assert ingestion_nodes[0]["data"]["type"] == "File"
@@ -172,7 +170,7 @@ def test_vector_store_rag_dump_components_and_edges(ingestion_graph, rag_graph):
     rag_edges = rag_data["edges"]
 
     # Sort nodes by id to check components
-    rag_nodes = sorted(rag_nodes, key=lambda x: x["id"])
+    rag_nodes = sorted(rag_nodes, key=operator.itemgetter("id"))
 
     # Check components in the RAG graph
     assert rag_nodes[0]["data"]["type"] == "ChatInput"
@@ -235,7 +233,7 @@ def test_vector_store_rag_add(ingestion_graph: Graph, rag_graph: Graph):
     combined_edges = combined_data["edges"]
 
     # Sort nodes by id to check components
-    combined_nodes = sorted(combined_nodes, key=lambda x: x["id"])
+    combined_nodes = sorted(combined_nodes, key=operator.itemgetter("id"))
 
     # Expected components in the combined graph (both ingestion and RAG nodes)
     expected_nodes = sorted(
@@ -252,10 +250,10 @@ def test_vector_store_rag_add(ingestion_graph: Graph, rag_graph: Graph):
             {"id": "prompt-123", "type": "Prompt"},
             {"id": "rag-vector-store-123", "type": "AstraDB"},
         ],
-        key=lambda x: x["id"],
+        key=operator.itemgetter("id"),
     )
 
-    for expected_node, combined_node in zip(expected_nodes, combined_nodes):
+    for expected_node, combined_node in zip(expected_nodes, combined_nodes, strict=True):
         assert combined_node["data"]["type"] == expected_node["type"]
         assert combined_node["id"] == expected_node["id"]
 

src/backend/tests/unit/inputs/test_inputs.py

@@ -1,6 +1,4 @@
 import pytest
-from pydantic import ValidationError
-
 from langflow.inputs.inputs import (
     BoolInput,
     CodeInput,
@@ -24,6 +22,7 @@ from langflow.inputs.inputs import (
 )
 from langflow.inputs.utils import instantiate_input
 from langflow.schema.message import Message
+from pydantic import ValidationError
 
 
 def test_table_input_valid():

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

@@ -1,10 +1,11 @@
-from typing import Literal
+from typing import TYPE_CHECKING, Literal
 
 import pytest
-from pydantic.fields import FieldInfo
-
 from langflow.components.inputs.ChatInput import ChatInput
 
+if TYPE_CHECKING:
+    from pydantic.fields import FieldInfo
+
 
 def test_create_input_schema():
     from langflow.io.schema import create_input_schema

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

@@ -1,7 +1,6 @@
 # Generated by qodo Gen
 
 import pytest
-
 from langflow.schema.table import Column, FormatterType
 
 

src/backend/tests/unit/schema/test_schema_message.py

@@ -1,6 +1,5 @@
 import pytest
 from langchain_core.prompts.chat import ChatPromptTemplate
-
 from langflow.schema.message import Message
 
 

src/backend/tests/unit/services/variable/test_service.py

@@ -3,13 +3,12 @@ from unittest.mock import patch
 from uuid import uuid4
 
 import pytest
-from sqlmodel import Session, SQLModel, create_engine
-
 from langflow.services.database.models.variable.model import VariableUpdate
 from langflow.services.deps import get_settings_service
+from langflow.services.settings.constants import VARIABLES_TO_GET_FROM_ENVIRONMENT
 from langflow.services.variable.constants import CREDENTIAL_TYPE, GENERIC_TYPE
 from langflow.services.variable.service import DatabaseVariableService
-from langflow.services.settings.constants import VARIABLES_TO_GET_FROM_ENVIRONMENT
+from sqlmodel import Session, SQLModel, create_engine
 
 
 @pytest.fixture
@@ -44,8 +43,8 @@ def test_initialize_user_variables__create_and_update(service, session):
         value = service.get_variable(user_id, name, field, session=session)
         assert value == env_vars[name]
 
-    assert all([i in variables for i in good_vars.keys()])
-    assert all([i not in variables for i in bad_vars.keys()])
+    assert all(i in variables for i in good_vars)
+    assert all(i not in variables for i in bad_vars)
 
 
 def test_initialize_user_variables__not_found_variable(service, session):

src/backend/tests/unit/test_api_key.py

@@ -1,6 +1,5 @@
 import pytest
 from httpx import AsyncClient
-
 from langflow.services.database.models.api_key import ApiKeyCreate
 
 
@@ -30,7 +29,8 @@ async def test_create_api_key(client: AsyncClient, logged_in_headers):
     response = await client.post("api/v1/api_key/", json={"name": api_key_name}, headers=logged_in_headers)
     assert response.status_code == 200
     data = response.json()
-    assert "name" in data and data["name"] == api_key_name
+    assert "name" in data
+    assert data["name"] == api_key_name
     assert "api_key" in data
     assert "**" not in data["api_key"]
 

src/backend/tests/unit/test_chat_endpoint.py

@@ -1,10 +1,9 @@
 import json
 from uuid import UUID
 
-from orjson import orjson
-
 from langflow.memory import get_messages
 from langflow.services.database.models.flow import FlowCreate, FlowUpdate
+from orjson import orjson
 
 
 async def test_build_flow(client, json_memory_chatbot_no_llm, logged_in_headers):
@@ -82,7 +81,8 @@ async def consume_and_assert_stream(r):
         elif count == 5:
             assert parsed["event"] == "end"
         else:
-            raise ValueError(f"Unexpected line: {line}")
+            msg = f"Unexpected line: {line}"
+            raise ValueError(msg)
         count += 1
 
 
@@ -92,5 +92,4 @@ async def _create_flow(client, json_memory_chatbot_no_llm, logged_in_headers):
     vector_store = FlowCreate(name="Flow", description="description", data=data, endpoint_name="f")
     response = await client.post("api/v1/flows/", json=vector_store.model_dump(), headers=logged_in_headers)
     response.raise_for_status()
-    flow_id = response.json()["id"]
-    return flow_id
+    return response.json()["id"]

src/backend/tests/unit/test_cli.py

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.__main__ import app
 from langflow.services import deps
 

src/backend/tests/unit/test_custom_component.py

@@ -5,7 +5,6 @@ from textwrap import dedent
 
 import pytest
 from langchain_core.documents import Document
-
 from langflow.custom import Component, CustomComponent
 from langflow.custom.code_parser.code_parser import CodeParser, CodeSyntaxError
 from langflow.custom.custom_component.base_component import BaseComponent, ComponentCodeNullError
@@ -14,7 +13,7 @@ from langflow.custom.utils import build_custom_component_template
 
 @pytest.fixture
 def code_component_with_multiple_outputs():
-    code = Path("src/backend/tests/data/component_multiple_outputs.py").read_text()
+    code = Path("src/backend/tests/data/component_multiple_outputs.py").read_text(encoding="utf-8")
     return Component(_code=code)
 
 
@@ -39,25 +38,20 @@ class YourComponent(CustomComponent):
 
 
 def test_code_parser_init():
-    """
-    Test the initialization of the CodeParser class.
-    """
+    """Test the initialization of the CodeParser class."""
     parser = CodeParser(code_default)
     assert parser.code == code_default
 
 
 def test_code_parser_get_tree():
-    """
-    Test the __get_tree method of the CodeParser class.
-    """
+    """Test the __get_tree method of the CodeParser class."""
     parser = CodeParser(code_default)
     tree = parser.get_tree()
     assert isinstance(tree, ast.AST)
 
 
 def test_code_parser_syntax_error():
-    """
-    Test the __get_tree method raises the
+    """Test the __get_tree method raises the
     CodeSyntaxError when given incorrect syntax.
     """
     code_syntax_error = "zzz import os"
@@ -68,26 +62,21 @@ def test_code_parser_syntax_error():
 
 
 def test_component_init():
-    """
-    Test the initialization of the Component class.
-    """
+    """Test the initialization of the Component class."""
     component = BaseComponent(_code=code_default, _function_entrypoint_name="build")
     assert component._code == code_default
     assert component._function_entrypoint_name == "build"
 
 
 def test_component_get_code_tree():
-    """
-    Test the get_code_tree method of the Component class.
-    """
+    """Test the get_code_tree method of the Component class."""
     component = BaseComponent(_code=code_default, _function_entrypoint_name="build")
     tree = component.get_code_tree(component._code)
     assert "imports" in tree
 
 
 def test_component_code_null_error():
-    """
-    Test the get_function method raises the
+    """Test the get_function method raises the
     ComponentCodeNullError when the code is empty.
     """
     component = BaseComponent(_code="", _function_entrypoint_name="")
@@ -96,9 +85,7 @@ def test_component_code_null_error():
 
 
 def test_custom_component_init():
-    """
-    Test the initialization of the CustomComponent class.
-    """
+    """Test the initialization of the CustomComponent class."""
     function_entrypoint_name = "build"
 
     custom_component = CustomComponent(_code=code_default, _function_entrypoint_name=function_entrypoint_name)
@@ -107,26 +94,21 @@ def test_custom_component_init():
 
 
 def test_custom_component_build_template_config():
-    """
-    Test the build_template_config property of the CustomComponent class.
-    """
+    """Test the build_template_config property of the CustomComponent class."""
     custom_component = CustomComponent(_code=code_default, _function_entrypoint_name="build")
     config = custom_component.build_template_config()
     assert isinstance(config, dict)
 
 
 def test_custom_component_get_function():
-    """
-    Test the get_function property of the CustomComponent class.
-    """
+    """Test the get_function property of the CustomComponent class."""
     custom_component = CustomComponent(_code="def build(): pass", _function_entrypoint_name="build")
     my_function = custom_component.get_function()
     assert isinstance(my_function, types.FunctionType)
 
 
 def test_code_parser_parse_imports_import():
-    """
-    Test the parse_imports method of the CodeParser
+    """Test the parse_imports method of the CodeParser
     class with an import statement.
     """
     parser = CodeParser(code_default)
@@ -138,8 +120,7 @@ def test_code_parser_parse_imports_import():
 
 
 def test_code_parser_parse_imports_importfrom():
-    """
-    Test the parse_imports method of the CodeParser
+    """Test the parse_imports method of the CodeParser
     class with an import from statement.
     """
     parser = CodeParser("from os import path")
@@ -151,9 +132,7 @@ def test_code_parser_parse_imports_importfrom():
 
 
 def test_code_parser_parse_functions():
-    """
-    Test the parse_functions method of the CodeParser class.
-    """
+    """Test the parse_functions method of the CodeParser class."""
     parser = CodeParser("def test(): pass")
     tree = parser.get_tree()
     for node in ast.walk(tree):
@@ -164,9 +143,7 @@ def test_code_parser_parse_functions():
 
 
 def test_code_parser_parse_classes():
-    """
-    Test the parse_classes method of the CodeParser class.
-    """
+    """Test the parse_classes method of the CodeParser class."""
     parser = CodeParser("from langflow.custom import Component\n\nclass Test(Component): pass")
     tree = parser.get_tree()
     for node in ast.walk(tree):
@@ -177,9 +154,7 @@ def test_code_parser_parse_classes():
 
 
 def test_code_parser_parse_classes_raises():
-    """
-    Test the parse_classes method of the CodeParser class.
-    """
+    """Test the parse_classes method of the CodeParser class."""
     parser = CodeParser("class Test: pass")
     tree = parser.get_tree()
     with pytest.raises(TypeError):
@@ -189,9 +164,7 @@ def test_code_parser_parse_classes_raises():
 
 
 def test_code_parser_parse_global_vars():
-    """
-    Test the parse_global_vars method of the CodeParser class.
-    """
+    """Test the parse_global_vars method of the CodeParser class."""
     parser = CodeParser("x = 1")
     tree = parser.get_tree()
     for node in ast.walk(tree):
@@ -202,8 +175,7 @@ def test_code_parser_parse_global_vars():
 
 
 def test_component_get_function_valid():
-    """
-    Test the get_function method of the Component
+    """Test the get_function method of the Component
     class with valid code and function_entrypoint_name.
     """
     component = BaseComponent(_code="def build(): pass", _function_entrypoint_name="build")
@@ -212,8 +184,7 @@ def test_component_get_function_valid():
 
 
 def test_custom_component_get_function_entrypoint_args():
-    """
-    Test the get_function_entrypoint_args
+    """Test the get_function_entrypoint_args
     property of the CustomComponent class.
     """
     custom_component = CustomComponent(_code=code_default, _function_entrypoint_name="build")
@@ -225,28 +196,23 @@ def test_custom_component_get_function_entrypoint_args():
 
 
 def test_custom_component_get_function_entrypoint_return_type():
-    """
-    Test the get_function_entrypoint_return_type
+    """Test the get_function_entrypoint_return_type
     property of the CustomComponent class.
     """
-
     custom_component = CustomComponent(_code=code_default, _function_entrypoint_name="build")
     return_type = custom_component.get_function_entrypoint_return_type
     assert return_type == [Document]
 
 
 def test_custom_component_get_main_class_name():
-    """
-    Test the get_main_class_name property of the CustomComponent class.
-    """
+    """Test the get_main_class_name property of the CustomComponent class."""
     custom_component = CustomComponent(_code=code_default, _function_entrypoint_name="build")
     class_name = custom_component.get_main_class_name
     assert class_name == "YourComponent"
 
 
 def test_custom_component_get_function_valid():
-    """
-    Test the get_function property of the CustomComponent
+    """Test the get_function property of the CustomComponent
     class with valid code and function_entrypoint_name.
     """
     custom_component = CustomComponent(_code="def build(): pass", _function_entrypoint_name="build")
@@ -255,9 +221,7 @@ def test_custom_component_get_function_valid():
 
 
 def test_code_parser_parse_arg_no_annotation():
-    """
-    Test the parse_arg method of the CodeParser class without an annotation.
-    """
+    """Test the parse_arg method of the CodeParser class without an annotation."""
     parser = CodeParser("")
     arg = ast.arg(arg="x", annotation=None)
     result = parser.parse_arg(arg, None)
@@ -266,9 +230,7 @@ def test_code_parser_parse_arg_no_annotation():
 
 
 def test_code_parser_parse_arg_with_annotation():
-    """
-    Test the parse_arg method of the CodeParser class with an annotation.
-    """
+    """Test the parse_arg method of the CodeParser class with an annotation."""
     parser = CodeParser("")
     arg = ast.arg(arg="x", annotation=ast.Name(id="int", ctx=ast.Load()))
     result = parser.parse_arg(arg, None)
@@ -277,8 +239,7 @@ def test_code_parser_parse_arg_with_annotation():
 
 
 def test_code_parser_parse_callable_details_no_args():
-    """
-    Test the parse_callable_details method of the
+    """Test the parse_callable_details method of the
     CodeParser class with a function with no arguments.
     """
     parser = CodeParser("")
@@ -295,9 +256,7 @@ def test_code_parser_parse_callable_details_no_args():
 
 
 def test_code_parser_parse_assign():
-    """
-    Test the parse_assign method of the CodeParser class.
-    """
+    """Test the parse_assign method of the CodeParser class."""
     parser = CodeParser("")
     stmt = ast.Assign(targets=[ast.Name(id="x", ctx=ast.Store())], value=ast.Num(n=1))
     result = parser.parse_assign(stmt)
@@ -306,9 +265,7 @@ def test_code_parser_parse_assign():
 
 
 def test_code_parser_parse_ann_assign():
-    """
-    Test the parse_ann_assign method of the CodeParser class.
-    """
+    """Test the parse_ann_assign method of the CodeParser class."""
     parser = CodeParser("")
     stmt = ast.AnnAssign(
         target=ast.Name(id="x", ctx=ast.Store()),
@@ -323,8 +280,7 @@ def test_code_parser_parse_ann_assign():
 
 
 def test_code_parser_parse_function_def_not_init():
-    """
-    Test the parse_function_def method of the
+    """Test the parse_function_def method of the
     CodeParser class with a function that is not __init__.
     """
     parser = CodeParser("")
@@ -341,8 +297,7 @@ def test_code_parser_parse_function_def_not_init():
 
 
 def test_code_parser_parse_function_def_init():
-    """
-    Test the parse_function_def method of the
+    """Test the parse_function_def method of the
     CodeParser class with an __init__ function.
     """
     parser = CodeParser("")
@@ -359,8 +314,7 @@ def test_code_parser_parse_function_def_init():
 
 
 def test_component_get_code_tree_syntax_error():
-    """
-    Test the get_code_tree method of the Component class
+    """Test the get_code_tree method of the Component class
     raises the CodeSyntaxError when given incorrect syntax.
     """
     component = BaseComponent(_code="import os as", _function_entrypoint_name="build")
@@ -369,8 +323,7 @@ def test_component_get_code_tree_syntax_error():
 
 
 def test_custom_component_class_template_validation_no_code():
-    """
-    Test the _class_template_validation method of the CustomComponent class
+    """Test the _class_template_validation method of the CustomComponent class
     raises the HTTPException when the code is None.
     """
     custom_component = CustomComponent(_code=None, _function_entrypoint_name="build")
@@ -379,8 +332,7 @@ def test_custom_component_class_template_validation_no_code():
 
 
 def test_custom_component_get_code_tree_syntax_error():
-    """
-    Test the get_code_tree method of the CustomComponent class
+    """Test the get_code_tree method of the CustomComponent class
     raises the CodeSyntaxError when given incorrect syntax.
     """
     custom_component = CustomComponent(_code="import os as", _function_entrypoint_name="build")
@@ -389,8 +341,7 @@ def test_custom_component_get_code_tree_syntax_error():
 
 
 def test_custom_component_get_function_entrypoint_args_no_args():
-    """
-    Test the get_function_entrypoint_args property of
+    """Test the get_function_entrypoint_args property of
     the CustomComponent class with a build method with no arguments.
     """
     my_code = """
@@ -405,8 +356,7 @@ class MyMainClass(CustomComponent):
 
 
 def test_custom_component_get_function_entrypoint_return_type_no_return_type():
-    """
-    Test the get_function_entrypoint_return_type property of the
+    """Test the get_function_entrypoint_return_type property of the
     CustomComponent class with a build method with no return type.
     """
     my_code = """
@@ -421,8 +371,7 @@ class MyClass(CustomComponent):
 
 
 def test_custom_component_get_main_class_name_no_main_class():
-    """
-    Test the get_main_class_name property of the
+    """Test the get_main_class_name property of the
     CustomComponent class when there is no main class.
     """
     my_code = """
@@ -435,8 +384,7 @@ def build():
 
 
 def test_custom_component_build_not_implemented():
-    """
-    Test the build method of the CustomComponent
+    """Test the build method of the CustomComponent
     class raises the NotImplementedError.
     """
     custom_component = CustomComponent(_code="def build(): pass", _function_entrypoint_name="build")
@@ -453,7 +401,7 @@ def test_build_config_no_code():
 
 @pytest.fixture
 def component():
-    yield CustomComponent(
+    return CustomComponent(
         field_config={
             "fields": {
                 "llm": {"type": "str"},

src/backend/tests/unit/test_custom_component_with_client.py

@@ -1,7 +1,6 @@
 from pathlib import Path
 
 import pytest
-
 from langflow.custom import Component
 from langflow.custom.custom_component.custom_component import CustomComponent
 from langflow.custom.utils import build_custom_component_template
@@ -11,7 +10,7 @@ from langflow.services.settings.feature_flags import FEATURE_FLAGS
 
 @pytest.fixture
 def code_component_with_multiple_outputs():
-    code = Path("src/backend/tests/data/component_multiple_outputs.py").read_text()
+    code = Path("src/backend/tests/data/component_multiple_outputs.py").read_text(encoding="utf-8")
     return Component(_code=code)
 
 

src/backend/tests/unit/test_data_class.py

@@ -1,6 +1,5 @@
 import pytest
 from langchain_core.documents import Document
-
 from langflow.schema import Data
 
 

src/backend/tests/unit/test_data_components.py

@@ -1,12 +1,11 @@
 import tempfile
 from pathlib import Path
-from unittest.mock import Mock, patch, ANY
+from unittest.mock import ANY, Mock, patch
 
 import httpx
 import pytest
 import respx
 from httpx import Response
-
 from langflow.components import data
 
 
@@ -163,16 +162,16 @@ def test_directory_without_mocks():
     directory_component = data.DirectoryComponent()
 
     with tempfile.TemporaryDirectory() as temp_dir:
-        (Path(temp_dir) / "test.txt").write_text("test")
+        (Path(temp_dir) / "test.txt").write_text("test", encoding="utf-8")
         # also add a json file
-        (Path(temp_dir) / "test.json").write_text('{"test": "test"}')
+        (Path(temp_dir) / "test.json").write_text('{"test": "test"}', encoding="utf-8")
 
         directory_component.set_attributes({"path": str(temp_dir), "use_multithreading": False})
         results = directory_component.load_directory()
         assert len(results) == 2
         values = ["test", '{"test":"test"}']
         assert all(result.text in values for result in results), [
-            (len(result.text), len(val)) for result, val in zip(results, values)
+            (len(result.text), len(val)) for result, val in zip(results, values, strict=True)
         ]
 
     # in ../docs/docs/components there are many mdx files

src/backend/tests/unit/test_database.py

@@ -5,8 +5,6 @@ from uuid import UUID, uuid4
 import orjson
 import pytest
 from fastapi.testclient import TestClient
-from sqlmodel import Session
-
 from langflow.api.v1.schemas import FlowListCreate, ResultDataResponse
 from langflow.graph.utils import log_transaction, log_vertex_build
 from langflow.initial_setup.setup import load_flows_from_directory, load_starter_projects
@@ -15,6 +13,7 @@ from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate
 from langflow.services.database.models.folder.model import FolderCreate
 from langflow.services.database.utils import session_getter
 from langflow.services.deps import get_db_service
+from sqlmodel import Session
 
 
 @pytest.fixture(scope="module")
@@ -520,5 +519,5 @@ def test_sqlite_pragmas():
     with db_service.with_session() as session:
         from sqlalchemy import text
 
-        assert "wal" == session.exec(text("PRAGMA journal_mode;")).scalar()
-        assert 1 == session.exec(text("PRAGMA synchronous;")).scalar()
+        assert session.exec(text("PRAGMA journal_mode;")).scalar() == "wal"
+        assert session.exec(text("PRAGMA synchronous;")).scalar() == 1

src/backend/tests/unit/test_endpoints.py

@@ -4,7 +4,6 @@ from uuid import UUID, uuid4
 import pytest
 from fastapi import status
 from httpx import AsyncClient
-
 from langflow.custom.directory_reader.directory_reader import DirectoryReader
 from langflow.services.deps import get_settings_service
 
@@ -377,7 +376,7 @@ async def test_invalid_prompt(client: AsyncClient):
 
 
 @pytest.mark.parametrize(
-    "prompt,expected_input_variables",
+    ("prompt", "expected_input_variables"),
     [
         ("{color} is my favorite color.", ["color"]),
         ("The weather is {weather} today.", ["weather"]),
@@ -442,13 +441,13 @@ async def test_successful_run_no_payload(client, simple_api_test, created_api_ke
     assert isinstance(outputs_dict.get("outputs"), list)
     assert len(outputs_dict.get("outputs")) == 1
     ids = [output.get("component_id") for output in outputs_dict.get("outputs")]
-    assert all(["ChatOutput" in _id for _id in ids])
+    assert all("ChatOutput" in _id for _id in ids)
     display_names = [output.get("component_display_name") for output in outputs_dict.get("outputs")]
-    assert all([name in display_names for name in ["Chat Output"]])
+    assert all(name in display_names for name in ["Chat Output"])
     output_results_has_results = all("results" in output.get("results") for output in outputs_dict.get("outputs"))
     inner_results = [output.get("results") for output in outputs_dict.get("outputs")]
 
-    assert all([result is not None for result in inner_results]), (outputs_dict, output_results_has_results)
+    assert all(result is not None for result in inner_results), (outputs_dict, output_results_has_results)
 
 
 async def test_successful_run_with_output_type_text(client, simple_api_test, created_api_key):
@@ -473,12 +472,12 @@ async def test_successful_run_with_output_type_text(client, simple_api_test, cre
     assert isinstance(outputs_dict.get("outputs"), list)
     assert len(outputs_dict.get("outputs")) == 1
     ids = [output.get("component_id") for output in outputs_dict.get("outputs")]
-    assert all(["ChatOutput" in _id for _id in ids]), ids
+    assert all("ChatOutput" in _id for _id in ids), ids
     display_names = [output.get("component_display_name") for output in outputs_dict.get("outputs")]
-    assert all([name in display_names for name in ["Chat Output"]]), display_names
+    assert all(name in display_names for name in ["Chat Output"]), display_names
     inner_results = [output.get("results") for output in outputs_dict.get("outputs")]
     expected_keys = ["message"]
-    assert all([key in result for result in inner_results for key in expected_keys]), outputs_dict
+    assert all(key in result for result in inner_results for key in expected_keys), outputs_dict
 
 
 async def test_successful_run_with_output_type_any(client, simple_api_test, created_api_key):
@@ -504,12 +503,12 @@ async def test_successful_run_with_output_type_any(client, simple_api_test, crea
     assert isinstance(outputs_dict.get("outputs"), list)
     assert len(outputs_dict.get("outputs")) == 1
     ids = [output.get("component_id") for output in outputs_dict.get("outputs")]
-    assert all(["ChatOutput" in _id or "TextOutput" in _id for _id in ids]), ids
+    assert all("ChatOutput" in _id or "TextOutput" in _id for _id in ids), ids
     display_names = [output.get("component_display_name") for output in outputs_dict.get("outputs")]
-    assert all([name in display_names for name in ["Chat Output"]]), display_names
+    assert all(name in display_names for name in ["Chat Output"]), display_names
     inner_results = [output.get("results") for output in outputs_dict.get("outputs")]
     expected_keys = ["message"]
-    assert all([key in result for result in inner_results for key in expected_keys]), outputs_dict
+    assert all(key in result for result in inner_results for key in expected_keys), outputs_dict
 
 
 async def test_successful_run_with_output_type_debug(client, simple_api_test, created_api_key):
@@ -566,7 +565,7 @@ async def test_successful_run_with_input_type_text(client, simple_api_test, crea
     # Now we check if the input_value is correct
     # We get text key twice because the output is now a Message
     assert all(
-        [output.get("results").get("text").get("text") == "value1" for output in text_input_outputs]
+        output.get("results").get("text").get("text") == "value1" for output in text_input_outputs
     ), text_input_outputs
 
 
@@ -599,7 +598,7 @@ async def test_successful_run_with_input_type_chat(client: AsyncClient, simple_a
     assert len(chat_input_outputs) == 1
     # Now we check if the input_value is correct
     assert all(
-        [output.get("results").get("message").get("text") == "value1" for output in chat_input_outputs]
+        output.get("results").get("message").get("text") == "value1" for output in chat_input_outputs
     ), chat_input_outputs
 
 
@@ -653,7 +652,7 @@ async def test_successful_run_with_input_type_any(client, simple_api_test, creat
         result_dict.get("message", result_dict.get("text")) for result_dict in all_result_dicts
     ]
     assert all(
-        [message_or_text_dict.get("text") == "value1" for message_or_text_dict in all_message_or_text_dicts]
+        message_or_text_dict.get("text") == "value1" for message_or_text_dict in all_message_or_text_dicts
     ), any_input_outputs
 
 

src/backend/tests/unit/test_files.py

@@ -8,10 +8,9 @@ from unittest.mock import MagicMock
 import pytest
 from asgi_lifespan import LifespanManager
 from httpx import ASGITransport, AsyncClient
-from sqlmodel import Session
-
 from langflow.services.deps import get_storage_service
 from langflow.services.storage.service import StorageService
+from sqlmodel import Session
 
 
 @pytest.fixture
@@ -26,7 +25,7 @@ def mock_storage_service():
     return service
 
 
-@pytest.fixture(name="files_client", scope="function")
+@pytest.fixture(name="files_client")
 async def files_client_fixture(session: Session, monkeypatch, request, load_flows_dir, mock_storage_service):
     # Set the database url to a test database
     if "noclient" in request.keywords:

src/backend/tests/unit/test_frontend_nodes.py

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.template.field.base import Input
 from langflow.template.frontend_node.base import FrontendNode
 from langflow.template.template.base import Template

src/backend/tests/unit/test_helper_components.py

@@ -5,7 +5,6 @@ from langflow.custom.utils import build_custom_component_template
 from langflow.schema import Data
 from langflow.schema.message import Message
 
-
 # def test_update_data_component():
 #     # Arrange
 #     update_data_component = helpers.UpdateDataComponent()
@@ -36,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()
+    uuid_generator_component._code = Path(helpers.IDGenerator.__file__).read_text(encoding="utf-8")
 
     frontend_node, _ = build_custom_component_template(uuid_generator_component)
 

src/backend/tests/unit/test_initial_setup.py

@@ -2,8 +2,6 @@ from datetime import datetime
 from pathlib import Path
 
 import pytest
-from sqlmodel import select
-
 from langflow.custom.directory_reader.utils import build_custom_component_list_from_path
 from langflow.initial_setup.setup import (
     STARTER_FOLDER_NAME,
@@ -14,6 +12,7 @@ from langflow.initial_setup.setup import (
 from langflow.interface.types import aget_all_types_dict
 from langflow.services.database.models.folder.model import Folder
 from langflow.services.deps import session_scope
+from sqlmodel import select
 
 
 def test_load_starter_projects():
@@ -101,10 +100,11 @@ async def test_create_or_update_starter_projects():
 
 
 def find_componeny_by_name(components, name):
-    for category, children in components.items():
+    for children in components.values():
         if name in children:
             return children[name]
-    raise ValueError(f"Component {name} not found in components")
+    msg = f"Component {name} not found in components"
+    raise ValueError(msg)
 
 
 def set_value(component, input_name, value):

src/backend/tests/unit/test_kubernetes_secrets.py

@@ -1,9 +1,9 @@
-import pytest
-from unittest.mock import MagicMock
-from kubernetes.client import V1ObjectMeta, V1Secret
 from base64 import b64encode
+from unittest.mock import MagicMock
 from uuid import UUID
 
+import pytest
+from kubernetes.client import V1ObjectMeta, V1Secret
 from langflow.services.variable.kubernetes_secrets import KubernetesSecretManager, encode_user_id
 
 

src/backend/tests/unit/test_loading.py

@@ -2,7 +2,6 @@ from langflow.graph import Graph
 from langflow.initial_setup.setup import load_starter_projects
 from langflow.load import load_flow_from_json
 
-
 # TODO: UPDATE BASIC EXAMPLE
 # def test_load_flow_from_json():
 #     """Test loading a flow from a json file"""
@@ -20,9 +19,8 @@ from langflow.load import load_flow_from_json
 
 
 def test_load_flow_from_json_object():
-    """Test loading a flow from a json file and applying tweaks"""
-    _, projects = zip(*load_starter_projects())
-    project = projects[0]
+    """Test loading a flow from a json file and applying tweaks."""
+    project = load_starter_projects()[0][1]
     loaded = load_flow_from_json(project)
     assert loaded is not None
     assert isinstance(loaded, Graph)

src/backend/tests/unit/test_logger.py

@@ -1,7 +1,8 @@
-import pytest
-import os
 import json
+import os
 from unittest.mock import patch
+
+import pytest
 from langflow.logging.logger import SizedLogBuffer
 
 
@@ -27,8 +28,8 @@ def test_write(sized_log_buffer):
     sized_log_buffer.max = 1  # Set max size to 1 for testing
     sized_log_buffer.write(message)
     assert len(sized_log_buffer.buffer) == 1
-    assert 1625097600124 == sized_log_buffer.buffer[0][0]
-    assert "Test log" == sized_log_buffer.buffer[0][1]
+    assert sized_log_buffer.buffer[0][0] == 1625097600124
+    assert sized_log_buffer.buffer[0][1] == "Test log"
 
 
 def test_write_overflow(sized_log_buffer):
@@ -38,8 +39,8 @@ def test_write_overflow(sized_log_buffer):
         sized_log_buffer.write(message)
 
     assert len(sized_log_buffer.buffer) == 2
-    assert 1625097601000 == sized_log_buffer.buffer[0][0]
-    assert 1625097602000 == sized_log_buffer.buffer[1][0]
+    assert sized_log_buffer.buffer[0][0] == 1625097601000
+    assert sized_log_buffer.buffer[1][0] == 1625097602000
 
 
 def test_len(sized_log_buffer):

src/backend/tests/unit/test_login.py

@@ -1,9 +1,8 @@
 import pytest
-from sqlalchemy.exc import IntegrityError
-
 from langflow.services.auth.utils import get_password_hash
 from langflow.services.database.models.user import User
 from langflow.services.deps import session_scope
+from sqlalchemy.exc import IntegrityError
 
 
 @pytest.fixture

src/backend/tests/unit/test_messages.py

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.memory import add_messages, add_messagetables, delete_messages, get_messages, store_message
 from langflow.schema.message import Message
 
@@ -10,17 +9,16 @@ from langflow.services.deps import session_scope
 from langflow.services.tracing.utils import convert_to_langchain_type
 
 
-@pytest.fixture()
+@pytest.fixture
 def created_message():
     with session_scope() as session:
         message = MessageCreate(text="Test message", sender="User", sender_name="User", session_id="session_id")
         messagetable = MessageTable.model_validate(message, from_attributes=True)
         messagetables = add_messagetables([messagetable], session)
-        message_read = MessageRead.model_validate(messagetables[0], from_attributes=True)
-        return message_read
+        return MessageRead.model_validate(messagetables[0], from_attributes=True)
 
 
-@pytest.fixture()
+@pytest.fixture
 def created_messages(session):
     with session_scope() as session:
         messages = [
@@ -30,10 +28,7 @@ def created_messages(session):
         ]
         messagetables = [MessageTable.model_validate(message, from_attributes=True) for message in messages]
         messagetables = add_messagetables(messagetables, session)
-        messages_read = [
-            MessageRead.model_validate(messagetable, from_attributes=True) for messagetable in messagetables
-        ]
-        return messages_read
+        return [MessageRead.model_validate(messagetable, from_attributes=True) for messagetable in messagetables]
 
 
 @pytest.mark.usefixtures("client")
@@ -87,10 +82,10 @@ def test_convert_to_langchain(method_name):
     def convert(value):
         if method_name == "message":
             return value.to_lc_message()
-        elif method_name == "convert_to_langchain_type":
+        if method_name == "convert_to_langchain_type":
             return convert_to_langchain_type(value)
-        else:
-            raise ValueError(f"Invalid method: {method_name}")
+        msg = f"Invalid method: {method_name}"
+        raise ValueError(msg)
 
     lc_message = convert(Message(text="Test message 1", sender="User", sender_name="User", session_id="session_id2"))
     assert lc_message.content == "Test message 1"

src/backend/tests/unit/test_messages_endpoints.py

@@ -2,7 +2,6 @@ from uuid import UUID
 
 import pytest
 from httpx import AsyncClient
-
 from langflow.memory import add_messagetables
 
 # Assuming you have these imports available
@@ -11,17 +10,16 @@ from langflow.services.database.models.message.model import MessageTable
 from langflow.services.deps import session_scope
 
 
-@pytest.fixture()
+@pytest.fixture
 async def created_message():
     with session_scope() as session:
         message = MessageCreate(text="Test message", sender="User", sender_name="User", session_id="session_id")
         messagetable = MessageTable.model_validate(message, from_attributes=True)
         messagetables = add_messagetables([messagetable], session)
-        message_read = MessageRead.model_validate(messagetables[0], from_attributes=True)
-        return message_read
+        return MessageRead.model_validate(messagetables[0], from_attributes=True)
 
 
-@pytest.fixture()
+@pytest.fixture
 def created_messages(session):
     with session_scope() as session:
         messages = [
@@ -30,9 +28,7 @@ def created_messages(session):
             MessageCreate(text="Test message 3", sender="User", sender_name="User", session_id="session_id2"),
         ]
         messagetables = [MessageTable.model_validate(message, from_attributes=True) for message in messages]
-        message_list = add_messagetables(messagetables, session)
-
-        return message_list
+        return add_messagetables(messagetables, session)
 
 
 @pytest.mark.api_key_required

src/backend/tests/unit/test_process.py

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.processing.process import process_tweaks
 from langflow.services.deps import get_session_service
 

src/backend/tests/unit/test_schema.py

@@ -1,14 +1,13 @@
+from collections.abc import Sequence as SequenceABC
 from types import NoneType
 from typing import Union
 
-from langflow.schema.data import Data
 import pytest
-from pydantic import ValidationError
-
+from langflow.schema.data import Data
 from langflow.template import Input, Output
 from langflow.template.field.base import UNDEFINED
 from langflow.type_extraction.type_extraction import post_process_type
-from collections.abc import Sequence as SequenceABC
+from pydantic import ValidationError
 
 
 class TestInput:
@@ -68,8 +67,8 @@ class TestInput:
         assert set(post_process_type(Union[None, list[None]])) == {None, NoneType}
 
         # Handling complex nested structures
-        assert set(post_process_type(Union[SequenceABC[Union[int, str]], list[float]])) == {int, str, float}
-        assert set(post_process_type(Union[Union[Union[int, list[str]], list[float]], str])) == {int, str, float}
+        assert set(post_process_type(Union[SequenceABC[int | str], list[float]])) == {int, str, float}
+        assert set(post_process_type(Union[int | list[str] | list[float], str])) == {int, str, float}
 
         # Non-generic types should return as is
         assert set(post_process_type(dict)) == {dict}
@@ -87,12 +86,12 @@ class TestInput:
         assert set(post_process_type(Data | Union[float, None])) == {Data, float, type(None)}
 
         # Multiple Data types combined
-        assert set(post_process_type(Union[Data, Union[str, float]])) == {Data, str, float}
+        assert set(post_process_type(Union[Data, str | float])) == {Data, str, float}
         assert set(post_process_type(Union[Data | float | str, int])) == {Data, int, float, str}
 
         # Testing with nested unions and lists
-        assert set(post_process_type(Union[list[Data], list[Union[int, str]]])) == {Data, int, str}
-        assert set(post_process_type(Data | list[Union[float, str]])) == {Data, float, str}
+        assert set(post_process_type(Union[list[Data], list[int | str]])) == {Data, int, str}
+        assert set(post_process_type(Data | list[float | str])) == {Data, float, str}
 
     def test_input_to_dict(self):
         input_obj = Input(field_type="str")

src/backend/tests/unit/test_telemetry.py

@@ -1,8 +1,8 @@
-import pytest
 import threading
-from langflow.services.telemetry.opentelemetry import OpenTelemetry
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
+import pytest
+from langflow.services.telemetry.opentelemetry import OpenTelemetry
 
 fixed_labels = {"flow_id": "this_flow_id", "service": "this", "user": "that"}
 
@@ -72,9 +72,9 @@ def test_missing_labels(opentelemetry_instance):
     with pytest.raises(ValueError, match="Labels must be provided for the metric"):
         opentelemetry_instance.up_down_counter("num_files_uploaded", 1, None)
     with pytest.raises(ValueError, match="Labels must be provided for the metric"):
-        opentelemetry_instance.update_gauge(metric_name="num_files_uploaded", value=1.0, labels=dict())
+        opentelemetry_instance.update_gauge(metric_name="num_files_uploaded", value=1.0, labels={})
     with pytest.raises(ValueError, match="Labels must be provided for the metric"):
-        opentelemetry_instance.observe_histogram("num_files_uploaded", 1, dict())
+        opentelemetry_instance.observe_histogram("num_files_uploaded", 1, {})
 
 
 def test_multithreaded_singleton():

src/backend/tests/unit/test_template.py

@@ -7,13 +7,13 @@ from pydantic import BaseModel
 
 # Dummy classes for testing purposes
 class Parent(BaseModel):
-    """Parent Class"""
+    """Parent Class."""
 
     parent_field: str
 
 
 class Child(Parent):
-    """Child Class"""
+    """Child Class."""
 
     child_field: int
 
@@ -85,7 +85,7 @@ def test_get_default_factory():
         return "default_value"
 
     # Add dummy_function to your_module
-    setattr(importlib.import_module(module_name), "dummy_function", dummy_function)
+    importlib.import_module(module_name).dummy_function = dummy_function
 
     default_value = get_default_factory(module_name, function_repr)
 

src/backend/tests/unit/test_user.py

@@ -2,13 +2,12 @@ from datetime import datetime
 
 import pytest
 from httpx import AsyncClient
-from sqlmodel import select
-
 from langflow.services.auth.utils import create_super_user, get_password_hash
 from langflow.services.database.models.user import UserUpdate
 from langflow.services.database.models.user.model import User
 from langflow.services.database.utils import session_getter
 from langflow.services.deps import get_db_service, get_settings_service
+from sqlmodel import select
 
 
 @pytest.fixture
@@ -86,7 +85,7 @@ async def test_user_waiting_for_approval(client):
     with session_getter(get_db_service()) as session:
         existing_user = session.exec(select(User).where(User.username == username)).first()
         if existing_user:
-            print(f"User {username} still exists after the test. This is expected.")
+            pass
         else:
             pytest.fail(f"User {username} does not exist after the test. This is unexpected.")
 

src/backend/tests/unit/test_validate_code.py

@@ -2,9 +2,8 @@ from pathlib import Path
 from unittest import mock
 
 import pytest
-from requests.exceptions import MissingSchema
-
 from langflow.utils.validate import create_function, execute_function, extract_function_name, validate_code
+from requests.exceptions import MissingSchema
 
 
 def test_create_function():
@@ -99,6 +98,5 @@ import requests
 def my_function(x):
     return requests.get(x).text
     """
-    with mock.patch("requests.get", side_effect=MissingSchema):
-        with pytest.raises(MissingSchema):
-            execute_function(code, "my_function", "invalid_url")
+    with mock.patch("requests.get", side_effect=MissingSchema), pytest.raises(MissingSchema):
+        execute_function(code, "my_function", "invalid_url")

src/backend/tests/unit/test_version.py

@@ -9,9 +9,9 @@ def test_version():
 
 
 def test_compute_main():
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10.post0")
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10.a1")
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10.b112")
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10.rc0")
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10.dev9")
-    assert "1.0.10" == _compute_non_prerelease_version("1.0.10")
+    assert _compute_non_prerelease_version("1.0.10.post0") == "1.0.10"
+    assert _compute_non_prerelease_version("1.0.10.a1") == "1.0.10"
+    assert _compute_non_prerelease_version("1.0.10.b112") == "1.0.10"
+    assert _compute_non_prerelease_version("1.0.10.rc0") == "1.0.10"
+    assert _compute_non_prerelease_version("1.0.10.dev9") == "1.0.10"
+    assert _compute_non_prerelease_version("1.0.10") == "1.0.10"

src/backend/tests/unit/utils/test_connection_string_parser.py

@@ -3,7 +3,7 @@ from langflow.utils.connection_string_parser import transform_connection_string
 
 
 @pytest.mark.parametrize(
-    "connection_string, expected",
+    ("connection_string", "expected"),
     [
         ("protocol:user:password@host", "protocol:user:password@host"),
         ("protocol:user@host", "protocol:user@host"),

src/backend/tests/unit/utils/test_format_directory_path.py

@@ -3,7 +3,7 @@ from langflow.base.data.utils import format_directory_path
 
 
 @pytest.mark.parametrize(
-    "input_path, expected",
+    ("input_path", "expected"),
     [
         # Test case 1: Standard path with no newlines (no change expected)
         ("/home/user/documents/file.txt", "/home/user/documents/file.txt"),

src/backend/tests/unit/utils/test_rewrite_file_path.py

@@ -1,9 +1,9 @@
-from langflow.base.data.utils import format_directory_path
 import pytest
+from langflow.base.data.utils import format_directory_path
 
 
 @pytest.mark.parametrize(
-    "input_path, expected",
+    ("input_path", "expected"),
     [
         # Test case 1: Standard path with no newlines
         ("/home/user/documents/file.txt", "/home/user/documents/file.txt"),

src/backend/tests/unit/utils/test_truncate_long_strings.py

@@ -1,9 +1,11 @@
+import math
+
 import pytest
 from langflow.utils.util_strings import truncate_long_strings
 
 
 @pytest.mark.parametrize(
-    "input_data, max_length, expected",
+    ("input_data", "max_length", "expected"),
     [
         # Test case 1: String shorter than max_length
         ("short string", 20, "short string"),
@@ -20,7 +22,7 @@ from langflow.utils.util_strings import truncate_long_strings
         # Test case 7: Integer input
         (12345, 3, 12345),
         # Test case 8: Float input
-        (3.14159, 4, 3.14159),
+        (math.pi, 4, math.pi),
         # Test case 9: Boolean input
         (True, 2, True),
         # Test case 10: None input

src/backend/tests/unit/utils/test_truncate_long_strings_on_objects.py

@@ -1,10 +1,10 @@
-from langflow.utils.util_strings import truncate_long_strings
-from langflow.utils.constants import MAX_TEXT_LENGTH
 import pytest
+from langflow.utils.constants import MAX_TEXT_LENGTH
+from langflow.utils.util_strings import truncate_long_strings
 
 
 @pytest.mark.parametrize(
-    "input_data, max_length, expected",
+    ("input_data", "max_length", "expected"),
     [
         # Test case 1: Simple string truncation
         ({"key": "a" * 100}, 10, {"key": "a" * 10 + "..."}),