fix more linting issues
This commit is contained in:
parent
99a4f01865
commit
200f5d647c
15 changed files with 18 additions and 38 deletions
|
|
@ -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"
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue