From 2687a5ab7238c66f44c91bb4865281dcfc6220f7 Mon Sep 17 00:00:00 2001
From: Jerry Liu <jerryjliu98@gmail.com>
Date: Wed, 10 Jan 2024 10:47:20 -0800
Subject: [PATCH] nit: fix set callback handler in base synthesizer (#9956)
---
llama_index/response_synthesizers/base.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/llama_index/response_synthesizers/base.py b/llama_index/response_synthesizers/base.py
index 884cc403b4..d38e33d72d 100644
--- a/llama_index/response_synthesizers/base.py
+++ b/llama_index/response_synthesizers/base.py
@@ -67,7 +67,13 @@ class BaseSynthesizer(ChainableMixin, PromptMixin):
@callback_manager.setter
def callback_manager(self, callback_manager: CallbackManager) -> None:
+ """Set callback manager."""
self._callback_manager = callback_manager
+ # TODO: please fix this later
+ self._service_context.callback_manager = callback_manager
+ self._service_context.llm.callback_manager = callback_manager
+ self._service_context.embed_model.callback_manager = callback_manager
+ self._service_context.node_parser.callback_manager = callback_manager
@abstractmethod
def get_response(
--
GitLab