Fix tracing output

This commit is contained in:
Nuno Campos
2025-01-20 13:55:13 -08:00
parent 07695f5c5a
commit b7c3ac4501
+3 -3
View File
@@ -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)