diff --git a/langchain/callbacks/file.py b/langchain/callbacks/file.py index 7c13295364..1c359e7a1d 100644 --- a/langchain/callbacks/file.py +++ b/langchain/callbacks/file.py @@ -24,7 +24,7 @@ class FileCallbackHandler(BaseCallbackHandler): self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs: Any ) -> None: """Print out that we are entering a chain.""" - class_name = serialized["name"] + class_name = serialized.get("name", serialized.get("id", [""])[-1]) print_text( f"\n\n\033[1m> Entering new {class_name} chain...\033[0m", end="\n",