diff --git a/embedchain/docs/components/llms.mdx b/embedchain/docs/components/llms.mdx index c0034dc1..183b8cd3 100644 --- a/embedchain/docs/components/llms.mdx +++ b/embedchain/docs/components/llms.mdx @@ -708,8 +708,6 @@ embedder: import os from embedchain import App -os.environ["AWS_ACCESS_KEY_ID"] = "xxx" -os.environ["AWS_SECRET_ACCESS_KEY"] = "xxx" os.environ["AWS_REGION"] = "us-west-2" app = App.from_config(config_path="config.yaml") diff --git a/mem0/llms/aws_bedrock.py b/mem0/llms/aws_bedrock.py index d56ff4a4..9f6c1b09 100644 --- a/mem0/llms/aws_bedrock.py +++ b/mem0/llms/aws_bedrock.py @@ -1,5 +1,5 @@ + import json -import os from typing import Any, Dict, List, Optional try: @@ -11,18 +11,13 @@ from mem0.configs.llms.base import BaseLlmConfig from mem0.llms.base import LLMBase -class AWSBedrockLLM(LLMBase): +class AWSBedrockLLM(LLMBase): def __init__(self, config: Optional[BaseLlmConfig] = None): super().__init__(config) if not self.config.model: - self.config.model = "anthropic.claude-3-5-sonnet-20240620-v1:0" - self.client = boto3.client( - "bedrock-runtime", - region_name=os.environ.get("AWS_REGION"), - aws_access_key_id=os.environ.get("AWS_ACCESS_KEY"), - aws_secret_access_key=os.environ.get("AWS_SECRET_ACCESS_KEY"), - ) + self.config.model="anthropic.claude-3-5-sonnet-20240620-v1:0" + self.client = boto3.client("bedrock-runtime") self.model_kwargs = { "temperature": self.config.temperature, "max_tokens_to_sample": self.config.max_tokens,