Merge pull request #258 from scaleapi/dm/checkpointer

feat: HTTP-proxy LangGraph checkpointer

Author: danielmillerp

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
   ".": "0.9.2"
-}
+}

pyproject.toml

@@ -48,6 +48,7 @@ dependencies = [
     "yaspin>=3.1.0",
     "claude-agent-sdk>=0.1.0",
     "anthropic>=0.40.0",
+    "langgraph-checkpoint>=2.0.0",
 ]
 
 requires-python = ">= 3.12,<4"

src/agentex/lib/adk/__init__.py

@@ -5,6 +5,10 @@
 from agentex.lib.adk._modules.acp import ACPModule
 from agentex.lib.adk._modules.agents import AgentsModule
 from agentex.lib.adk._modules.agent_task_tracker import AgentTaskTrackerModule
+from agentex.lib.adk._modules.checkpointer import create_checkpointer
+from agentex.lib.adk._modules._langgraph_tracing import create_langgraph_tracing_handler
+from agentex.lib.adk._modules._langgraph_async import stream_langgraph_events
+from agentex.lib.adk._modules._langgraph_sync import convert_langgraph_to_agentex_events
 from agentex.lib.adk._modules.events import EventsModule
 from agentex.lib.adk._modules.messages import MessagesModule
 from agentex.lib.adk._modules.state import StateModule
@@ -27,16 +31,22 @@
 
 __all__ = [
     # Core
-    "acp", 
+    "acp",
     "agents",
-    "tasks", 
-    "messages", 
-    "state", 
-    "streaming", 
-    "tracing", 
+    "tasks",
+    "messages",
+    "state",
+    "streaming",
+    "tracing",
     "events",
     "agent_task_tracker",
 
+    # Checkpointing / LangGraph
+    "create_checkpointer",
+    "create_langgraph_tracing_handler",
+    "stream_langgraph_events",
+    "convert_langgraph_to_agentex_events",
+
     # Providers
     "providers",
     # Utils