diff --git a/src/ai_powered/decorators.py b/src/ai_powered/decorators.py
index 890b8fa..77e43e3 100644
--- a/src/ai_powered/decorators.py
+++ b/src/ai_powered/decorators.py
@@ -1,5 +1,5 @@
 from functools import wraps
-from typing import Any, Callable, Generic, ParamSpec, TypeVar
+from typing_extensions import Any, Callable, Generic, ParamSpec, TypeVar
 import json
 import msgspec
 
diff --git a/src/ai_powered/llm_adapter/generic_adapter.py b/src/ai_powered/llm_adapter/generic_adapter.py
index 08c0bf7..5c89509 100644
--- a/src/ai_powered/llm_adapter/generic_adapter.py
+++ b/src/ai_powered/llm_adapter/generic_adapter.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass
 import re
-from typing import Any, Callable, Set
+from typing_extensions import Any, Callable, Set
 from ai_powered.colors import green, yellow
 from ai_powered.constants import DEBUG
 from .definitions import FunctionSimulator, ModelFeature
diff --git a/src/ai_powered/llm_adapter/known_models.py b/src/ai_powered/llm_adapter/known_models.py
index 75578a8..7856a34 100644
--- a/src/ai_powered/llm_adapter/known_models.py
+++ b/src/ai_powered/llm_adapter/known_models.py
@@ -1,6 +1,6 @@
 from copy import deepcopy
 from dataclasses import dataclass, field
-from typing import Any, Callable, Optional, Set, TypeAlias
+from typing_extensions import Any, Callable, Optional, Set, TypeAlias
 
 from ai_powered.llm_adapter.definitions import ALL_FEATURES, ModelFeature
 
diff --git a/src/ai_powered/schema_deref.py b/src/ai_powered/schema_deref.py
index 875dbb9..4114380 100644
--- a/src/ai_powered/schema_deref.py
+++ b/src/ai_powered/schema_deref.py
@@ -1,5 +1,5 @@
 
-from typing import Any
+from typing_extensions import Any
 
 def deref(schema: dict[str, Any]) -> dict[str, Any]:
     '''
diff --git a/test/examples/ai_powered_decorator/gen_json_data.py b/test/examples/ai_powered_decorator/gen_json_data.py
index 0a516c9..527aacb 100644
--- a/test/examples/ai_powered_decorator/gen_json_data.py
+++ b/test/examples/ai_powered_decorator/gen_json_data.py
@@ -1,4 +1,4 @@
-from typing import Any
+from typing_extensions import Any
 from ai_powered import ai_powered
 
 @ai_powered