Lint and formatting fixes

embedchain/__init__.py

@@ -6,5 +6,4 @@
 from embedchain.apps.CustomApp import CustomApp  # noqa: F401
 from embedchain.apps.Llama2App import Llama2App  # noqa: F401
 from embedchain.apps.OpenSourceApp import OpenSourceApp  # noqa: F401
-from embedchain.apps.PersonApp import (PersonApp,  # noqa: F401
-                                       PersonOpenSourceApp)
+from embedchain.apps.PersonApp import PersonApp, PersonOpenSourceApp  # noqa: F401

embedchain/bots/__init__.py

@@ -1,4 +1,3 @@
-from embedchain.bots.poe import PoeBot
-from embedchain.bots.whatsapp import WhatsAppBot
-# TODO: fix discord import
-# from embedchain.bots.discord import DiscordBot
+from embedchain.bots.poe import PoeBot  # noqa: F401
+from embedchain.bots.whatsapp import WhatsAppBot  # noqa: F401
+from embedchain.bots.discord import DiscordBot  # noqa: F401

embedchain/bots/base.py

@@ -1,7 +1,6 @@
 from embedchain import CustomApp
 from embedchain.config import AddConfig, CustomAppConfig, QueryConfig
-from embedchain.helper_classes.json_serializable import (
-    JSONSerializable, register_deserializable)
+from embedchain.helper_classes.json_serializable import JSONSerializable, register_deserializable
 from embedchain.models import EmbeddingFunctions, Providers
 
 

embedchain/bots/whatsapp.py

@@ -3,6 +3,9 @@
 import signal
 import sys
 
+from flask import Flask, request
+from twilio.twiml.messaging_response import MessagingResponse
+
 from embedchain.helper_classes.json_serializable import register_deserializable
 
 from .base import BaseBot
@@ -11,8 +14,6 @@
 @register_deserializable
 class WhatsAppBot(BaseBot):
     def __init__(self):
-        from flask import Flask, request
-        from twilio.twiml.messaging_response import MessagingResponse
         super().__init__()
 
     def handle_message(self, message):

embedchain/config/__init__.py

@@ -5,5 +5,4 @@
 from .BaseConfig import BaseConfig  # noqa: F401
 from .ChatConfig import ChatConfig  # noqa: F401
 from .QueryConfig import QueryConfig  # noqa: F401
-from .vectordbs.ElasticsearchDBConfig import \
-    ElasticsearchDBConfig  # noqa: F401
+from .vectordbs.ElasticsearchDBConfig import ElasticsearchDBConfig  # noqa: F401

embedchain/config/apps/CustomAppConfig.py

@@ -5,8 +5,7 @@
 
 from embedchain.config.vectordbs import ElasticsearchDBConfig
 from embedchain.helper_classes.json_serializable import register_deserializable
-from embedchain.models import (EmbeddingFunctions, Providers, VectorDatabases,
-                               VectorDimensions)
+from embedchain.models import EmbeddingFunctions, Providers, VectorDatabases, VectorDimensions
 
 from .BaseAppConfig import BaseAppConfig
 

embedchain/utils.py

@@ -128,8 +128,7 @@ def detect_datatype(source: Any) -> DataType:
     formatted_source = format_source(str(source), 30)
 
     if url:
-        from langchain.document_loaders.youtube import \
-            ALLOWED_NETLOCK as YOUTUBE_ALLOWED_NETLOCS
+        from langchain.document_loaders.youtube import ALLOWED_NETLOCK as YOUTUBE_ALLOWED_NETLOCS
 
         if url.netloc in YOUTUBE_ALLOWED_NETLOCS:
             logging.debug(f"Source of `{formatted_source}` detected as `youtube_video`.")