fix more linting issues

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-22 18:15:23 -03:00
commit 200f5d647c
15 changed files with 18 additions and 38 deletions

View file

@ -10,7 +10,7 @@ from langflow.services.database.models.flow.flow import Flow, FlowCreate
from langflow.services.database.models.user.user import User, UserCreate
import orjson
from langflow.services.database.utils import session_getter
from langflow.services.getters import get_db_service, get_session
from langflow.services.getters import get_db_service
import pytest
from fastapi.testclient import TestClient
from httpx import AsyncClient
@ -88,10 +88,7 @@ def setup_env(monkeypatch):
@pytest.fixture(name="distributed_client")
def distributed_client_fixture(session: Session, monkeypatch, distributed_env):
# Here we load the .env from ../deploy/.env
from dotenv import load_dotenv
from langflow.services.task import manager
from langflow.core import celery_app
from langflow.services.manager import reinitialize_services, initialize_services
db_dir = tempfile.mkdtemp()
db_path = Path(db_dir) / "test.db"

View file

@ -3,7 +3,6 @@ import os
from pathlib import Path
import pickle
from typing import Type, Union
from langflow import graph
from langflow.graph.edge.base import Edge
from langflow.graph.vertex.base import Vertex
from langchain.agents import AgentExecutor
@ -12,11 +11,9 @@ from langchain.chains.base import Chain
from langchain.llms.fake import FakeListLLM
from langflow.graph import Graph
from langflow.graph.vertex.types import (
AgentVertex,
FileToolVertex,
LLMVertex,
ToolkitVertex,
VectorStoreVertex,
)
from langflow.processing.process import get_result_and_thought
from langflow.utils.payload import get_root_node

View file

@ -1,4 +1,3 @@
from unittest import mock
from unittest.mock import patch, Mock, MagicMock, call
from langflow.services.database.models.user.user import User
from langflow.services.settings.constants import (
@ -6,7 +5,6 @@ from langflow.services.settings.constants import (
DEFAULT_SUPERUSER_PASSWORD,
)
from langflow.services.utils import (
initialize_settings_service,
setup_superuser,
teardown_superuser,
)