Skip to content
Snippets Groups Projects
Unverified Commit 2687a5ab authored by Jerry Liu's avatar Jerry Liu Committed by GitHub
Browse files

nit: fix set callback handler in base synthesizer (#9956)

parent 80d732c4
Branches
Tags
No related merge requests found
......@@ -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(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment