if debug is used, also use the notificaition callback for the error

Author: thiswillbeyourgithub

WDoc/WDoc.py

@@ -135,14 +135,27 @@ def __init__(
         **cli_kwargs,
     ) -> None:
         "This docstring is dynamically replaced by the content of WDoc/docs/USAGE.md"
+        if notification_callback is not None:
+            @optional_typecheck
+            def ntfy(text: str) -> str:
+                out = notification_callback(text)
+                assert out == text, "The notification callback must return the same string"
+                return out
+            ntfy("Starting WDoc")
+        else:
+            @optional_typecheck
+            def ntfy(text: str) -> str:
+                return text
+        self.ntfy = ntfy
+
         if debug:
             def handle_exception(exc_type, exc_value, exc_traceback):
                 if not issubclass(exc_type, KeyboardInterrupt):
                     @optional_typecheck
                     def p(message: str) -> None:
                         "print error, in red if possible"
                         try:
-                            red(message)
+                            red(self.ntfy(message))
                         except Exception as err:
                             print(message)
                     p("\n--verbose was used so opening debug console at the "
@@ -376,19 +389,6 @@ def p(message: str) -> None:
                 raise Exception(
                     red(f"Can't find the price of {query_eval_modelname}"))
 
-        if notification_callback is not None:
-            @optional_typecheck
-            def ntfy(text: str) -> str:
-                out = notification_callback(text)
-                assert out == text, "The notification callback must return the same string"
-                return out
-            ntfy("Starting WDoc")
-        else:
-            @optional_typecheck
-            def ntfy(text: str) -> str:
-                return text
-        self.ntfy = ntfy
-
         if is_verbose:
             # os.environ["LANGCHAIN_TRACING_V2"] = "true"
             set_verbose(True)