diff --git a/llama_cpp/llama_types.py b/llama_cpp/llama_types.py
index bfc7342..e8f4ce1 100644
--- a/llama_cpp/llama_types.py
+++ b/llama_cpp/llama_types.py
@@ -1,4 +1,4 @@
-from typing import List, Optional, Dict, Union
+from typing import List, Optional, Dict
 from typing_extensions import TypedDict, NotRequired, Literal