Fix CI tests for Mem0 (#1498)

This commit is contained in:
Dev Khant
2024-07-19 01:58:19 +05:30
committed by GitHub
parent a546a9f56a
commit e9136c1aa0
4 changed files with 78 additions and 8 deletions

View File

@@ -14,7 +14,11 @@ class AWSBedrockLLM(LLMBase):
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.model_kwargs = {"temperature": self.config.temperature, "max_tokens_to_sample": self.config.max_tokens, "top_p": self.config.top_p}
self.model_kwargs = {
"temperature": self.config.temperature,
"max_tokens_to_sample": self.config.max_tokens,
"top_p": self.config.top_p
}
def _format_messages(self, messages: List[Dict[str, str]]) -> str:
"""