chore: unify python examples

Author: ranst91

typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/examples/agents/__init__.py renamed to typescript-sdk/integrations/langgraph/examples/python/agents/__init__.py

@@ -2,7 +2,8 @@
 A simple agentic chat flow using LangGraph instead of CrewAI.
 """
 
-from typing import Dict, List, Any, Optional
+from typing import List, Any
+import os
 
 # Updated imports for LangGraph
 from langchain_core.runnables import RunnableConfig
@@ -76,5 +77,19 @@ async def chat_node(state: AgentState, config: RunnableConfig):
 workflow.add_edge(START, "chat_node")
 workflow.add_edge("chat_node", END)
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-agentic_chat_graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)

typescript-sdk/integrations/langgraph/examples/python/agents/agentic_chat/agent.py

@@ -2,9 +2,10 @@
 An example demonstrating agentic generative UI using LangGraph.
 """
 
-import json
 import asyncio
-from typing import Dict, List, Any, Optional, Literal
+from typing import List, Any
+import os
+
 # LangGraph imports
 from langchain_core.runnables import RunnableConfig
 from langgraph.graph import StateGraph, END, START
@@ -189,5 +190,19 @@ async def chat_node(state: AgentState, config: RunnableConfig):
 workflow.add_edge("start_flow", "chat_node")
 workflow.add_edge("chat_node", END)
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)

typescript-sdk/integrations/langgraph/examples/python/agents/agentic_generative_ui/agent.py

@@ -6,12 +6,12 @@
 load_dotenv()
 
 from ag_ui_langgraph import LangGraphAgent, add_langgraph_fastapi_endpoint
-from .human_in_the_loop import human_in_the_loop_graph
-from .predictive_state_updates import predictive_state_updates_graph
-from .shared_state import shared_state_graph
-from .tool_based_generative_ui import tool_based_generative_ui_graph
-from .agentic_chat import agentic_chat_graph
-from .agentic_generative_ui import graph
+from .human_in_the_loop.agent import graph as human_in_the_loop_graph
+from .predictive_state_updates.agent import graph as predictive_state_updates_graph
+from .shared_state.agent import graph as shared_state_graph
+from .tool_based_generative_ui.agent import graph as tool_based_generative_ui_graph
+from .agentic_chat.agent import graph as agentic_chat_graph
+from .agentic_generative_ui.agent import graph as agentic_generative_ui_graph
 
 app = FastAPI(title="LangGraph Dojo Example Server")
 
@@ -30,7 +30,7 @@
     "agentic_generative_ui": LangGraphAgent(
         name="agentic_generative_ui",
         description="An example for an agentic generative UI flow.",
-        graph=graph,
+        graph=agentic_generative_ui_graph,
     ),
     "human_in_the_loop": LangGraphAgent(
         name="human_in_the_loop",

typescript-sdk/integrations/langgraph/python/ag_ui_langgraph/examples/agents/dojo.py renamed to typescript-sdk/integrations/langgraph/examples/python/agents/dojo.py

@@ -4,6 +4,7 @@
 
 import json
 from typing import Dict, List, Any
+import os
 
 # LangGraph imports
 from langchain_core.runnables import RunnableConfig
@@ -269,5 +270,19 @@ def should_continue(command: Command):
     },
 )
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-human_in_the_loop_graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)

typescript-sdk/integrations/langgraph/examples/python/agents/human_in_the_loop/agent.py

@@ -2,9 +2,9 @@
 A demo of predictive state updates using LangGraph.
 """
 
-import json
 import uuid
-from typing import Dict, List, Any, Optional
+from typing import List, Any, Optional
+import os
 
 # LangGraph imports
 from langchain_core.runnables import RunnableConfig
@@ -174,5 +174,19 @@ async def chat_node(state: AgentState, config: RunnableConfig):
 workflow.add_edge("start_flow", "chat_node")
 workflow.add_edge("chat_node", END)
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-predictive_state_updates_graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)

typescript-sdk/integrations/langgraph/examples/python/agents/predictive_state_updates/agent.py

@@ -5,6 +5,7 @@
 import json
 from enum import Enum
 from typing import Dict, List, Any, Optional
+import os
 
 # LangGraph imports
 from langchain_core.runnables import RunnableConfig
@@ -297,5 +298,19 @@ async def chat_node(state: Dict[str, Any], config: RunnableConfig):
 workflow.add_edge("start_flow", "chat_node")
 workflow.add_edge("chat_node", END)
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-shared_state_graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)

typescript-sdk/integrations/langgraph/examples/python/agents/shared_state/agent.py

@@ -2,7 +2,8 @@
 An example demonstrating tool-based generative UI using LangGraph.
 """
 
-from typing import Dict, List, Any, Optional
+from typing import List, Any
+import os
 
 # LangGraph imports
 from langchain_core.runnables import RunnableConfig
@@ -124,6 +125,20 @@ async def chat_node(state: AgentState, config: RunnableConfig):
 workflow.add_edge(START, "chat_node")
 workflow.add_edge("chat_node", END)
 
+# Conditionally use a checkpointer based on the environment
+# Check for multiple indicators that we're running in LangGraph dev/API mode
+is_langgraph_api = (
+        os.environ.get("LANGGRAPH_API", "false").lower() == "true" or
+        os.environ.get("LANGGRAPH_API_DIR") is not None
+)
+
 # Compile the graph
-tool_based_generative_ui_graph = workflow.compile()
+if is_langgraph_api:
+    # When running in LangGraph API/dev, don't use a custom checkpointer
+    graph = workflow.compile()
+else:
+    # For CopilotKit and other contexts, use MemorySaver
+    from langgraph.checkpoint.memory import MemorySaver
+    memory = MemorySaver()
+    graph = workflow.compile(checkpointer=memory)
 

typescript-sdk/integrations/langgraph/examples/python/agents/tool_based_generative_ui/agent.py

@@ -3,12 +3,12 @@
   "dockerfile_lines": [],
   "dependencies": ["."],
   "graphs": {
-    "agentic_chat": "./agents/agentic_chat/agent.py:agentic_chat_graph",
+    "agentic_chat": "./agents/agentic_chat/agent.py:graph",
     "agentic_generative_ui": "./agents/agentic_generative_ui/agent.py:graph",
-    "human_in_the_loop": "./agents/human_in_the_loop/agent.py:human_in_the_loop_graph",
-    "predictive_state_updates": "./agents/predictive_state_updates/agent.py:predictive_state_updates_graph",
-    "shared_state": "./agents/shared_state/agent.py:shared_state_graph",
-    "tool_based_generative_ui": "./agents/tool_based_generative_ui/agent.py:tool_based_generative_ui_graph"
+    "human_in_the_loop": "./agents/human_in_the_loop/agent.py:graph",
+    "predictive_state_updates": "./agents/predictive_state_updates/agent.py:graph",
+    "shared_state": "./agents/shared_state/agent.py:graph",
+    "tool_based_generative_ui": "./agents/tool_based_generative_ui/agent.py:graph"
   },
   "env": ".env"
 }