From 224199083bbc644e9bfac60cfedfab7b9841eb54 Mon Sep 17 00:00:00 2001 From: Sam Coward Date: Thu, 13 Jul 2023 03:05:36 -0400 Subject: [PATCH] Fix missing chain classname in StdOutCallbackHandler.on_chain_start (#6124) Retrieves the name of the class from new location as of commit 18af149e91e62b3ac7728ddea420688d41043734 Co-authored-by: Zander Chase <130414180+vowelparrot@users.noreply.github.com> --- langchain/callbacks/stdout.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/callbacks/stdout.py b/langchain/callbacks/stdout.py index baa03de521..add431d970 100644 --- a/langchain/callbacks/stdout.py +++ b/langchain/callbacks/stdout.py @@ -37,7 +37,7 @@ class StdOutCallbackHandler(BaseCallbackHandler): self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs: Any ) -> None: """Print out that we are entering a chain.""" - class_name = serialized.get("name", "") + class_name = serialized.get("name", serialized.get("id", [""])[-1]) print(f"\n\n\033[1m> Entering new {class_name} chain...\033[0m") def on_chain_end(self, outputs: Dict[str, Any], **kwargs: Any) -> None: