TY linting (replaces mypi)

.cursor/worktrees.json

@@ -0,0 +1,5 @@
+{
+  "setup-worktree": [
+    "npm install"
+  ]
+}

.github/workflows/run_tests.yml

@@ -29,6 +29,37 @@ jobs:
       - name: Run Ruff formatter
         run: uv run ruff format --check .
 
+  typecheck:
+    name: Type Check (ty)
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+      - name: Install dependencies
+        uses: ./.github/actions/python-uv-setup
+      - name: Run ty type checker
+        run: |
+          uvx ty check getstream/ \
+            --exclude "getstream/models/" \
+            --exclude "getstream/video/rtc/pb/" \
+            --exclude "**/rest_client.py" \
+            --exclude "**/async_rest_client.py" \
+            --exclude "getstream/chat/channel.py" \
+            --exclude "getstream/chat/async_channel.py" \
+            --exclude "getstream/chat/client.py" \
+            --exclude "getstream/chat/async_client.py" \
+            --exclude "getstream/common/client.py" \
+            --exclude "getstream/common/async_client.py" \
+            --exclude "getstream/moderation/client.py" \
+            --exclude "getstream/moderation/async_client.py" \
+            --exclude "getstream/video/client.py" \
+            --exclude "getstream/video/async_client.py" \
+            --exclude "getstream/video/call.py" \
+            --exclude "getstream/video/async_call.py" \
+            --exclude "getstream/feeds/client.py" \
+            --exclude "getstream/feeds/feeds.py" \
+            --exclude "getstream/stream.py"
+
   test:
     name: Test "${{ inputs.marker }}"
     environment:

DEVELOPMENT.md

@@ -29,10 +29,10 @@ pre-commit install
 
 ### Check
 
-Shortcut to ruff, mypy and non integration tests:
+Shortcut to ruff, ty (type checker) and non integration tests:
 
 ```
-uv run python dev.py check
+uv run python dev.py
 ```
 
 ### Formatting
@@ -41,10 +41,17 @@ uv run python dev.py check
 uv run ruff check --fix
 ```
 
-### Mypy type checks
+### Type checking (ty)
 
+Type checking is run via the `ty` type checker, excluding generated code:
+
+```
+uv run python dev.py ty
+```
+
+Or manually (note: requires exclude flags for generated code - see dev.py for the full list):
 ```
-uv run mypy --install-types --non-interactive --exclude 'getstream/models/.*' .
+uvx ty check getstream/ --exclude "getstream/models/" --exclude "getstream/video/rtc/pb/" ...
 ```
 
 ## Release

dev.py

@@ -15,7 +15,7 @@
 
 def run(
     command: str, env: Optional[dict] = None, check: bool = True
-) -> subprocess.CompletedProcess:
+) -> subprocess.CompletedProcess[str]:
     """Run a shell command with automatic argument parsing."""
     click.echo(f"Running: {command}")
 
@@ -24,8 +24,8 @@ def run(
     if env:
         full_env.update(env)
 
+    cmd_list = shlex.split(command)
     try:
-        cmd_list = shlex.split(command)
         result = subprocess.run(
             cmd_list, check=check, capture_output=False, env=full_env, text=True
         )
@@ -34,15 +34,18 @@ def run(
         if check:
             click.echo(f"Command failed with exit code {e.returncode}", err=True)
             sys.exit(e.returncode)
-        return e
+        # Return a CompletedProcess with the error info when check=False
+        return subprocess.CompletedProcess(
+            cmd_list, e.returncode, stdout=None, stderr=None
+        )
 
 
 @click.group(invoke_without_command=True)
 @click.pass_context
 def cli(ctx):
-    """Development CLI tool for getstream SDK."""
+    """Development CLI tool for getstream SDK. Runs 'check' by default."""
     if ctx.invoked_subcommand is None:
-        click.echo(ctx.get_help())
+        ctx.invoke(check)
 
 
 @cli.command()
@@ -75,26 +78,63 @@ def lint():
     run("uv run ruff format --check .")
 
 
+# Generated code patterns to exclude from type checking
+TY_EXCLUDES = [
+    "getstream/models/",
+    "getstream/video/rtc/pb/",
+    "**/rest_client.py",
+    "**/async_rest_client.py",
+    "getstream/chat/channel.py",
+    "getstream/chat/async_channel.py",
+    "getstream/chat/client.py",
+    "getstream/chat/async_client.py",
+    "getstream/common/client.py",
+    "getstream/common/async_client.py",
+    "getstream/moderation/client.py",
+    "getstream/moderation/async_client.py",
+    "getstream/video/client.py",
+    "getstream/video/async_client.py",
+    "getstream/video/call.py",
+    "getstream/video/async_call.py",
+    "getstream/feeds/client.py",
+    "getstream/feeds/feeds.py",
+    "getstream/stream.py",
+]
+
+
+def _run_typecheck():
+    """Internal function to run ty type checks."""
+    click.echo("Running ty type checker on getstream...")
+    excludes = " ".join(f'--exclude "{e}"' for e in TY_EXCLUDES)
+    run(f"uvx ty check getstream/ {excludes}")
+
+
 @cli.command()
-def mypy():
-    """Run mypy type checks on getstream package."""
-    click.echo("Running mypy on getstream...")
-    run("uv run mypy --install-types --non-interactive getstream")
+def typecheck():
+    """Run ty type checks on getstream package (excludes generated code)."""
+    _run_typecheck()
+
+
+@cli.command("ty")
+def ty_alias():
+    """Run ty type checks (alias for 'typecheck')."""
+    _run_typecheck()
 
 
 @cli.command()
 def check():
-    """Run full check: ruff, mypy, and unit tests."""
+    """Run full check: ruff, ty, and unit tests."""
     click.echo("Running full development check...")
 
     # Run ruff
     click.echo("\n=== 1. Ruff Linting ===")
     run("uv run ruff check .")
     run("uv run ruff format --check .")
 
-    # Run mypy on main package
-    click.echo("\n=== 2. MyPy Type Checking ===")
-    run("uv run mypy --install-types --non-interactive getstream")
+    # Run ty type checker on main package (excludes generated code)
+    click.echo("\n=== 2. Type Checking (ty) ===")
+    excludes = " ".join(f'--exclude "{e}"' for e in TY_EXCLUDES)
+    run(f"uvx ty check getstream/ {excludes}")
 
     # Run unit tests
     click.echo("\n=== 3. Unit Tests ===")
@@ -103,5 +143,19 @@ def check():
     click.echo("\n✅ All checks passed!")
 
 
+@cli.command()
+def regen():
+    """Regenerate all generated code (OpenAPI + WebRTC protobuf)."""
+    click.echo("Regenerating all generated code...")
+
+    click.echo("\n=== 1. OpenAPI Code Generation ===")
+    run("./generate.sh")
+
+    click.echo("\n=== 2. WebRTC Protobuf Generation ===")
+    run("./generate_webrtc.sh")
+
+    click.echo("\n✅ Code regeneration complete!")
+
+
 if __name__ == "__main__":
     cli()

getstream/base.py

@@ -2,7 +2,7 @@
 import time
 import uuid
 import asyncio
-from typing import Any, Dict, Optional, Type, get_origin
+from typing import Any, Dict, Optional, Type, cast, get_origin
 
 from getstream.models import APIError
 from getstream.rate_limit import extract_rate_limit
@@ -25,7 +25,7 @@
 import ijson
 
 
-def build_path(path: str, path_params: dict) -> str:
+def build_path(path: str, path_params: Optional[Dict[str, Any]]) -> str:
     if path_params is None:
         return path
     for k, v in path_params.items():
@@ -46,12 +46,14 @@ def _parse_response(
         try:
             parsed_result = json.loads(response.text) if response.text else {}
 
+            data: T
             if hasattr(data_type, "from_dict"):
-                data = data_type.from_dict(parsed_result, infer_missing=True)
+                from_dict = getattr(data_type, "from_dict")
+                data = from_dict(parsed_result, infer_missing=True)
             elif get_origin(data_type) is not dict:
                 raise AttributeError(f"{data_type.__name__} has no 'from_dict' method")
             else:
-                data = parsed_result
+                data = cast(T, parsed_result)
 
         except ValueError:
             raise StreamAPIException(
@@ -140,7 +142,7 @@ def __init__(
             user_agent=user_agent,
         )
         self.client = httpx.Client(
-            base_url=self.base_url,
+            base_url=self.base_url or "",
             headers=self.headers,
             params=self.params,
             timeout=httpx.Timeout(self.timeout),
@@ -154,7 +156,7 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 
     def _endpoint_name(self, path: str) -> str:
         op = getattr(self, "_operation_name", None)
-        return op or current_operation(self._normalize_endpoint_from_path(path))
+        return op or current_operation(self._normalize_endpoint_from_path(path)) or ""
 
     def _request_sync(
         self,
@@ -315,25 +317,25 @@ def __init__(
             user_agent=user_agent,
         )
         self.client = httpx.AsyncClient(
-            base_url=self.base_url,
+            base_url=self.base_url or "",
             headers=self.headers,
             params=self.params,
             timeout=httpx.Timeout(self.timeout),
         )
 
-    def __aenter__(self):
+    async def __aenter__(self):
         return self
 
-    def __aexit__(self, exc_type, exc_val, exc_tb):
-        self.close()
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.aclose()
 
     async def aclose(self):
         """Close HTTPX async client (closes pools/keep-alives)."""
         await self.client.aclose()
 
     def _endpoint_name(self, path: str) -> str:
         op = getattr(self, "_operation_name", None)
-        return op or current_operation(self._normalize_endpoint_from_path(path))
+        return op or current_operation(self._normalize_endpoint_from_path(path)) or ""
 
     async def _request_async(
         self,

getstream/common/telemetry.py

@@ -176,7 +176,7 @@ def span_request(
     if tracer is None:
         yield _NullSpan()  # pragma: no cover
         return
-    with tracer.start_as_current_span(name, kind=SpanKind.CLIENT) as span:  # type: ignore[arg-type]
+    with tracer.start_as_current_span(name, kind=SpanKind.CLIENT) as span:
         base_attrs: Dict[str, Any] = dict(attributes or {})
         # auto-propagate contextual IDs to request spans
         try:
@@ -190,22 +190,22 @@ def span_request(
             pass
         if base_attrs:
             try:
-                span.set_attributes(base_attrs)  # type: ignore[attr-defined]
+                span.set_attributes(base_attrs)
             except Exception:
                 pass
         if include_bodies and request_body is not None:
             try:
                 span.add_event(
                     "request.body", {"level": "INFO", "body": safe_dump(request_body)}
-                )  # type: ignore[attr-defined]
+                )
             except Exception:
                 pass
         try:
             yield span
         except BaseException as e:
             try:
-                span.record_exception(e)  # type: ignore[attr-defined]
-                span.set_status(Status(StatusCode.ERROR, str(e)))  # type: ignore[attr-defined]
+                span.record_exception(e)
+                span.set_status(Status(StatusCode.ERROR, str(e)))
             except Exception:
                 pass
             raise
@@ -328,7 +328,7 @@ def start_as_current_span(
     if tracer is None:  # pragma: no cover
         yield _NullSpan()
         return
-    with tracer.start_as_current_span(name, kind=use_kind) as span:  # type: ignore[arg-type]
+    with tracer.start_as_current_span(name, kind=use_kind) as span:
         base_attrs: Dict[str, Any] = dict(attributes or {})
         # auto-propagate contextual IDs
         try:
@@ -342,7 +342,7 @@ def start_as_current_span(
             pass
         if base_attrs:
             try:
-                span.set_attributes(base_attrs)  # type: ignore[attr-defined]
+                span.set_attributes(base_attrs)
             except Exception:
                 pass
         yield span
@@ -372,7 +372,7 @@ def detach_context(token: Optional[object]) -> None:
     if not _HAS_OTEL or otel_context is None or token is None:  # pragma: no cover
         return
     try:
-        otel_context.detach(token)
+        otel_context.detach(token)  # type: ignore[arg-type]
     except Exception:
         # Best-effort; do not raise during detach
         pass

getstream/stream_response.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional, Generic
+from typing import Any, Dict, Optional, Generic
 import typing
 
 import httpx
@@ -27,9 +27,9 @@ def rate_limit(self) -> Optional[RateLimitInfo]:
         """Returns the ratelimit info of your API operation."""
         return self.__rate_limit
 
-    def headers(self) -> typing.Dict[str, Any]:
+    def headers(self) -> Dict[str, Any]:
         """Returns the headers of the response."""
-        return self.__headers
+        return dict(self.__headers)
 
     def status_code(self) -> int:
         """Returns the HTTP status code of the response."""

getstream/utils/__init__.py

@@ -1,6 +1,6 @@
 import json
 import asyncio
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union, cast
 from urllib.parse import quote
 from datetime import datetime
 from datetime import timezone
@@ -77,10 +77,12 @@ def datetime_from_unix_ns(
         return None
 
     if isinstance(ts, dict):
-        return {k: datetime_from_unix_ns(v) for k, v in ts.items()}
+        # Values in dict are int/float/str, so recursive calls return datetime
+        return {k: cast(datetime, datetime_from_unix_ns(v)) for k, v in ts.items()}
 
     if isinstance(ts, list):
-        return [datetime_from_unix_ns(v) for v in ts]
+        # Values in list are int/float/str, so recursive calls return datetime
+        return [cast(datetime, datetime_from_unix_ns(v)) for v in ts]
 
     if isinstance(ts, str):
         ts = int(ts)