refactor(langflow): organize imports in multiple files

This commit organizes imports in multiple files to follow PEP8 guidelines
and improve code readability. No functionality changes were made.
This commit is contained in:
Gabriel Almeida 2023-05-10 11:42:36 -03:00
commit f83160cf21
7 changed files with 6 additions and 9 deletions

View file

@ -1,8 +1,7 @@
from fastapi import APIRouter, WebSocket
from fastapi import APIRouter, WebSocket, WebSocketDisconnect, WebSocketException, status
from langflow.api.chat_manager import ChatManager
from langflow.utils.logger import logger
from fastapi import status, WebSocketDisconnect, WebSocketException
router = APIRouter()
chat_manager = ChatManager()

View file

@ -5,7 +5,6 @@ from typing import Dict, List
from fastapi import WebSocket
from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
from langflow.cache import cache_manager
from langflow.cache.manager import Subject

View file

@ -3,14 +3,14 @@ from typing import Any, Dict
from fastapi import APIRouter, HTTPException
from langflow.interface.run import process_graph_cached
from langflow.interface.types import build_langchain_types_dict
from langflow.api.schemas import (
ExportedFlow,
GraphData,
PredictRequest,
PredictResponse,
)
from langflow.interface.run import process_graph_cached
from langflow.interface.types import build_langchain_types_dict
# build router
router = APIRouter()

View file

@ -1,4 +1,4 @@
from typing import Any, Union, Dict, List
from typing import Any, Dict, List, Union
from pydantic import BaseModel, validator

View file

@ -27,7 +27,6 @@ from langchain.agents.agent_toolkits.vectorstore.prompt import (
from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS
from langchain.agents.mrkl.prompt import FORMAT_INSTRUCTIONS as SQL_FORMAT_INSTRUCTIONS
from langchain.base_language import BaseLanguageModel
from langchain.memory.chat_memory import BaseChatMemory
from langchain.sql_database import SQLDatabase
from langchain.tools.python.tool import PythonAstREPLTool

View file

@ -5,9 +5,9 @@ from typing import Any, Type
from langchain import PromptTemplate
from langchain.agents import Agent
from langchain.base_language import BaseLanguageModel
from langchain.chains.base import Chain
from langchain.chat_models.base import BaseChatModel
from langchain.base_language import BaseLanguageModel
from langchain.tools import BaseTool

View file

@ -3,8 +3,8 @@ import io
from typing import Any, Dict
from chromadb.errors import NotEnoughElementsException
from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler # type: ignore
from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler # type: ignore
from langflow.cache.base import compute_dict_hash, load_cache, memoize_dict
from langflow.graph.graph import Graph
from langflow.interface import loading