fix: fix typing

qdrant · Jan 22, 2025 · 721fed5 · 721fed5
1 parent 3980758
commit 721fed5
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/qdrant_client/embed/embedder.py b/qdrant_client/embed/embedder.py
@@ -17,13 +17,15 @@
     ImageInput,
 )
 
-T = TypeVar("T", TextEmbedding, SparseTextEmbedding, LateInteractionTextEmbedding, ImageEmbedding)
+if TextEmbedding is not None:
+    T = TypeVar(
+        "T", *[TextEmbedding, SparseTextEmbedding, LateInteractionTextEmbedding, ImageEmbedding]
+    )
+else:
+    T = TypeVar("T")
 
 
-class ModelInstance(BaseModel, Generic[T]):
-    class Config:
-        arbitrary_types_allowed = True
-
+class ModelInstance(BaseModel, Generic[T], arbitrary_types_allowed=True):
     model: T
     options: dict[str, Any]
     deprecated: bool = False