move to correct audit

Author: benfdking

sqlmesh/lsp/main.py

@@ -6,20 +6,20 @@
 from contextlib import suppress
 from pathlib import Path
 
+from sqlmesh.core.audit.definition import ModelAudit
 from sqlmesh.core.context import Context
 from lsprotocol import types
 from pygls.server import LanguageServer
 from pygls.workspace import TextDocument
 from sqlmesh._version import __version__
-from sqlmesh.core.dialect import Audit
 from sqlmesh.core.model import Model
 
 logger = logging.getLogger(__name__)
 
 GLOBAL_CONTEXT: t.Optional[
     t.Tuple[
         Context,
-        t.Dict[str, t.Tuple[Context, t.Union[Model, Audit]]],
+        t.Dict[str, t.Tuple[Context, t.Union[Model, ModelAudit]]],
     ]
 ] = None