Fix/type annotations (#211)

* correct missing/old references in state_manage.pyi

* update changelog

* bump version

* fix init method type annotation

Author: NodeJSmith

CHANGELOG.md

@@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+## [0.19.1] - 2026-01-25
+
+### Fixed
+- Update `state_manager.pyi` to fix type hints
+
 ## [0.19.0] - 2026-01-25
 
 ### Fixed

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "hassette"
-version = "0.19.0"
+version = "0.19.1"
 description = "Hassette is a simple, modern, async-first Python framework for building Home Assistant automations."
 readme = "README.md"
 authors = [{ name = "Jessica", email = "12jessicasmith34@gmail.com" }]

src/hassette/state_manager/state_manager.pyi

@@ -9,30 +9,46 @@ For custom domains not listed here, users should use:
 """
 
 import typing
+from collections.abc import Iterator
+from typing import Generic, NamedTuple
+
+from frozendict import frozendict
 
 from hassette.models import states
-from hassette.models.states import BaseState, StateT
+from hassette.models.states import StateT
 from hassette.resources.base import Resource
 
 if typing.TYPE_CHECKING:
     from hassette import Hassette
     from hassette.core.state_proxy import StateProxy
+    from hassette.events import HassStateDict
 
-class _TypedStateGetter(typing.Generic[StateT]):
-    def __init__(self, proxy: StateProxy, model: type[StateT]): ...
-    def __call__(self, entity_id: str) -> StateT: ...
-    def get(self, entity_id: str) -> StateT | None: ...
-
-class _StateGetter:
-    def __init__(self, proxy: StateProxy): ...
-    def __getitem__(self, model: type[StateT]) -> _TypedStateGetter[StateT]: ...
+class CacheValue(Generic[StateT], NamedTuple):
+    context_id: str | None
+    frozen_state: frozendict
+    model: StateT
 
 class DomainStates(typing.Generic[StateT]):
-    def __init__(self, states_dict: dict[str, BaseState], domain: str) -> None: ...
+    _state_proxy: StateProxy
+    _model: type[StateT]
+    _domain: str
+    _cache: dict[str, CacheValue[StateT]]
+
+    def _validate_or_return_from_cache(self, entity_id: str, state: HassStateDict) -> StateT: ...
+    def __init__(self, state_proxy: StateProxy, model: type[StateT]) -> None: ...
     def __iter__(self) -> typing.Generator[tuple[str, StateT], typing.Any, None]: ...
     def __len__(self) -> int: ...
-    def get(self, entity_id: str) -> StateT | None: ...
     def __getitem__(self, entity_id: str) -> StateT: ...
+    def __contains__(self, entity_id: str) -> bool: ...
+    def __repr__(self) -> str: ...
+    def __bool__(self) -> bool: ...
+    def get(self, entity_id: str) -> StateT | None: ...
+    def items(self) -> Iterator[tuple[str, StateT]]: ...
+    def keys(self) -> list[str]: ...
+    def iterkeys(self) -> typing.Iterator[str]: ...
+    def values(self) -> list[StateT]: ...
+    def itervalues(self) -> Iterator[StateT]: ...
+    def to_dict(self) -> dict[str, StateT]: ...
 
 class StateManager(Resource):
     """Resource for managing Home Assistant states.