From df0eef4713414023d3bf9e3a7d1ae61dc86ccdba Mon Sep 17 00:00:00 2001
From: "open-swe[bot]" <open-swe@users.noreply.github.com>
Date: Tue, 29 Jul 2025 17:27:05 +0000
Subject: [PATCH] Apply patch

---
 .../langgraph/_internal/_runnable.py          | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/libs/langgraph/langgraph/_internal/_runnable.py b/libs/langgraph/langgraph/_internal/_runnable.py
index c6c5ed239..2622d92b2 100644
--- a/libs/langgraph/langgraph/_internal/_runnable.py
+++ b/libs/langgraph/langgraph/_internal/_runnable.py
@@ -487,6 +487,52 @@ def is_async_generator(
     )
 
 
+class _PregelWrapper(Runnable):
+    """Wrapper for PregelProtocol instances to handle runtime context propagation.
+    
+    When a compiled subgraph (PregelProtocol) is added as a node, this wrapper
+    extracts the runtime context from the config and passes it explicitly to
+    the subgraph's invoke method.
+    """
+    
+    def __init__(self, pregel: "PregelProtocol", name: str | None = None):
+        self.pregel = pregel
+        self._name = name or getattr(pregel, "name", None) or pregel.__class__.__name__
+    
+    def get_name(self, suffix: str | None = None, *, name: str | None = None) -> str:
+        """Get the name of the runnable."""
+        name = name or self._name
+        return f"{name}{suffix}" if suffix else name
+    
+    def invoke(
+        self, input: Any, config: RunnableConfig | None = None, **kwargs: Any
+    ) -> Any:
+        """Invoke the wrapped PregelProtocol with runtime context extracted from config."""
+        if config is None:
+            config = ensure_config()
+        
+        # Extract runtime context from config
+        runtime = config.get(CONF, {}).get(CONFIG_KEY_RUNTIME)
+        context = runtime.context if runtime else None
+        
+        # Invoke the subgraph with the extracted context
+        return self.pregel.invoke(input, config, context=context, **kwargs)
+    
+    async def ainvoke(
+        self, input: Any, config: RunnableConfig | None = None, **kwargs: Any
+    ) -> Any:
+        """Async invoke the wrapped PregelProtocol with runtime context extracted from config."""
+        if config is None:
+            config = ensure_config()
+        
+        # Extract runtime context from config
+        runtime = config.get(CONF, {}).get(CONFIG_KEY_RUNTIME)
+        context = runtime.context if runtime else None
+        
+        # Invoke the subgraph with the extracted context
+        return await self.pregel.ainvoke(input, config, context=context, **kwargs)
+
+
 def coerce_to_runnable(
     thing: RunnableLike, *, name: str | None, trace: bool
 ) -> Runnable:
@@ -902,3 +948,4 @@ async def _consume_aiter(it: AsyncIterator[Any]) -> Any:
             output = chunk
     return output
 
+