chore: add correct TypeIs annotation for is_coroutine_fn (#1306)

Author: georgeh0

pyproject.toml

@@ -10,7 +10,7 @@ authors = [{ name = "CocoIndex", email = "[email protected]" }]
 readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
-    "typing-extensions>=4.12; python_version < '3.13'",
+    "typing-extensions>=4.12",
     "click>=8.1.8",
     "rich>=14.0.0",
     "python-dotenv>=1.1.0",

python/cocoindex/query_handler.py

@@ -4,11 +4,7 @@
 from typing import Generic, Any
 from .index import VectorSimilarityMetric
 import sys
-
-if sys.version_info >= (3, 13):
-    from typing import TypeVar
-else:
-    from typing_extensions import TypeVar  # PEP 696 backport
+from typing_extensions import TypeVar
 
 
 @dataclasses.dataclass

python/cocoindex/runtime.py

@@ -8,9 +8,11 @@
 import inspect
 import warnings
 
-from typing import Any, Callable, Awaitable, TypeVar, Coroutine
+from typing import Any, Callable, Awaitable, TypeVar, Coroutine, ParamSpec
+from typing_extensions import TypeIs
 
 T = TypeVar("T")
+P = ParamSpec("P")
 
 
 class _ExecutionContext:
@@ -68,14 +70,16 @@ def run(self, coro: Coroutine[Any, Any, T]) -> T:
 execution_context = _ExecutionContext()
 
 
-def is_coroutine_fn(fn: Callable[..., Any]) -> bool:
+def is_coroutine_fn(
+    fn: Callable[P, T] | Callable[P, Coroutine[Any, Any, T]],
+) -> TypeIs[Callable[P, Coroutine[Any, Any, T]]]:
     if isinstance(fn, (staticmethod, classmethod)):
         return inspect.iscoroutinefunction(fn.__func__)
     else:
         return inspect.iscoroutinefunction(fn)
 
 
-def to_async_call(fn: Callable[..., Any]) -> Callable[..., Awaitable[Any]]:
+def to_async_call(fn: Callable[P, T]) -> Callable[P, Awaitable[T]]:
     if is_coroutine_fn(fn):
         return fn
     return lambda *args, **kwargs: asyncio.to_thread(fn, *args, **kwargs)