From 6a6d0e738965ab1c666f8c88ba420149a8d6a12d Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Wed, 19 Jun 2024 01:25:48 -0300
Subject: [PATCH] fix: rearrange imports

---
 src/backend/base/langflow/components/data/Directory.py    | 8 +++-----
 .../langflow/components/models/BaiduQianfanChatModel.py   | 3 ++-
 .../base/langflow/components/models/ChatLiteLLMModel.py   | 2 +-
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/backend/base/langflow/components/data/Directory.py b/src/backend/base/langflow/components/data/Directory.py
index 0cf69eb69..0b381274d 100644
--- a/src/backend/base/langflow/components/data/Directory.py
+++ b/src/backend/base/langflow/components/data/Directory.py
@@ -1,12 +1,10 @@
 from typing import List, Optional
 
-from langflow.base.data.utils import (parallel_load_data,
-                                      parse_text_file_to_data,
+from langflow.base.data.utils import parallel_load_data, parse_text_file_to_data, retrieve_file_paths
+from langflow.custom import Component
 from langflow.io import BoolInput, IntInput, TextInput
 from langflow.schema import Data
-from langflow.io import Output
-from langflow.schema import Data
-from langflow.io import Output
+from langflow.template import Output
 
 
 class DirectoryComponent(Component):
diff --git a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
index 879ed8c5d..c4ef67aa9 100644
--- a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
+++ b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
@@ -1,4 +1,4 @@
-from langchain_community.chat_models.baidu_qianfan_endpoint import \
+from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint
 from pydantic.v1 import SecretStr
 
 from langflow.base.constants import STREAM_INFO_TEXT
@@ -6,6 +6,7 @@ from langflow.base.models.model import LCModelComponent
 from langflow.field_typing import BaseLanguageModel, Text
 from langflow.io import BoolInput, FloatInput, Output, SecretStrInput, StrInput
 
+
 class QianfanChatEndpointComponent(LCModelComponent):
     display_name: str = "Qianfan"
     description: str = "Generate text using Baidu Qianfan LLMs."
diff --git a/src/backend/base/langflow/components/models/ChatLiteLLMModel.py b/src/backend/base/langflow/components/models/ChatLiteLLMModel.py
index 5cd799445..cb1380dfb 100644
--- a/src/backend/base/langflow/components/models/ChatLiteLLMModel.py
+++ b/src/backend/base/langflow/components/models/ChatLiteLLMModel.py
@@ -4,9 +4,9 @@ from langchain_community.chat_models.litellm import ChatLiteLLM, ChatLiteLLMExce
 
 from langflow.base.constants import STREAM_INFO_TEXT
 from langflow.base.models.model import LCModelComponent
+from langflow.field_typing import BaseLanguageModel
 from langflow.io import BoolInput, DictInput, DropdownInput, FloatInput, IntInput, MessageInput, Output, StrInput
 from langflow.schema.message import Message
-from langflow.schema.message import Message
 
 
 class ChatLiteLLMModelComponent(LCModelComponent):