Removing langmem style tools from this PR

Author: Jack Gordley

libs/langgraph-checkpoint-aws/langgraph_checkpoint_aws/__init__.py

@@ -8,11 +8,6 @@
 from langgraph_checkpoint_aws.agentcore.store import (
     AgentCoreMemoryStore,
 )
-from langgraph_checkpoint_aws.agentcore.tools import (
-    create_manage_memory_tool,
-    create_search_memory_tool,
-    create_store_event_tool,
-)
 
 __version__ = "0.1.2"
 SDK_USER_AGENT = f"LangGraphCheckpointAWS#{__version__}"
@@ -21,8 +16,5 @@
 __all__ = [
     "AgentCoreMemorySaver",
     "AgentCoreMemoryStore",
-    "create_manage_memory_tool",
-    "create_search_memory_tool",
-    "create_store_event_tool",
     "SDK_USER_AGENT",
 ]

libs/langgraph-checkpoint-aws/langgraph_checkpoint_aws/agentcore/__init__.py

@@ -1,15 +1,7 @@
 from langgraph_checkpoint_aws.agentcore.saver import AgentCoreMemorySaver
 from langgraph_checkpoint_aws.agentcore.store import AgentCoreMemoryStore
-from langgraph_checkpoint_aws.agentcore.tools import (
-    create_manage_memory_tool,
-    create_search_memory_tool,
-    create_store_event_tool,
-)
 
 __all__ = [
     "AgentCoreMemorySaver",
     "AgentCoreMemoryStore",
-    "create_manage_memory_tool",
-    "create_search_memory_tool",
-    "create_store_event_tool",
 ]

libs/langgraph-checkpoint-aws/langgraph_checkpoint_aws/agentcore/helpers.py

@@ -12,7 +12,6 @@
 from typing import Any, Dict, List, Union
 
 import boto3
-from bedrock_agentcore.memory.constants import MessageRole
 from botocore.config import Config
 from langchain_core.messages import BaseMessage
 from langgraph.checkpoint.base import CheckpointTuple, SerializerProtocol
@@ -347,13 +346,13 @@ def convert_langchain_messages_to_event_messages(
 
         # Map LangChain roles to Bedrock Agent Core roles
         if msg.type == "human":
-            role = MessageRole.USER.value
+            role = "USER"
         elif msg.type == "ai":
-            role = MessageRole.ASSISTANT.value
+            role = "ASSISTANT"
         elif msg.type == "tool":
-            role = MessageRole.TOOL.value
+            role = "TOOL"
         elif msg.type == "system":
-            role = MessageRole.OTHER.value
+            role = "OTHER"
         else:
             logger.warning(f"Skipping unsupported message type: {msg.type}")
             continue