Remove unwanted libraries and lighten package (#1391)

This commit is contained in:
Dev Khant
2024-07-09 04:30:16 +05:30
committed by GitHub
parent ebbf90f4aa
commit 1a5d0d236a
28 changed files with 235 additions and 1541 deletions

View File

@@ -23,7 +23,7 @@ class AWSBedrockLlm(BaseLlm):
except ModuleNotFoundError:
raise ModuleNotFoundError(
"The required dependencies for AWSBedrock are not installed."
'Please install with `pip install --upgrade "embedchain[aws-bedrock]"`'
"Please install with `pip install boto3==1.34.20`"
) from None
self.boto_client = boto3.client("bedrock-runtime", "us-west-2" or os.environ.get("AWS_REGION"))
@@ -38,8 +38,7 @@ 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)