uipath-langchain-client 1.0.1__tar.gz → 1.0.2__tar.gz

{uipath_langchain_client-1.0.1 → uipath_langchain_client-1.0.2}/CHANGELOG.md

@@ -2,6 +2,11 @@
 
 All notable changes to `uipath_langchain_client` will be documented in this file.
 
+## [1.0.2] - 2026-02-02
+
+### Buf Fix
+- Removed old fix on Gemini streaming and updated with a new cleaner one
+
 ## [1.0.1] - 2026-02-02
 
 ### Bug Fix

{uipath_langchain_client-1.0.1 → uipath_langchain_client-1.0.2}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: uipath-langchain-client
-Version: 1.0.1
+Version: 1.0.2
 Summary: LangChain-compatible chat models and embeddings for UiPath's LLM services
 Requires-Python: >=3.11
 Requires-Dist: langchain>=1.2.7
-Requires-Dist: uipath-llm-client>=1.0.0
+Requires-Dist: uipath-llm-client>=1.0.2
 Provides-Extra: all
 Requires-Dist: langchain-anthropic>=1.3.1; extra == 'all'
 Requires-Dist: langchain-aws>=1.2.1; extra == 'all'

{uipath_langchain_client-1.0.1 → uipath_langchain_client-1.0.2}/pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 requires-python = ">=3.11"
 dependencies = [
     "langchain>=1.2.7",
-    "uipath-llm-client>=1.0.0",
+    "uipath-llm-client>=1.0.2",
 ]
 
 [project.optional-dependencies]

{uipath_langchain_client-1.0.1 → uipath_langchain_client-1.0.2}/src/uipath_langchain_client/__version__.py

@@ -1,3 +1,3 @@
 __title__ = "UiPath LangChain Client"
 __description__ = "A Python client for interacting with UiPath's LLM services via LangChain."
-__version__ = "1.0.1"
+__version__ = "1.0.2"

uipath_langchain_client-1.0.2/src/uipath_langchain_client/clients/google/chat_models.py

@@ -0,0 +1,60 @@
+from typing import Self
+
+from httpx import URL, Request
+from pydantic import Field, model_validator
+from uipath_langchain_client.base_client import UiPathBaseLLMClient
+from uipath_langchain_client.settings import UiPathAPIConfig
+
+try:
+    from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
+
+    from google.genai.client import Client
+    from google.genai.types import HttpOptions
+except ImportError as e:
+    raise ImportError(
+        "The 'google' extra is required to use UiPathChatGoogleGenerativeAI. "
+        "Install it with: uv add uipath-langchain-client[google]"
+    ) from e
+
+
+class UiPathChatGoogleGenerativeAI(UiPathBaseLLMClient, ChatGoogleGenerativeAI):
+    api_config: UiPathAPIConfig = UiPathAPIConfig(
+        api_type="completions",
+        client_type="passthrough",
+        vendor_type="vertexai",
+        api_flavor="generate-content",
+        api_version="v1beta1",
+        freeze_base_url=True,
+    )
+
+    # Override fields to avoid errors when instantiating the class
+    model: str = Field(default="", alias="model_name")
+    project: str | None = "PLACEHOLDER"
+    location: str | None = "PLACEHOLDER"
+
+    @model_validator(mode="after")
+    def setup_uipath_client(self) -> Self:
+        def fix_url_for_streaming(request: Request):
+            if request.headers.get("X-UiPath-Streaming-Enabled") == "true":
+                request.url = URL(request.url).copy_add_param("alt", "sse")
+
+        async def fix_url_for_streaming_async(request: Request):
+            if request.headers.get("X-UiPath-Streaming-Enabled") == "true":
+                request.url = URL(request.url).copy_add_param("alt", "sse")
+
+        self.uipath_sync_client.event_hooks["request"].append(fix_url_for_streaming)
+        self.uipath_async_client.event_hooks["request"].append(fix_url_for_streaming_async)
+
+        self.client = Client(
+            vertexai=True,
+            api_key="PLACEHOLDER",
+            http_options=HttpOptions(
+                base_url=str(self.uipath_sync_client.base_url),
+                headers=dict(self.uipath_sync_client.headers),
+                timeout=None,  # handled by the UiPath client
+                retry_options=None,  # handled by the UiPath client
+                httpx_client=self.uipath_sync_client,
+                httpx_async_client=self.uipath_async_client,
+            ),
+        )
+        return self

{uipath_langchain_client-1.0.1 → uipath_langchain_client-1.0.2}/src/uipath_langchain_client/clients/vertexai/chat_models.py

@@ -23,6 +23,10 @@ class UiPathChatAnthropicVertex(UiPathBaseLLMClient, ChatAnthropicVertex):  # ty
         freeze_base_url=True,
     )
 
+    # Override fields to avoid errors when instantiating the class
+    project: str | None = "PLACEHOLDER"
+    location: str = "PLACEHOLDER"
+
     @model_validator(mode="after")
     def setup_uipath_client(self) -> Self:
         self.client = AnthropicVertex(

uipath_langchain_client-1.0.1/src/uipath_langchain_client/clients/google/chat_models.py

@@ -1,203 +0,0 @@
-from collections.abc import AsyncIterator, Iterator
-from typing import Self
-
-from httpx import Response
-from pydantic import Field, SecretStr, model_validator
-from uipath_langchain_client.base_client import UiPathBaseLLMClient
-from uipath_langchain_client.settings import UiPathAPIConfig
-
-try:
-    from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
-
-    from google.genai.client import Client
-    from google.genai.types import HttpOptions
-except ImportError as e:
-    raise ImportError(
-        "The 'google' extra is required to use UiPathChatGoogleGenerativeAI. "
-        "Install it with: uv add uipath-langchain-client[google]"
-    ) from e
-
-
-def _wrap_iter_lines(original: Iterator[str]) -> Iterator[str]:
-    """Wrap iter_lines to extract individual JSON objects from streaming responses.
-
-    The LLM Gateway wraps streaming JSON responses in an array like [{...}, {...}].
-    This extracts each complete JSON object and yields them individually.
-    Handles multiple JSON objects on a single line (e.g., {...},{...}).
-
-    We prefix output with 'data: ' so the SDK's _iter_response_stream bypasses its
-    broken brace counting (which doesn't handle braces inside strings) and yields
-    our JSON objects directly.
-
-    Temporal Fix until it's fixed in the main package.
-    """
-    buffer = ""
-    balance = 0
-    in_string = False
-    escape_next = False
-
-    for line in original:
-        # Handle data: prefix (SSE format)
-        if line.startswith("data:"):
-            line = line[5:].lstrip()
-
-        for char in line:
-            # Handle escape sequences in strings
-            if escape_next:
-                buffer += char
-                escape_next = False
-                continue
-
-            if char == "\\" and in_string:
-                buffer += char
-                escape_next = True
-                continue
-
-            if char == '"':
-                in_string = not in_string
-                buffer += char
-                continue
-
-            # Only track braces outside of strings
-            if not in_string:
-                if char == "{":
-                    balance += 1
-                    buffer += char
-                elif char == "}":
-                    buffer += char
-                    balance -= 1
-                    if balance == 0 and buffer:
-                        # Complete JSON object found - yield with 'data: ' prefix
-                        # so SDK bypasses its broken brace counting
-                        yield "data: " + buffer
-                        buffer = ""
-                elif balance == 0:
-                    # Skip characters outside JSON objects (array brackets, commas, whitespace)
-                    continue
-                else:
-                    buffer += char
-            else:
-                buffer += char
-
-    # Yield any remaining buffer (handles incomplete streams)
-    if buffer:
-        yield "data: " + buffer
-
-
-async def _wrap_aiter_lines(original: AsyncIterator[str]) -> AsyncIterator[str]:
-    """Async version of _wrap_iter_lines.
-
-    Extracts individual JSON objects from streaming responses.
-    Handles multiple JSON objects on a single line (e.g., {...},{...}).
-
-    We prefix output with 'data: ' so the SDK's _iter_response_stream bypasses its
-    broken brace counting (which doesn't handle braces inside strings) and yields
-    our JSON objects directly.
-    """
-    buffer = ""
-    balance = 0
-    in_string = False
-    escape_next = False
-
-    async for line in original:
-        # Handle data: prefix (SSE format)
-        if line.startswith("data:"):
-            line = line[5:].lstrip()
-
-        for char in line:
-            # Handle escape sequences in strings
-            if escape_next:
-                buffer += char
-                escape_next = False
-                continue
-
-            if char == "\\" and in_string:
-                buffer += char
-                escape_next = True
-                continue
-
-            if char == '"':
-                in_string = not in_string
-                buffer += char
-                continue
-
-            # Only track braces outside of strings
-            if not in_string:
-                if char == "{":
-                    balance += 1
-                    buffer += char
-                elif char == "}":
-                    buffer += char
-                    balance -= 1
-                    if balance == 0 and buffer:
-                        # Complete JSON object found - yield with 'data: ' prefix
-                        # so SDK bypasses its broken brace counting
-                        yield "data: " + buffer
-                        buffer = ""
-                elif balance == 0:
-                    # Skip characters outside JSON objects (array brackets, commas, whitespace)
-                    continue
-                else:
-                    buffer += char
-            else:
-                buffer += char
-
-    # Yield any remaining buffer (handles incomplete streams)
-    if buffer:
-        yield "data: " + buffer
-
-
-class UiPathChatGoogleGenerativeAI(UiPathBaseLLMClient, ChatGoogleGenerativeAI):
-    api_config: UiPathAPIConfig = UiPathAPIConfig(
-        api_type="completions",
-        client_type="passthrough",
-        vendor_type="vertexai",
-        api_flavor="generate-content",
-        api_version="v1beta1",
-        freeze_base_url=True,
-    )
-
-    # Override fields to avoid errors when instantiating the class
-    model: str = Field(default="", alias="model_name")
-    google_api_key: SecretStr | None = Field(default=SecretStr("PLACEHOLDER"))
-
-    @model_validator(mode="after")
-    def setup_uipath_client(self) -> Self:
-        def fix_streaming_response(response: Response):
-            """Monkey-patch iter_lines to strip JSON array brackets."""
-            original_iter_lines = response.iter_lines
-            response.iter_lines = lambda: _wrap_iter_lines(original_iter_lines())
-
-        async def fix_streaming_response_async(response: Response):
-            """Monkey-patch aiter_lines to strip JSON array brackets."""
-            original_aiter_lines = response.aiter_lines
-            response.aiter_lines = lambda: _wrap_aiter_lines(original_aiter_lines())
-
-        self.uipath_sync_client.event_hooks["response"].append(fix_streaming_response)
-        self.uipath_async_client.event_hooks["response"].append(fix_streaming_response_async)
-
-        # TODO: in exactly 2 weeks, we need to uncomment this part of the code because it will work, 5 february 2026 is the date.
-        # def fix_url_for_streaming(request: Request):
-        #     if request.headers.get("X-UiPath-Streaming-Enabled") == "true":
-        #         request.url = URL(request.url).copy_add_param("alt", "sse")
-
-        # async def fix_url_for_streaming_async(request: Request):
-        #     if request.headers.get("X-UiPath-Streaming-Enabled") == "true":
-        #         request.url = URL(request.url).copy_add_param("alt", "sse")
-
-        # self.uipath_sync_client.event_hooks["request"].append(fix_url_for_streaming)
-        # self.uipath_async_client.event_hooks["request"].append(fix_url_for_streaming_async)
-
-        self.client = Client(
-            vertexai=True,
-            api_key="PLACEHOLDER",
-            http_options=HttpOptions(
-                base_url=str(self.uipath_sync_client.base_url),
-                headers=dict(self.uipath_sync_client.headers),
-                timeout=None,  # handled by the UiPath client
-                retry_options=None,  # handled by the UiPath client
-                httpx_client=self.uipath_sync_client,
-                httpx_async_client=self.uipath_async_client,
-            ),
-        )
-        return self