diff --git a/src/opengradient/types.py b/src/opengradient/types.py index fa89c98..bf25053 100644 --- a/src/opengradient/types.py +++ b/src/opengradient/types.py @@ -301,6 +301,12 @@ def __iter__(self): """Iterate over stream chunks.""" return self + def __aiter__(self): + """Return async iterator (required for async for loops).""" + if not self._is_async: + raise TypeError("Use __iter__ for sync iterators") + return self + def __next__(self) -> StreamChunk: """Get next stream chunk.""" import json