Python 3.10 type hinting compatibility

Author: ncoghlan

src/lmstudio/plugin/_dev_runner.py

@@ -8,7 +8,13 @@
 from contextlib import asynccontextmanager
 from pathlib import Path
 from functools import partial
-from typing import AsyncGenerator, Iterable, TypeAlias, assert_never
+from typing import AsyncGenerator, Iterable, TypeAlias
+
+from typing_extensions import (
+    # Native in 3.11+
+    assert_never,
+)
+
 
 from .runner import (
     ENV_CLIENT_ID,

src/lmstudio/plugin/config_schemas.py

@@ -5,10 +5,14 @@
     Any,
     ClassVar,
     Generic,
-    Self,
     Sequence,
     TypeVar,
     cast,
+)
+
+from typing_extensions import (
+    # Native in 3.11+
+    Self,
     dataclass_transform,
 )
 

src/lmstudio/plugin/hooks/common.py

@@ -3,7 +3,7 @@
 import asyncio
 
 from contextlib import asynccontextmanager
-from datetime import datetime, UTC
+from datetime import datetime, timezone
 from pathlib import Path
 from random import randrange
 from typing import (
@@ -90,7 +90,7 @@ def _parse_config(
 
     @classmethod
     def _create_ui_block_id(self) -> str:
-        return f"{datetime.now(UTC)}-{randrange(0, 2**32):08x}"
+        return f"{datetime.now(timezone.utc)}-{randrange(0, 2**32):08x}"
 
 
 StatusUpdateCallback: TypeAlias = Callable[[str, StatusStepStatus, str], Any]

src/lmstudio/plugin/hooks/prompt_preprocessor.py

@@ -13,6 +13,9 @@
     Generic,
     Iterable,
     TypeAlias,
+)
+from typing_extensions import (
+    # Native in 3.11+
     assert_never,
 )