fix: replace Iterator/AsyncIterator with Generator/AsyncGenerator on all generator functions

Every function annotated -> Iterator[T] or -> AsyncIterator[T] that contains
a yield statement is actually a generator function, not a plain function that
returns an iterator. The correct annotations are:

  sync  yield  ->  Generator[YieldType, SendType, ReturnType]
  async yield  ->  AsyncGenerator[YieldType, SendType]

The previous async fix (Iterator -> AsyncGenerator) was applied inconsistently:
the same issue existed on all sync counterparts and was left behind. This commit
fixes all remaining sites for full consistency:

base_client.py:
  uipath_stream       Iterator[str | bytes]                  -> Generator[str | bytes, None, None]
  _stream             Iterator[ChatGenerationChunk]          -> Generator[ChatGenerationChunk, None, None]
  _uipath_stream      Iterator[ChatGenerationChunk]          -> Generator[ChatGenerationChunk, None, None]

clients/normalized/chat_models.py:
  _uipath_stream      Iterator[ChatGenerationChunk]          -> Generator[ChatGenerationChunk, None, None]

clients/bedrock/utils.py:
  _stream_generator   Iterator[dict[str, Any]]               -> Generator[dict[str, Any], None, None]

Added Generator to collections.abc imports in all three files.

Author: cosminacho

packages/uipath_langchain_client/src/uipath_langchain_client/base_client.py

@@ -25,7 +25,7 @@
 
 import logging
 from abc import ABC
-from collections.abc import AsyncGenerator, AsyncIterator, Iterator, Mapping, Sequence
+from collections.abc import AsyncGenerator, AsyncIterator, Generator, Iterator, Mapping, Sequence
 from functools import cached_property
 from typing import Any, Literal
 
@@ -239,7 +239,7 @@ def uipath_stream(
         stream_type: Literal["text", "bytes", "lines", "raw"] = "lines",
         raise_status_error: bool = False,
         **kwargs: Any,
-    ) -> Iterator[str | bytes]:
+    ) -> Generator[str | bytes, None, None]:
         """Make a synchronous streaming HTTP request to the UiPath API.
 
         Args:
@@ -393,7 +393,7 @@ def _stream(
         stop: list[str] | None = None,
         run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
-    ) -> Iterator[ChatGenerationChunk]:
+    ) -> Generator[ChatGenerationChunk, None, None]:
         set_captured_response_headers({})
         try:
             first = True
@@ -413,7 +413,7 @@ def _uipath_stream(
         stop: list[str] | None = None,
         run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
-    ) -> Iterator[ChatGenerationChunk]:
+    ) -> Generator[ChatGenerationChunk, None, None]:
         """Override in subclasses to provide the core (non-wrapped) stream logic."""
         yield from super()._stream(messages, stop=stop, run_manager=run_manager, **kwargs)
 

packages/uipath_langchain_client/src/uipath_langchain_client/clients/bedrock/utils.py

@@ -1,6 +1,7 @@
 import base64
 import json
-from typing import Any, Iterator
+from typing import Any
+from collections.abc import Generator, Iterator
 
 from httpx import Client
 
@@ -50,7 +51,7 @@ def __init__(self, httpx_client: Client | None = None, region_name: str = "PLACE
         self.httpx_client = httpx_client
         self.meta = _MockClientMeta(region_name=region_name)
 
-    def _stream_generator(self, request_body: dict[str, Any]) -> Iterator[dict[str, Any]]:
+    def _stream_generator(self, request_body: dict[str, Any]) -> Generator[dict[str, Any], None, None]:
         if self.httpx_client is None:
             raise ValueError("httpx_client is not set")
         with self.httpx_client.stream("POST", "/", json=_serialize_bytes(request_body)) as response:

packages/uipath_langchain_client/src/uipath_langchain_client/clients/normalized/chat_models.py

@@ -24,7 +24,7 @@
 """
 
 import json
-from collections.abc import AsyncGenerator, AsyncIterator, Callable, Iterator, Sequence
+from collections.abc import AsyncGenerator, Callable, Generator, Iterator, Sequence
 from typing import Any
 
 from langchain_core.callbacks import (
@@ -392,15 +392,15 @@ def _uipath_stream(
         stop: list[str] | None = None,
         run_manager: CallbackManagerForLLMRun | None = None,
         **kwargs: Any,
-    ) -> Iterator[ChatGenerationChunk]:
+    ) -> Generator[ChatGenerationChunk, None, None]:
         request_body = self._preprocess_request(messages, stop=stop, **kwargs)
         request_body["stream"] = True
         for chunk in self.uipath_stream(
             request_body=request_body, stream_type="lines", raise_status_error=True
         ):
             chunk = str(chunk)
             if chunk.startswith("data:"):
-                chunk = chunk[len("data:"):].strip()
+                chunk = chunk[len("data:") :].strip()
             try:
                 json_data = json.loads(chunk)
             except json.JSONDecodeError:
@@ -423,7 +423,7 @@ async def _uipath_astream(
         ):
             chunk = str(chunk)
             if chunk.startswith("data:"):
-                chunk = chunk[len("data:"):].strip()
+                chunk = chunk[len("data:") :].strip()
             try:
                 json_data = json.loads(chunk)
             except json.JSONDecodeError: