fix: add callback

main.py

@@ -55,14 +55,17 @@ class MyCustomSyncHandler(BaseCallbackHandler):
         self.redisClient.publish(f'{kwargs["tags"][0]}',  self.message)
 
     def on_llm_end(self, response: LLMResult, **kwargs: Any) -> Any:
+        print("on_llm_end end")
         self.redisClient.publish(f'{kwargs["tags"][0]}',  'end')
 
     def on_llm_error(
         self, error: Union[Exception, KeyboardInterrupt], **kwargs: Any
     ) -> Any:
+        print("on_llm_error end")
         self.redisClient.publish(f'{kwargs["tags"][0]}',  'end')
 
     def on_chain_end(self, outputs: Dict[str, Any], **kwargs: Any) -> Any:
+        print("on_chain_end end")
         self.redisClient.publish(f'{kwargs["tags"][0]}',  'end')
 
 handleCallback = MyCustomSyncHandler(redisClient)