Remove GetLocal, PutLocal (#12133)

Do you agree?

libs/langchain/langchain/schema/runnable/__init__.py

@@ -14,7 +14,6 @@
 
 This module contains schema and implementation of LangChain Runnables primitives.
 """
-from langchain.schema.runnable._locals import GetLocalVar, PutLocalVar
 from langchain.schema.runnable.base import (
     Runnable,
     RunnableBinding,
@@ -40,9 +39,7 @@
     "ConfigurableField",
     "ConfigurableFieldSingleOption",
     "ConfigurableFieldMultiOption",
-    "GetLocalVar",
     "patch_config",
-    "PutLocalVar",
     "RouterInput",
     "RouterRunnable",
     "Runnable",

libs/langchain/langchain/schema/runnable/base.py

@@ -1656,7 +1656,6 @@ def invoke(
                         # mark each step as a child run
                         patch_config(
                             config,
-                            copy_locals=True,
                             callbacks=run_manager.get_child(f"map:key:{key}"),
                         ),
                     )
@@ -2534,10 +2533,7 @@ def batch(
                 [merge_configs(self.config, conf) for conf in config],
             )
         else:
-            configs = [
-                patch_config(merge_configs(self.config, config), copy_locals=True)
-                for _ in range(len(inputs))
-            ]
+            configs = [merge_configs(self.config, config) for _ in range(len(inputs))]
         return self.bound.batch(
             inputs,
             configs,
@@ -2559,10 +2555,7 @@ async def abatch(
                 [merge_configs(self.config, conf) for conf in config],
             )
         else:
-            configs = [
-                patch_config(merge_configs(self.config, config), copy_locals=True)
-                for _ in range(len(inputs))
-            ]
+            configs = [merge_configs(self.config, config) for _ in range(len(inputs))]
         return await self.bound.abatch(
             inputs,
             configs,

libs/langchain/langchain/schema/runnable/config.py

@@ -64,13 +64,6 @@ class RunnableConfig(TypedDict, total=False):
     Name for the tracer run for this call. Defaults to the name of the class.
     """
 
-    locals: Dict[str, Any]
-    """
-    Variables scoped to this call and any sub-calls. Usually used with
-    GetLocalVar() and PutLocalVar(). Care should be taken when placing mutable
-    objects in locals, as they will be shared between parallel sub-calls.
-    """
-
     max_concurrency: Optional[int]
     """
     Maximum number of parallel calls to make. If not provided, defaults to 
@@ -96,7 +89,6 @@ def ensure_config(config: Optional[RunnableConfig] = None) -> RunnableConfig:
         tags=[],
         metadata={},
         callbacks=None,
-        locals={},
         recursion_limit=25,
     )
     if config is not None:
@@ -124,23 +116,20 @@ def get_config_list(
     return (
         list(map(ensure_config, config))
         if isinstance(config, list)
-        else [patch_config(config, copy_locals=True) for _ in range(length)]
+        else [ensure_config(config) for _ in range(length)]
     )
 
 
 def patch_config(
     config: Optional[RunnableConfig],
     *,
-    copy_locals: bool = False,
     callbacks: Optional[BaseCallbackManager] = None,
     recursion_limit: Optional[int] = None,
     max_concurrency: Optional[int] = None,
     run_name: Optional[str] = None,
     configurable: Optional[Dict[str, Any]] = None,
 ) -> RunnableConfig:
     config = ensure_config(config)
-    if copy_locals:
-        config["locals"] = config["locals"].copy()
     if callbacks is not None:
         # If we're replacing callbacks we need to unset run_name
         # As that should apply only to the same run as the original callbacks

libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py

@@ -1209,7 +1209,6 @@ async def test_with_config(mocker: MockerFixture) -> None:
                 metadata={"key": "value"},
                 tags=["c"],
                 callbacks=None,
-                locals={},
                 recursion_limit=5,
             ),
         ),
@@ -1219,7 +1218,6 @@ async def test_with_config(mocker: MockerFixture) -> None:
                 metadata={"key": "value"},
                 tags=["c"],
                 callbacks=None,
-                locals={},
                 recursion_limit=5,
             ),
         ),
@@ -1290,7 +1288,6 @@ async def test_default_method_implementations(mocker: MockerFixture) -> None:
                 metadata={"key": "value"},
                 tags=[],
                 callbacks=None,
-                locals={},
                 recursion_limit=25,
             ),
         ),
@@ -1300,7 +1297,6 @@ async def test_default_method_implementations(mocker: MockerFixture) -> None:
                 metadata={"key": "value"},
                 tags=[],
                 callbacks=None,
-                locals={},
                 recursion_limit=25,
             ),
         ),