[autofix.ci] apply automated fixes

src/backend/tests/conftest.py

@@ -15,11 +15,6 @@
 from dotenv import load_dotenv
 from fastapi.testclient import TestClient
 from httpx import ASGITransport, AsyncClient
-from loguru import logger
-from sqlmodel import Session, SQLModel, create_engine, select
-from sqlmodel.pool import StaticPool
-from typer.testing import CliRunner
-
 from langflow.components.inputs import ChatInput
 from langflow.graph import Graph
 from langflow.initial_setup.setup import STARTER_FOLDER_NAME
@@ -32,6 +27,11 @@
 from langflow.services.database.models.vertex_builds.crud import delete_vertex_builds_by_flow_id
 from langflow.services.database.utils import session_getter
 from langflow.services.deps import get_db_service
+from loguru import logger
+from sqlmodel import Session, SQLModel, create_engine, select
+from sqlmodel.pool import StaticPool
+from typer.testing import CliRunner
+
 from tests.api_keys import get_openai_api_key
 
 if TYPE_CHECKING:

src/backend/tests/integration/backward_compatibility/test_starter_projects.py

@@ -1,6 +1,6 @@
 import pytest
-
 from langflow.schema.message import Message
+
 from tests.api_keys import get_openai_api_key
 from tests.integration.utils import download_flow_from_github, run_json_flow
 

src/backend/tests/integration/components/astra/test_astra_component.py

@@ -3,10 +3,10 @@
 import pytest
 from astrapy.db import AstraDB
 from langchain_core.documents import Document
-
 from langflow.components.embeddings import OpenAIEmbeddingsComponent
 from langflow.components.vectorstores import AstraVectorStoreComponent
 from langflow.schema.data import Data
+
 from tests.api_keys import get_astradb_api_endpoint, get_astradb_application_token, get_openai_api_key
 from tests.integration.components.mock_components import TextToData
 from tests.integration.utils import ComponentInputHandle, run_single_component

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

@@ -1,8 +1,8 @@
 import pytest
-
 from langflow.components.helpers import ParseJSONDataComponent
 from langflow.components.inputs import ChatInput
 from langflow.schema import Data
+
 from tests.integration.components.mock_components import TextToData
 from tests.integration.utils import ComponentInputHandle, run_single_component
 

src/backend/tests/integration/components/inputs/test_chat_input.py

@@ -1,8 +1,8 @@
 import pytest
-
 from langflow.components.inputs import ChatInput
 from langflow.memory import get_messages
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_single_component
 
 

src/backend/tests/integration/components/inputs/test_text_input.py

@@ -1,7 +1,7 @@
 import pytest
-
 from langflow.components.inputs import TextInputComponent
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_single_component
 
 

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

@@ -1,10 +1,10 @@
 import os
 
 import pytest
-
 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/integration/components/outputs/test_chat_output.py

@@ -1,8 +1,8 @@
 import pytest
-
 from langflow.components.outputs import ChatOutput
 from langflow.memory import get_messages
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_single_component
 
 

src/backend/tests/integration/components/outputs/test_text_output.py

@@ -1,7 +1,7 @@
 import pytest
-
 from langflow.components.outputs import TextOutputComponent
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_single_component
 
 

src/backend/tests/integration/components/prompts/test_prompt.py

@@ -1,7 +1,7 @@
 import pytest
-
 from langflow.components.prompts import PromptComponent
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_single_component
 
 

src/backend/tests/integration/flows/test_basic_prompting.py

@@ -1,10 +1,10 @@
 import pytest
-
 from langflow.components.inputs import ChatInput
 from langflow.components.outputs import ChatOutput
 from langflow.components.prompts import PromptComponent
 from langflow.graph import Graph
 from langflow.schema.message import Message
+
 from tests.integration.utils import run_flow
 
 

src/backend/tests/integration/test_misc.py

@@ -3,7 +3,6 @@
 import pytest
 from fastapi import status
 from fastapi.testclient import TestClient
-
 from langflow.graph.schema import RunOutputs
 from langflow.initial_setup.setup import load_starter_projects
 from langflow.load import run_flow_from_json

src/backend/tests/integration/utils.py

@@ -5,7 +5,6 @@
 
 import requests
 from astrapy.admin import parse_api_endpoint
-
 from langflow.api.v1.schemas import InputValueRequest
 from langflow.custom import Component
 from langflow.field_typing import Embeddings

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 import ToolCallingAgentComponent
 from langflow.components.inputs import ChatInput

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

@@ -2,11 +2,10 @@
 
 import pytest
 from langchain_core.language_models import BaseLanguageModel
-from pydantic import BaseModel
-from typing_extensions import override
-
 from langflow.components.helpers.structured_output import StructuredOutputComponent
 from langflow.schema.data import Data
+from pydantic import BaseModel
+from typing_extensions import override
 
 
 class TestStructuredOutputComponent:

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

@@ -3,7 +3,6 @@
 
 import pytest
 from langchain_community.chat_models import ChatOllama
-
 from langflow.components.models import ChatOllamaComponent
 
 

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

@@ -1,5 +1,4 @@
 import pytest
-
 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,4 @@
 import pytest
-
 from langflow.components.prototypes import UpdateDataComponent
 from langflow.schema import Data
 

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

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.components.agents import CrewAIAgentComponent, ToolCallingAgentComponent
 from langflow.components.helpers import SequentialTaskComponent
 from langflow.components.inputs import ChatInput

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

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

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

@@ -1,5 +1,4 @@
 import pytest
-
 from langflow.components.inputs import ChatInput
 from langflow.components.models import OpenAIModelComponent
 from langflow.components.outputs 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 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
+from pydantic import Field
 
 
 @pytest.fixture

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

@@ -2,7 +2,6 @@
 from collections import deque
 
 import pytest
-
 from langflow.components.agents import ToolCallingAgentComponent
 from langflow.components.inputs import ChatInput
 from langflow.components.outputs import ChatOutput, TextOutputComponent

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

@@ -1,7 +1,6 @@
 import os
 
 import pytest
-
 from langflow.components.inputs import ChatInput
 from langflow.components.models import OpenAIModelComponent
 from langflow.components.outputs import ChatOutput, TextOutputComponent

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

@@ -1,7 +1,6 @@
 from typing import TYPE_CHECKING
 
 import pytest
-
 from langflow.components.helpers import MemoryComponent
 from langflow.components.inputs import ChatInput
 from langflow.components.models import OpenAIModelComponent

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

@@ -2,7 +2,6 @@
 from typing import TYPE_CHECKING
 
 import pytest
-
 from langflow.graph.graph.runnable_vertices_manager import RunnableVerticesManager
 
 if TYPE_CHECKING:

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 json
 
 import pytest
-
 from langflow.graph import Graph
 from langflow.graph.graph.utils import (
     find_last_node,

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

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

@@ -3,7 +3,6 @@
 from typing import TYPE_CHECKING
 
 import pytest
-
 from langflow.components.helpers import MemoryComponent
 from langflow.components.inputs import ChatInput
 from langflow.components.models import OpenAIModelComponent

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

@@ -3,7 +3,6 @@
 from textwrap import dedent
 
 import pytest
-
 from langflow.components.data import FileComponent
 from langflow.components.embeddings import OpenAIEmbeddingsComponent
 from langflow.components.helpers import ParseDataComponent, SplitTextComponent

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.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,7 +1,6 @@
 from typing import TYPE_CHECKING, Literal
 
 import pytest
-
 from langflow.components.inputs import ChatInput
 
 if TYPE_CHECKING:

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 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 sqlmodel import Session, SQLModel, create_engine
 
 
 @pytest.fixture

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
 
 

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):

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 @@
 
 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

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

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