Update llama_types.py

Smartappli · Aug 15, 2024 · f747d46 · f747d46
1 parent d8d9c4d
commit f747d46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llama_cpp/llama_types.py b/llama_cpp/llama_types.py
@@ -7,9 +7,9 @@
 
 """
 
-from typing import Any, Dict, List, Literal, NotRequired, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
-from typing_extensions import TypedDict
+from typing_extensions import Literal, NotRequired, TypedDict
 
 # NOTE: Defining this correctly using annotations seems to break pydantic validation.
 #       This is a workaround until we can figure out how to do this correctly