Store audit results via callback rather than local console reference

Author: treysp

sqlmesh/core/scheduler.py

@@ -168,6 +168,8 @@ def evaluate(
         batch_index: int,
         environment_naming_info: EnvironmentNamingInfo,
         default_catalog: t.Optional[str],
+        on_audits_complete: t.Optional[t.Callable[[Snapshot, t.List[AuditResult]], None]] = None,
+        on_audit_warning: t.Optional[t.Callable[[str, t.Optional[str]], None]] = None,
         **kwargs: t.Any,
     ) -> t.Tuple[t.List[AuditResult], t.List[AuditError]]:
         """Evaluate a snapshot and add the processed interval to the state sync.
@@ -211,7 +213,8 @@ def evaluate(
             wap_id=wap_id,
             **kwargs,
         )
-        get_console().store_evaluation_audit_results(snapshot, audit_results)
+        if on_audits_complete:
+            on_audits_complete(snapshot, audit_results)
 
         audit_errors_to_raise: t.List[AuditError] = []
         audit_errors_to_warn: t.List[AuditError] = []