From a39c9983421b099c7479395604fed701cf31e96e Mon Sep 17 00:00:00 2001 From: Hansen Qian Date: Mon, 28 Nov 2022 14:11:30 -0500 Subject: [PATCH] Add chain name to verbose logging (#214) Adds some context over what chain is running, thereby making it more obvious how different chains are entered and existed Screen Shot 2022-11-28 at 11 55 34 AM (note that the `...` is because the output is too long and VSCode truncated it) --- langchain/chains/base.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/langchain/chains/base.py b/langchain/chains/base.py index 8e8005e7740..529e86dda41 100644 --- a/langchain/chains/base.py +++ b/langchain/chains/base.py @@ -81,10 +81,12 @@ class Chain(BaseModel, ABC): inputs = dict(inputs, **external_context) self._validate_inputs(inputs) if self.verbose: - print("\n\n\033[1m> Entering new chain...\033[0m") + print( + f"\n\n\033[1m> Entering new {self.__class__.__name__} chain...\033[0m" + ) outputs = self._call(inputs) if self.verbose: - print("\n\033[1m> Finished chain.\033[0m") + print(f"\n\033[1m> Finished {self.__class__.__name__} chain.\033[0m") self._validate_outputs(outputs) if self.memory is not None: self.memory.save_context(inputs, outputs)