From b7c3ac4501e1a24cec06ad76deed7d1b8e19527e Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 20 Jan 2025 13:55:13 -0800 Subject: [PATCH] Fix tracing output --- libs/langgraph/langgraph/pregel/call.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/call.py b/libs/langgraph/langgraph/pregel/call.py index f8c6b22f3..22feb46bb 100644 --- a/libs/langgraph/langgraph/pregel/call.py +++ b/libs/langgraph/langgraph/pregel/call.py @@ -10,7 +10,7 @@ from typing import Any, Callable, Optional, TypeVar, Union from typing_extensions import ParamSpec -from langgraph.constants import CONF, CONFIG_KEY_CALL, RETURN +from langgraph.constants import CONF, CONFIG_KEY_CALL, RETURN, TAG_HIDDEN from langgraph.pregel.write import ChannelWrite, ChannelWriteEntry from langgraph.types import RetryPolicy from langgraph.utils.config import get_config @@ -157,7 +157,7 @@ def get_runnable_for_entrypoint(func: Callable[..., Any]) -> RunnableSeq: ) seq = RunnableSeq( run, - ChannelWrite([ChannelWriteEntry(RETURN)]), + ChannelWrite([ChannelWriteEntry(RETURN)], tags=[TAG_HIDDEN]), name=func.__name__, ) if not _lookup_module_and_qualname(func): @@ -184,7 +184,7 @@ def get_runnable_for_task(func: Callable[..., Any]) -> RunnableSeq: ) seq = RunnableSeq( run, - ChannelWrite([ChannelWriteEntry(RETURN)]), + ChannelWrite([ChannelWriteEntry(RETURN)], tags=[TAG_HIDDEN]), name=func.__name__, trace_inputs=functools.partial( _explode_args_trace_inputs, inspect.signature(func)