From b82ad19ed25815f381fa59b719e21fb11c49d377 Mon Sep 17 00:00:00 2001 From: William FH <13333726+hinthornw@users.noreply.github.com> Date: Wed, 30 Aug 2023 15:30:22 -0700 Subject: [PATCH] Check memory address (#9971) Don't want to dup the collector but can have multiple --- libs/langchain/langchain/callbacks/manager.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/libs/langchain/langchain/callbacks/manager.py b/libs/langchain/langchain/callbacks/manager.py index 2f7a7fad478..52832338fcb 100644 --- a/libs/langchain/langchain/callbacks/manager.py +++ b/libs/langchain/langchain/callbacks/manager.py @@ -1799,6 +1799,9 @@ def _configure( for handler in callback_manager.handlers ): callback_manager.add_handler(open_ai, True) - if run_collector_ is not None: + if run_collector_ is not None and not any( + handler is run_collector_ # direct pointer comparison + for handler in callback_manager.handlers + ): callback_manager.add_handler(run_collector_, False) return callback_manager