diff --git a/mem0/llms/aws_bedrock.py b/mem0/llms/aws_bedrock.py index 2bc963c2b2..d56ff4a43b 100644 --- a/mem0/llms/aws_bedrock.py +++ b/mem0/llms/aws_bedrock.py @@ -172,6 +172,7 @@ def _convert_tool_format(self, original_tools): def generate_response( self, messages: List[Dict[str, str]], + response_format=None, tools: Optional[List[Dict]] = None, tool_choice: str = "auto", ):