Code Formatting (#1828)

embedchain/embedchain/llm/aws_bedrock.py

@@ -45,8 +45,9 @@ class AWSBedrockLlm(BaseLlm):
         }
 
         if config.stream:
-            from langchain.callbacks.streaming_stdout import \
-                StreamingStdOutCallbackHandler
+            from langchain.callbacks.streaming_stdout import (
+                StreamingStdOutCallbackHandler,
+            )
 
             kwargs["streaming"] = True
             kwargs["callbacks"] = [StreamingStdOutCallbackHandler()]

embedchain/embedchain/llm/base.py

@@ -5,7 +5,6 @@ from typing import Any, Optional
 
 from langchain.schema import BaseMessage as LCBaseMessage
 
-from embedchain.constants import SQLITE_PATH
 from embedchain.config import BaseLlmConfig
 from embedchain.config.llm.base import (
     DEFAULT_PROMPT,
@@ -13,6 +12,7 @@ from embedchain.config.llm.base import (
     DEFAULT_PROMPT_WITH_MEM0_MEMORY_TEMPLATE,
     DOCS_SITE_PROMPT_TEMPLATE,
 )
+from embedchain.constants import SQLITE_PATH
 from embedchain.core.db.database import init_db, setup_engine
 from embedchain.helpers.json_serializable import JSONSerializable
 from embedchain.memory.base import ChatHistory

embedchain/embedchain/llm/gpt4all.py

@@ -26,8 +26,7 @@ class GPT4ALLLlm(BaseLlm):
     @staticmethod
     def _get_instance(model):
         try:
-            from langchain_community.llms.gpt4all import \
-                GPT4All as LangchainGPT4All
+            from langchain_community.llms.gpt4all import GPT4All as LangchainGPT4All
         except ModuleNotFoundError:
             raise ModuleNotFoundError(
                 "The GPT4All python package is not installed. Please install it with `pip install --upgrade embedchain[opensource]`"  # noqa E501

embedchain/embedchain/llm/jina.py

@@ -35,8 +35,9 @@ class JinaLlm(BaseLlm):
         if config.top_p:
             kwargs["model_kwargs"]["top_p"] = config.top_p
         if config.stream:
-            from langchain.callbacks.streaming_stdout import \
-                StreamingStdOutCallbackHandler
+            from langchain.callbacks.streaming_stdout import (
+                StreamingStdOutCallbackHandler,
+            )
 
             chat = JinaChat(**kwargs, streaming=config.stream, callbacks=[StreamingStdOutCallbackHandler()])
         else: