[Refactor] Improve logging package wide (#1315)

2024-03-13 17:13:30 -07:00
parent ef69c91b60
commit 3616eaadb4
54 changed files with 263 additions and 231 deletions
--- a/embedchain/llm/aws_bedrock.py
+++ b/embedchain/llm/aws_bedrock.py
@@ -38,7 +38,8 @@ class AWSBedrockLlm(BaseLlm):
        }

        if config.stream:
-            from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
+            from langchain.callbacks.streaming_stdout import \
+                StreamingStdOutCallbackHandler

            callbacks = [StreamingStdOutCallbackHandler()]
            llm = Bedrock(**kwargs, streaming=config.stream, callbacks=callbacks)