Simplify the boilerplate

Signed-off-by: William Fu-Hinthorn <[email protected]>

Author: hinthornw

.gitignore

@@ -161,3 +161,4 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 uv.lock
+.langgraph_api/

pyproject.toml

@@ -54,3 +54,9 @@ lint.ignore = [
 "tests/*" = ["D", "UP"]
 [tool.ruff.lint.pydocstyle]
 convention = "google"
+
+[dependency-groups]
+dev = [
+    "anyio>=4.7.0",
+    "langgraph-cli[inmem]>=0.2.8",
+]

src/agent/configuration.py

@@ -1,38 +1,52 @@
-"""Define a simple chatbot agent.
+"""LangGraph single-node graph template.
 
-This agent returns a predefined response without using an actual LLM.
+Returns a predefined response. Replace logic and configuration as needed.
 """
 
-from typing import Any, Dict
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict, TypedDict
 
 from langchain_core.runnables import RunnableConfig
 from langgraph.graph import StateGraph
 
-from agent.configuration import Configuration
-from agent.state import State
+
+class Configuration(TypedDict):
+    """Configurable parameters for the agent.
+
+    Set these when creating assistants OR when invoking the graph.
+    See: https://langchain-ai.github.io/langgraph/cloud/how-tos/configuration_cloud/
+    """
+    my_configurable_param: str
+
+
+@dataclass
+class State:
+    """Input state for the agent.
+
+    Defines the initial structure of incoming data.
+    See: https://langchain-ai.github.io/langgraph/concepts/low_level/#state
+    """
+    changeme: str = "example"
 
 
 async def my_node(state: State, config: RunnableConfig) -> Dict[str, Any]:
-    """Each node does work."""
-    configuration = Configuration.from_runnable_config(config)
-    # configuration = Configuration.from_runnable_config(config)
-    # You can use runtime configuration to alter the behavior of your
-    # graph.
+    """Example node: processes input and returns output.
+
+    Can use runtime configuration to alter behavior.
+    """
+    configuration = config["configurable"]
     return {
         "changeme": "output from my_node. "
-        f"Configured with {configuration.my_configurable_param}"
+        f'Configured with {configuration.get("my_configurable_param")}'
     }
 
 
-# Define a new graph
-workflow = StateGraph(State, config_schema=Configuration)
-
-# Add the node to the graph
-workflow.add_node("my_node", my_node)
-
-# Set the entrypoint as `call_model`
-workflow.add_edge("__start__", "my_node")
-
-# Compile the workflow into an executable graph
-graph = workflow.compile()
-graph.name = "New Graph"  # This defines the custom name in LangSmith
+# Define the graph
+graph = (
+    StateGraph(State, config_schema=Configuration)
+    .add_node(my_node)
+    .add_edge("__start__", "my_node")
+    .compile(name="New Graph")
+)

src/agent/graph.py

@@ -1,11 +1,9 @@
 import pytest
-from langsmith import unit
 
 from agent import graph
 
 
-@pytest.mark.asyncio
-@unit
+@pytest.mark.langsmith
 async def test_agent_simple_passthrough() -> None:
     res = await graph.ainvoke({"changeme": "some_val"})
     assert res is not None

src/agent/state.py

@@ -1,5 +1,9 @@
-from agent.configuration import Configuration
+from langgraph.pregel import Pregel
 
+from agent.graph import graph
 
-def test_configuration_empty() -> None:
-    Configuration.from_runnable_config({})
+
+def test_placeholder() -> None:
+    # TODO: You can add actual unit tests
+    # for your graph and other logic here.
+    assert isinstance(graph, Pregel)