github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-28 22:14:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Unset context to None in var (#30380)

Browse Source
This commit is contained in:
William FH 2025-03-19 11:53:17 -07:00 committed by GitHub
parent 4130e6476b
commit 8265be4d3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libs/core/langchain_core/runnables/config.py
View File

@ -153,9 +153,7 @@ def _set_config_context(
@contextmanager
def set_config_context(
config: RunnableConfig, ctx: Optional[Context] = None
) -> Generator[Context, None, None]:
def set_config_context(config: RunnableConfig) -> Generator[Context, None, None]:
"""Set the child Runnable config + tracing context.
Args:
@ -163,14 +161,23 @@ def set_config_context(
"""
from langsmith.run_helpers import _set_tracing_context
ctx = ctx if ctx is not None else copy_context()
config_token, current_context = ctx.run(_set_config_context, config)
ctx = copy_context()
config_token, _ = ctx.run(_set_config_context, config)
try:
yield ctx
finally:
ctx.run(var_child_runnable_config.reset, config_token)
if current_context is not None:
ctx.run(_set_tracing_context, current_context)
ctx.run(
_set_tracing_context,
{
"parent": None,
"project_name": None,
"tags": None,
"metadata": None,
"enabled": None,
"client": None,
},
)
def ensure_config(config: Optional[RunnableConfig] = None) -> RunnableConfig: