Update llama_types.py

Smartappli · web-flow · commit 683868b213cb · 2024-08-15T07:54:41.000+02:00
diff --git a/llama_cpp/llama_types.py b/llama_cpp/llama_types.py
@@ -25,7 +25,7 @@ class EmbeddingUsage(TypedDict):
 class Embedding(TypedDict):
     index: int
     object: str
-    embedding: list[float] | list[list[float]]
+    embedding: Union[List[float], List[List[float]]]
 
 
 class CreateEmbeddingResponse(TypedDict):
