diff --git a/01-tutorials/01-fundamentals/07-memory-persistent-agents/personal_agent_with_memory.ipynb b/01-tutorials/01-fundamentals/07-memory-persistent-agents/personal_agent_with_memory.ipynb index ccf94f2a0..2d011476f 100644 --- a/01-tutorials/01-fundamentals/07-memory-persistent-agents/personal_agent_with_memory.ipynb +++ b/01-tutorials/01-fundamentals/07-memory-persistent-agents/personal_agent_with_memory.ipynb @@ -76,20 +76,24 @@ "metadata": {}, "outputs": [], "source": [ - "\n", "# Import Required Libraries\n", "import os\n", "from strands import Agent, tool\n", "from strands.models import bedrock\n", "from strands_tools import mem0_memory\n", + "from strands_tools.mem0_memory import Mem0ServiceClient\n", "import time\n", "\n", "from ddgs import DDGS\n", "from ddgs.exceptions import DDGSException, RatelimitException\n", "\n", + "DEFAULT_BEDROCK_MODEL_ID = \"us.anthropic.claude-3-7-sonnet-20250219-v1:0\"\n", "\n", "os.environ[\"AWS_REGION\"] = os.environ.get(\"AWS_REGION\", \"us-east-1\")\n", - "bedrock.DEFAULT_BEDROCK_MODEL_ID = \"us.anthropic.claude-3-7-sonnet-20250219-v1:0\" #Optional: Set a default model for Bedrock" + "bedrock.DEFAULT_BEDROCK_MODEL_ID = DEFAULT_BEDROCK_MODEL_ID # Optional: Set a default model for Bedrock\n", + "\n", + "# Required as Mem0 defaults to Claude 3.5 Haiku which now has limited support\n", + "Mem0ServiceClient.DEFAULT_CONFIG[\"llm\"][\"config\"][\"model\"] = DEFAULT_BEDROCK_MODEL_ID " ] }, {