diff --git a/src/backend/langflow/api/chat.py b/src/backend/langflow/api/chat.py index 436f64f7c..490aed12c 100644 --- a/src/backend/langflow/api/chat.py +++ b/src/backend/langflow/api/chat.py @@ -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() diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py index 90ad2632b..6a4111fb4 100644 --- a/src/backend/langflow/api/chat_manager.py +++ b/src/backend/langflow/api/chat_manager.py @@ -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 diff --git a/src/backend/langflow/api/endpoints.py b/src/backend/langflow/api/endpoints.py index d4e592901..64973e163 100644 --- a/src/backend/langflow/api/endpoints.py +++ b/src/backend/langflow/api/endpoints.py @@ -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() diff --git a/src/backend/langflow/api/schemas.py b/src/backend/langflow/api/schemas.py index 2d14bad50..f73b0642d 100644 --- a/src/backend/langflow/api/schemas.py +++ b/src/backend/langflow/api/schemas.py @@ -1,4 +1,4 @@ -from typing import Any, Union, Dict, List +from typing import Any, Dict, List, Union from pydantic import BaseModel, validator diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index 6d2efed68..5fab9bf77 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -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 diff --git a/src/backend/langflow/interface/importing/utils.py b/src/backend/langflow/interface/importing/utils.py index 0d14249cb..8c65bf60e 100644 --- a/src/backend/langflow/interface/importing/utils.py +++ b/src/backend/langflow/interface/importing/utils.py @@ -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 diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py index bab211c2a..05e4b49d6 100644 --- a/src/backend/langflow/interface/run.py +++ b/src/backend/langflow/interface/run.py @@ -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