langchain-dev-utils 1.3.6__tar.gz → 1.4.0__tar.gz

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.3.6
+Version: 1.4.0
 Summary: A practical utility library for LangChain and LangGraph development
 Project-URL: Source Code, https://github.com/TBice123123/langchain-dev-utils
 Project-URL: repository, https://github.com/TBice123123/langchain-dev-utils
@@ -27,10 +27,11 @@ Description-Content-Type: text/markdown
   <a href="https://tbice123123.github.io/langchain-dev-utils/zh/">中文</a>
 </p>
 
-[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package)](https://pypi.org/project/langchain-dev-utils/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![Python](https://img.shields.io/badge/python-3.11|3.12|3.13|3.14-%2334D058)](https://www.python.org/downloads)
-[![Downloads](https://static.pepy.tech/badge/langchain-dev-utils/month)](https://pepy.tech/project/langchain-dev-utils)
+[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-black.svg?logo=github)](https://github.com/TBice123123/langchain-dev-utils)
+[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package&logo=python)](https://pypi.org/project/langchain-dev-utils/)
+[![Python Version](https://img.shields.io/badge/python-3.11%2B-blue.svg?logo=python&label=Python)](https://python.org)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg?label=License)](https://opensource.org/licenses/MIT)
+[![Last Commit](https://img.shields.io/github/last-commit/TBice123123/langchain-dev-utils)](https://github.com/TBice123123/langchain-dev-utils)
 [![Documentation](https://img.shields.io/badge/docs-latest-blue)](https://tbice123123.github.io/langchain-dev-utils)
 
 > This is the English version. For the Chinese version, please visit [中文版本](https://github.com/TBice123123/langchain-dev-utils/blob/master/README_cn.md)
@@ -51,7 +52,7 @@ Tired of writing repetitive code in LangChain development? `langchain-dev-utils`
 - **💬 Flexible message handling** - Support for chain-of-thought concatenation, streaming processing, and message formatting
 - **🛠️ Powerful tool calling** - Built-in tool call detection, parameter parsing, and human review functionality
 - **🤖 Efficient Agent development** - Simplify agent creation process, expand more common middleware
-- **📊 Flexible state graph composition** - Support for serial and parallel composition of multiple StateGraphs
+- **📊 Convenient State Graph Construction** - Provides pre-built functions to easily create sequential or parallel state graphs
 
 ## ⚡ Quick Start
 

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/README.md

@@ -9,10 +9,11 @@
   <a href="https://tbice123123.github.io/langchain-dev-utils/zh/">中文</a>
 </p>
 
-[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package)](https://pypi.org/project/langchain-dev-utils/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![Python](https://img.shields.io/badge/python-3.11|3.12|3.13|3.14-%2334D058)](https://www.python.org/downloads)
-[![Downloads](https://static.pepy.tech/badge/langchain-dev-utils/month)](https://pepy.tech/project/langchain-dev-utils)
+[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-black.svg?logo=github)](https://github.com/TBice123123/langchain-dev-utils)
+[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package&logo=python)](https://pypi.org/project/langchain-dev-utils/)
+[![Python Version](https://img.shields.io/badge/python-3.11%2B-blue.svg?logo=python&label=Python)](https://python.org)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg?label=License)](https://opensource.org/licenses/MIT)
+[![Last Commit](https://img.shields.io/github/last-commit/TBice123123/langchain-dev-utils)](https://github.com/TBice123123/langchain-dev-utils)
 [![Documentation](https://img.shields.io/badge/docs-latest-blue)](https://tbice123123.github.io/langchain-dev-utils)
 
 > This is the English version. For the Chinese version, please visit [中文版本](https://github.com/TBice123123/langchain-dev-utils/blob/master/README_cn.md)
@@ -33,7 +34,7 @@ Tired of writing repetitive code in LangChain development? `langchain-dev-utils`
 - **💬 Flexible message handling** - Support for chain-of-thought concatenation, streaming processing, and message formatting
 - **🛠️ Powerful tool calling** - Built-in tool call detection, parameter parsing, and human review functionality
 - **🤖 Efficient Agent development** - Simplify agent creation process, expand more common middleware
-- **📊 Flexible state graph composition** - Support for serial and parallel composition of multiple StateGraphs
+- **📊 Convenient State Graph Construction** - Provides pre-built functions to easily create sequential or parallel state graphs
 
 ## ⚡ Quick Start
 

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/README_cn.md

@@ -9,10 +9,11 @@
   <a href="https://tbice123123.github.io/langchain-dev-utils/zh/">中文</a>
 </p>
 
-[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package)](https://pypi.org/project/langchain-dev-utils/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![Python](https://img.shields.io/badge/python-3.11|3.12|3.13|3.14-%2334D058)](https://www.python.org/downloads)
-[![Downloads](https://static.pepy.tech/badge/langchain-dev-utils/month)](https://pepy.tech/project/langchain-dev-utils)
+[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-black.svg?logo=github)](https://github.com/TBice123123/langchain-dev-utils)
+[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package&logo=python)](https://pypi.org/project/langchain-dev-utils/)
+[![Python Version](https://img.shields.io/badge/python-3.11%2B-blue.svg?logo=python&label=Python)](https://python.org)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg?label=License)](https://opensource.org/licenses/MIT)
+[![Last Commit](https://img.shields.io/github/last-commit/TBice123123/langchain-dev-utils)](https://github.com/TBice123123/langchain-dev-utils)
 [![Documentation](https://img.shields.io/badge/docs-latest-blue)](https://tbice123123.github.io/langchain-dev-utils/zh/)
 
 > 当前为中文版，英文版请访问[English Version](https://github.com/TBice123123/langchain-dev-utils/blob/master/README.md)
@@ -33,7 +34,7 @@
 - **💬 灵活的消息处理** - 支持思维链拼接、流式处理和消息格式化
 - **🛠️ 强大的工具调用** - 内置工具调用检测、参数解析和人工审核功能
 - **🤖 高效的 Agent 开发** - 简化智能体创建流程，扩充更多的常用中间件
-- **📊 灵活的状态图组合** - 支持串行和并行方式组合多个 StateGraph
+- **📊 便捷的状态图构建** - 提供预构建函数方便构建顺序或者并行的状态图
 
 ## ⚡ 快速开始
 

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/pyproject.toml

@@ -1,47 +1,51 @@
-[project]
-name = "langchain-dev-utils"
-version = "1.3.6"
-description = "A practical utility library for LangChain and LangGraph development"
-readme = "README.md"
-authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]
-requires-python = ">=3.11"
-dependencies = ["langchain>=1.2.0", "langchain-core>=1.2.5", "langgraph>=1.0.0"]
-
-[project.urls]
-"Source Code" = "https://github.com/TBice123123/langchain-dev-utils"
-repository = "https://github.com/TBice123123/langchain-dev-utils"
-documentation = "https://tbice123123.github.io/langchain-dev-utils"
-
-
-[project.optional-dependencies]
-standard = ["json-repair>=0.53.1", "langchain-openai"]
-
-[build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
-
-[tool.hatch.build]
-exclude = ["/data", "/docs", "mkdocs.yml"]
-
-[tool.pytest.ini_options]
-asyncio_mode = "auto"
-testpaths = ["tests"]
-python_files = ["test_*.py"]
-python_functions = ["test_*"]
-
-[dependency-groups]
-dev = ["langchain-model-profiles>=0.0.5", "ruff>=0.14.5"]
-docs = ["mkdocs-material>=9.7.0", "mkdocs-static-i18n>=1.3.0"]
-tests = [
-    "python-dotenv>=1.1.1",
-    "langchain-tests>=1.0.0",
-    "langchain-deepseek>=1.0.0",
-    "langchain-qwq>=0.3.0",
-    "langchain-ollama>=1.0.0",
-    "langchain-community>=0.4.1",
-]
-
-
-[tool.ruff.lint]
-select = ["E", "F", "I", "PGH003", "T201"]
-ignore = ["E501"]
+[project]
+name = "langchain-dev-utils"
+version = "1.4.0"
+description = "A practical utility library for LangChain and LangGraph development"
+readme = "README.md"
+authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]
+requires-python = ">=3.11"
+dependencies = ["langchain>=1.2.0", "langchain-core>=1.2.5", "langgraph>=1.0.0"]
+
+[project.urls]
+"Source Code" = "https://github.com/TBice123123/langchain-dev-utils"
+repository = "https://github.com/TBice123123/langchain-dev-utils"
+documentation = "https://tbice123123.github.io/langchain-dev-utils"
+
+
+[project.optional-dependencies]
+standard = ["json-repair>=0.53.1", "langchain-openai"]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build]
+exclude = ["/data", "/docs", "mkdocs.yml"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]
+python_files = ["test_*.py"]
+python_functions = ["test_*"]
+
+[dependency-groups]
+dev = ["langchain-model-profiles>=0.0.5", "ruff>=0.14.5"]
+docs = [
+    "jupyter>=1.1.1",
+    "mkdocs-material>=9.7.0",
+    "mkdocs-static-i18n>=1.3.0",
+]
+tests = [
+    "python-dotenv>=1.1.1",
+    "langchain-tests>=1.0.0",
+    "langchain-deepseek>=1.0.0",
+    "langchain-qwq>=0.3.0",
+    "langchain-ollama>=1.0.0",
+    "langchain-community>=0.4.1",
+]
+
+
+[tool.ruff.lint]
+select = ["E", "F", "I", "PGH003", "T201"]
+ignore = ["E501"]

langchain_dev_utils-1.4.0/src/langchain_dev_utils/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.4.0"

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/_utils.py

@@ -1,9 +1,27 @@
 from importlib import util
-from typing import Literal, Optional
+from typing import Literal, Optional, cast
 
+from langgraph.graph import StateGraph
+from langgraph.graph.state import StateNode
 from pydantic import BaseModel
 
 
+def _transform_node_to_tuple(
+    node: StateNode | tuple[str, StateNode],
+) -> tuple[str, StateNode]:
+    if not isinstance(node, tuple):
+        if isinstance(node, StateGraph):
+            node = node.compile()
+            name = node.name
+            return name, node
+        name = cast(str, getattr(node, "name", getattr(node, "__name__", None)))
+        if name is None:
+            raise ValueError("Node name must be provided if action is not a function")
+        return name, node
+    else:
+        return node
+
+
 def _check_pkg_install(
     pkg: Literal["langchain_openai", "json_repair"],
 ) -> None:

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/agents/middleware/handoffs.py

@@ -1,4 +1,4 @@
-from typing import Any, Awaitable, Callable, Literal
+from typing import Any, Awaitable, Callable, Literal, cast
 
 from langchain.agents import AgentState
 from langchain.agents.middleware import AgentMiddleware, ModelRequest, ModelResponse
@@ -128,6 +128,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
     Args:
         agents_config (dict[str, AgentConfig]): A dictionary of agent configurations.
         custom_handoffs_tool_descriptions (Optional[dict[str, str]]): A dictionary of custom tool descriptions for handoffs tools. Defaults to None.
+        handoffs_tool_overrides (Optional[dict[str, BaseTool]]): A dictionary of handoffs tools to override. Defaults to None.
 
     Examples:
         ```python
@@ -142,6 +143,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
         self,
         agents_config: dict[str, AgentConfig],
         custom_handoffs_tool_descriptions: Optional[dict[str, str]] = None,
+        handoffs_tool_overrides: Optional[dict[str, BaseTool]] = None,
     ) -> None:
         default_agent_name = _get_default_active_agent(agents_config)
         if default_agent_name is None:
@@ -152,13 +154,23 @@ class HandoffAgentMiddleware(AgentMiddleware):
         if custom_handoffs_tool_descriptions is None:
             custom_handoffs_tool_descriptions = {}
 
-        handoffs_tools = [
-            _create_handoffs_tool(
-                agent_name,
-                custom_handoffs_tool_descriptions.get(agent_name),
-            )
-            for agent_name in agents_config.keys()
-        ]
+        if handoffs_tool_overrides is None:
+            handoffs_tool_overrides = {}
+
+        handoffs_tools = []
+        for agent_name in agents_config.keys():
+            if not handoffs_tool_overrides.get(agent_name):
+                handoffs_tools.append(
+                    _create_handoffs_tool(
+                        agent_name,
+                        custom_handoffs_tool_descriptions.get(agent_name),
+                    )
+                )
+            else:
+                handoffs_tools.append(
+                    cast(BaseTool, handoffs_tool_overrides.get(agent_name))
+                )
+
         self.default_agent_name = default_agent_name
         self.agents_config = _transform_agent_config(
             agents_config,
@@ -166,7 +178,7 @@ class HandoffAgentMiddleware(AgentMiddleware):
         )
         self.tools = handoffs_tools
 
-    def _get_active_agent_config(self, request: ModelRequest) -> dict[str, Any]:
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
         active_agent_name = request.state.get("active_agent", self.default_agent_name)
 
         _config = self.agents_config[active_agent_name]
@@ -181,24 +193,22 @@ class HandoffAgentMiddleware(AgentMiddleware):
             params["system_prompt"] = _config.get("prompt")
         if _config.get("tools"):
             params["tools"] = _config.get("tools")
-        return params
+
+        if params:
+            return request.override(**params)
+        else:
+            return request
 
     def wrap_model_call(
         self, request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
     ) -> ModelCallResult:
-        override_kwargs = self._get_active_agent_config(request)
-        if override_kwargs:
-            return handler(request.override(**override_kwargs))
-        else:
-            return handler(request)
+        override_request = self._get_override_request(request)
+        return handler(override_request)
 
     async def awrap_model_call(
         self,
         request: ModelRequest,
         handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelCallResult:
-        override_kwargs = self._get_active_agent_config(request)
-        if override_kwargs:
-            return await handler(request.override(**override_kwargs))
-        else:
-            return await handler(request)
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/agents/middleware/model_router.py

@@ -150,7 +150,7 @@ class ModelRouterMiddleware(AgentMiddleware):
         model_name = await self._aselect_model(state["messages"])
         return {"router_model_selection": model_name}
 
-    def _get_override_kwargs(self, request: ModelRequest) -> dict[str, Any]:
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
         model_dict = {
             item["model_name"]: {
                 "tools": item.get("tools", None),
@@ -180,24 +180,21 @@ class ModelRouterMiddleware(AgentMiddleware):
                     content=model_values["system_prompt"]
                 )
 
-        return override_kwargs
+        if override_kwargs:
+            return request.override(**override_kwargs)
+        else:
+            return request
 
     def wrap_model_call(
         self, request: ModelRequest, handler: Callable[[ModelRequest], ModelResponse]
     ) -> ModelCallResult:
-        override_kwargs = self._get_override_kwargs(request)
-        if override_kwargs:
-            return handler(request.override(**override_kwargs))
-        else:
-            return handler(request)
+        override_request = self._get_override_request(request)
+        return handler(override_request)
 
     async def awrap_model_call(
         self,
         request: ModelRequest,
         handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelCallResult:
-        override_kwargs = self._get_override_kwargs(request)
-        if override_kwargs:
-            return await handler(request.override(**override_kwargs))
-        else:
-            return await handler(request)
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/agents/middleware/plan.py

@@ -335,12 +335,8 @@ class PlanMiddleware(AgentMiddleware):
         self.system_prompt = system_prompt
         self.tools = tools
 
-    def wrap_model_call(
-        self,
-        request: ModelRequest,
-        handler: Callable[[ModelRequest], ModelResponse],
-    ) -> ModelCallResult:
-        """Update the system message to include the plan system prompt."""
+    def _get_override_request(self, request: ModelRequest) -> ModelRequest:
+        """Add the plan system prompt to the system message."""
         if request.system_message is not None:
             new_system_content = [
                 *request.system_message.content_blocks,
@@ -351,7 +347,15 @@ class PlanMiddleware(AgentMiddleware):
         new_system_message = SystemMessage(
             content=cast("list[str | dict[str, str]]", new_system_content)
         )
-        return handler(request.override(system_message=new_system_message))
+        return request.override(system_message=new_system_message)
+
+    def wrap_model_call(
+        self,
+        request: ModelRequest,
+        handler: Callable[[ModelRequest], ModelResponse],
+    ) -> ModelCallResult:
+        override_request = self._get_override_request(request)
+        return handler(override_request)
 
     async def awrap_model_call(
         self,
@@ -359,14 +363,5 @@ class PlanMiddleware(AgentMiddleware):
         handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelCallResult:
         """Update the system message to include the plan system prompt."""
-        if request.system_message is not None:
-            new_system_content = [
-                *request.system_message.content_blocks,
-                {"type": "text", "text": f"\n\n{self.system_prompt}"},
-            ]
-        else:
-            new_system_content = [{"type": "text", "text": self.system_prompt}]
-        new_system_message = SystemMessage(
-            content=cast("list[str | dict[str, str]]", new_system_content)
-        )
-        return await handler(request.override(system_message=new_system_message))
+        override_request = self._get_override_request(request)
+        return await handler(override_request)

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/agents/wrap.py

@@ -1,12 +1,13 @@
-import asyncio
-from typing import Any, Awaitable, Callable, Optional
+import inspect
+from typing import Any, Awaitable, Callable, Optional, cast
 
 from langchain.tools import ToolRuntime
-from langchain_core.messages import HumanMessage
+from langchain_core.messages import AIMessage, HumanMessage
 from langchain_core.tools import BaseTool, StructuredTool
 from langgraph.graph.state import CompiledStateGraph
 
 from langchain_dev_utils.message_convert import format_sequence
+from langchain_dev_utils.tool_calling import parse_tool_calling
 
 
 def _process_input(request: str, runtime: ToolRuntime) -> str:
@@ -19,6 +20,18 @@ def _process_output(
     return response["messages"][-1].content
 
 
+def get_subagent_name(runtime: ToolRuntime) -> str:
+    messages = runtime.state.get("messages", [])
+    last_ai_msg = cast(
+        AIMessage,
+        next((msg for msg in reversed(messages) if isinstance(msg, AIMessage)), None),
+    )
+
+    _, args = parse_tool_calling(last_ai_msg, first_tool_call_only=True)
+    args = cast(dict[str, Any], args)
+    return args["agent_name"]
+
+
 def wrap_agent_as_tool(
     agent: CompiledStateGraph,
     tool_name: Optional[str] = None,
@@ -115,7 +128,7 @@ def wrap_agent_as_tool(
         request: str,
         runtime: ToolRuntime,
     ):
-        if asyncio.iscoroutinefunction(process_input_async):
+        if inspect.iscoroutinefunction(process_input_async):
             _processed_input = await process_input_async(request, runtime)
         else:
             _processed_input = (
@@ -135,7 +148,7 @@ def wrap_agent_as_tool(
 
         response = await agent.ainvoke(agent_input)
 
-        if asyncio.iscoroutinefunction(process_output_async):
+        if inspect.iscoroutinefunction(process_output_async):
             response = await process_output_async(request, response, runtime)
         else:
             response = (
@@ -277,7 +290,7 @@ def wrap_all_agents_as_tool(
         if agent_name not in agents_map:
             raise ValueError(f"Agent {agent_name} not found")
 
-        if asyncio.iscoroutinefunction(process_input_async):
+        if inspect.iscoroutinefunction(process_input_async):
             _processed_input = await process_input_async(description, runtime)
         else:
             _processed_input = (
@@ -297,7 +310,7 @@ def wrap_all_agents_as_tool(
 
         response = await agents_map[agent_name].ainvoke(agent_input)
 
-        if asyncio.iscoroutinefunction(process_output_async):
+        if inspect.iscoroutinefunction(process_output_async):
             response = await process_output_async(description, response, runtime)
         else:
             response = (

{langchain_dev_utils-1.3.6 → langchain_dev_utils-1.4.0}/src/langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -218,14 +218,15 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
         stop: list[str] | None = None,
         **kwargs: Any,
     ) -> dict:
+        if stop is not None:
+            kwargs["stop"] = stop
+
         payload = {**self._default_params, **kwargs}
 
         if self._use_responses_api(payload):
             return super()._get_request_payload(input_, stop=stop, **kwargs)
 
         messages = self._convert_input(input_).to_messages()
-        if stop is not None:
-            kwargs["stop"] = stop
 
         payload_messages = []
         last_human_index = -1

langchain_dev_utils-1.4.0/src/langchain_dev_utils/graph/__init__.py

@@ -0,0 +1,7 @@
+from .parallel import create_parallel_graph
+from .sequential import create_sequential_graph
+
+__all__ = [
+    "create_parallel_graph",
+    "create_sequential_graph",
+]

langchain_dev_utils-1.4.0/src/langchain_dev_utils/graph/parallel.py

@@ -0,0 +1,119 @@
+from typing import Awaitable, Callable, Optional, Union, cast
+
+from langgraph.cache.base import BaseCache
+from langgraph.graph import StateGraph
+from langgraph.graph.state import CompiledStateGraph, StateNode
+from langgraph.store.base import BaseStore
+from langgraph.types import Checkpointer, Send
+from langgraph.typing import ContextT, InputT, OutputT, StateT
+
+from langchain_dev_utils._utils import _transform_node_to_tuple
+
+from .types import Node
+
+
+def create_parallel_graph(
+    nodes: list[Node],
+    state_schema: type[StateT],
+    graph_name: Optional[str] = None,
+    branches_fn: Optional[
+        Union[
+            Callable[..., list[Send]],
+            Callable[..., Awaitable[list[Send]]],
+        ]
+    ] = None,
+    context_schema: type[ContextT] | None = None,
+    input_schema: type[InputT] | None = None,
+    output_schema: type[OutputT] | None = None,
+    checkpointer: Checkpointer | None = None,
+    store: BaseStore | None = None,
+    cache: BaseCache | None = None,
+) -> CompiledStateGraph[StateT, ContextT, InputT, OutputT]:
+    """
+    Create a parallel graph from a list of nodes.
+
+    This function lets you build a parallel StateGraph simply by writing the corresponding Nodes.
+
+    Args:
+        nodes: List of nodes to execute in parallel
+        state_schema: state schema of the final state graph
+        graph_name: Name of the final state graph
+        branches_fn: Optional function to determine which nodes to execute
+            in parallel
+        context_schema: context schema of the final state graph
+        input_schema: input schema of the final state graph
+        output_schema: output schema of the final state graph
+        checkpointer: Optional LangGraph checkpointer for the final state graph
+        store: Optional LangGraph store for the final state graph
+        cache: Optional LangGraph cache for the final state graph
+
+    Returns:
+        CompiledStateGraph[StateT, ContextT, InputT, OutputT]: Compiled state graph
+
+    Example:
+        # Basic parallel pipeline: multiple specialized agents run concurrently
+        >>> from langchain_dev_utils.graph import create_parallel_graph
+        >>>
+        >>> graph = create_parallel_graph(
+        ...     nodes=[
+        ...        node1, node2, node3
+        ...     ],
+        ...     state_schema=StateT,
+        ...     graph_name="parallel_graph",
+        ... )
+        >>>
+        >>> response = graph.invoke({"messages": [HumanMessage("Hello")]})
+
+        # Dynamic parallel pipeline: decide which nodes to run based on conditional branches
+        >>> graph = create_parallel_graph(
+        ...     nodes=[
+        ...         node1, node2, node3
+        ...     ],
+        ...     state_schema=StateT,
+        ...     branches_fn=lambda state: [
+        ...         Send("node1", arg={"messages": [HumanMessage("Hello")]}),
+        ...         Send("node2", arg={"messages": [HumanMessage("Hello")]}),
+        ...     ],
+        ...     graph_name="parallel_graph",
+        ... )
+        >>>
+        >>> response = graph.invoke({"messages": [HumanMessage("Hello")]})
+    """
+    graph = StateGraph(
+        state_schema=state_schema,
+        context_schema=context_schema,
+        input_schema=input_schema,
+        output_schema=output_schema,
+    )
+
+    node_list: list[tuple[str, StateNode]] = []
+
+    for node in nodes:
+        node_list.append(_transform_node_to_tuple(node))
+
+    if branches_fn:
+        for name, node in node_list:
+            node = cast(StateNode[StateT, ContextT], node)
+            graph.add_node(name, node)
+        graph.add_conditional_edges(
+            "__start__",
+            branches_fn,
+            [node_name for node_name, _ in node_list],
+        )
+        return graph.compile(
+            name=graph_name or "parallel graph",
+            checkpointer=checkpointer,
+            store=store,
+            cache=cache,
+        )
+    else:
+        for node_name, node in node_list:
+            node = cast(StateNode[StateT, ContextT], node)
+            graph.add_node(node_name, node)
+            graph.add_edge("__start__", node_name)
+        return graph.compile(
+            name=graph_name or "parallel graph",
+            checkpointer=checkpointer,
+            store=store,
+            cache=cache,
+        )