fix: Fix py38 type union unsupported issue

alignai/integration/langchain.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import importlib
 from datetime import datetime
 from importlib import metadata as importlib_metadata

alignai/integration/llama_index.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import importlib
 from datetime import datetime
 from importlib import metadata as importlib_metadata

alignai/logger.py

@@ -1,17 +1,13 @@
 import logging
 import traceback
-import types
-from typing import Type
 
 
 class LimitedTracebackFormatter(logging.Formatter):
     def __init__(self, *args, stack_trace_limit=5, **kwargs):
         super().__init__(*args, **kwargs)
         self._stack_trace_limit = stack_trace_limit
 
-    def formatException(
-        self, exc_info: tuple[Type[BaseException], BaseException, types.TracebackType] | tuple[None, None, None]
-    ) -> str:
+    def formatException(self, exc_info) -> str:
         _, _, tb = exc_info
         return "".join(traceback.format_tb(tb)[-self._stack_trace_limit :])