HAT-301, HAT-302: Otel Constraints + Step Types (hatchet-dev#290)

* feat: better typing for step decorator

* fix: patch version

* fix: min allowed with current proto version

* fix: try adding a py.typed marker

* fix: lock

Author: hatchet-temporary

hatchet_sdk/hatchet.py

@@ -1,6 +1,6 @@
 import asyncio
 import logging
-from typing import Any, Callable, Optional, Type, TypeVar, Union
+from typing import Any, Callable, Optional, ParamSpec, Type, TypeVar, Union
 
 from pydantic import BaseModel
 from typing_extensions import deprecated
@@ -35,6 +35,9 @@
 )
 
 T = TypeVar("T", bound=BaseModel)
+R = TypeVar("R")
+P = ParamSpec("P")
+
 TWorkflow = TypeVar("TWorkflow", bound=object)
 
 
@@ -87,10 +90,10 @@ def step(
     desired_worker_labels: dict[str, DesiredWorkerLabel] = {},
     backoff_factor: float | None = None,
     backoff_max_seconds: int | None = None,
-) -> Callable[..., Any]:
+) -> Callable[[Callable[P, R]], Callable[P, R]]:
     parents = parents or []
 
-    def inner(func: Callable[[Context], Any]) -> Callable[[Context], Any]:
+    def inner(func: Callable[P, R]) -> Callable[P, R]:
         limits = None
         if rate_limits:
             limits = [rate_limit._req for rate_limit in rate_limits or []]
@@ -99,7 +102,6 @@ def inner(func: Callable[[Context], Any]) -> Callable[[Context], Any]:
         setattr(func, "_step_parents", parents)
         setattr(func, "_step_timeout", timeout)
         setattr(func, "_step_retries", retries)
-        setattr(func, "_step_rate_limits", retries)
         setattr(func, "_step_rate_limits", limits)
         setattr(func, "_step_backoff_factor", backoff_factor)
         setattr(func, "_step_backoff_max_seconds", backoff_max_seconds)

hatchet_sdk/py.typed

@@ -1,9 +1,10 @@
 [tool.poetry]
 name = "hatchet-sdk"
-version = "0.43.0"
+version = "0.43.1"
 description = ""
 authors = ["Alexander Belanger <[email protected]>"]
 readme = "README.md"
+include = ["hatchet_sdk/py.typed"]
 
 [tool.poetry.dependencies]
 python = "^3.10"
@@ -22,12 +23,12 @@ aiohttp = "^3.10.5"
 aiohttp-retry = "^2.8.3"
 tenacity = ">=8.4.1"
 cel-python = "^0.1.5"
-opentelemetry-api = "^1.27.0"
-opentelemetry-sdk = "^1.27.0"
-opentelemetry-instrumentation = "^0.48b0"
-opentelemetry-distro = "^0.48b0"
-opentelemetry-exporter-otlp = "^1.27.0"
-opentelemetry-exporter-otlp-proto-http = "^1.27.0"
+opentelemetry-api = "^1.28.0"
+opentelemetry-sdk = "^1.28.0"
+opentelemetry-instrumentation = ">=0.49b0"
+opentelemetry-distro = ">=0.49b0"
+opentelemetry-exporter-otlp = "^1.28.0"
+opentelemetry-exporter-otlp-proto-http = "^1.28.0"
 prometheus-client = "^0.21.1"
 
 [tool.poetry.group.dev.dependencies]
@@ -36,7 +37,7 @@ pytest-asyncio = "^0.23.8"
 psutil = "^6.0.0"
 
 [tool.poetry.group.lint.dependencies]
-mypy = "^1.13.0"
+mypy = "^1.14.0"
 types-protobuf = "^5.28.3.20241030"
 black = "^24.10.0"
 isort = "^5.13.2"