handle OR and other checks

Signed-off-by: Raayan Dhar [email protected] <[email protected]>

Author: raayandhar

python/sglang/utils.py

@@ -323,6 +323,31 @@ def __call__(self, *args, **kwargs):
         module = self._load()
         return module(*args, **kwargs)
 
+    def __or__(self, other):
+        self_type = self._load()
+        if isinstance(other, LazyImport):
+            other_type = other._load()
+        elif isinstance(other, tuple):
+            # Handle chaining: A | (B, C) -> (A, B, C)
+            return (self_type,) + other
+        else:
+            other_type = other
+        return (self_type, other_type)
+
+    def __ror__(self, other):
+        self_type = self._load()
+        if isinstance(other, LazyImport):
+            other_type = other._load()
+        elif isinstance(other, tuple):
+            # Handle chaining: (A, B) | C -> (A, B, C)
+            return other + (self_type,)
+        else:
+            other_type = other
+        return (other_type, self_type)
+
+    def __instancecheck__(self, instance):
+        return isinstance(instance, self._load())
+
 
 def download_and_cache_file(url: str, filename: Optional[str] = None):
     """Read and cache a file from a url."""