uipath-langchain-client 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

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/clients/google/chat_models.py

@@ -1,8 +1,7 @@
-from collections.abc import AsyncIterator, Iterator
 from typing import Self
 
-from httpx import Response
-from pydantic import Field, SecretStr, model_validator
+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
 
@@ -18,135 +17,6 @@ except ImportError as e:
     ) 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",
@@ -159,34 +29,21 @@ class UiPathChatGoogleGenerativeAI(UiPathBaseLLMClient, ChatGoogleGenerativeAI):
 
     # 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"))
+    project: str | None = "PLACEHOLDER"
+    location: str | None = "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")
+        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")
+        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.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,

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.dist-info → uipath_langchain_client-1.0.2.dist-info}/METADATA

@@ -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.dist-info → uipath_langchain_client-1.0.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 uipath_langchain_client/__init__.py,sha256=3rOLh-mBT_DWUNShlnOzzcU9OxT98V7i_fQPcFRAGuQ,1605
-uipath_langchain_client/__version__.py,sha256=R_aA1X_hA-yDDjWv0aiM-6RBV3TdDB-fVpRK9_sckes,154
+uipath_langchain_client/__version__.py,sha256=4nhzfK4r7krVivStesBcwAOfNmRJXk2QNvz_mT5-iMY,154
 uipath_langchain_client/base_client.py,sha256=ZOv51sfGpJLS0fQq0kUIy0v0BA_WF6jEFky9EgO7fpQ,10911
 uipath_langchain_client/factory.py,sha256=6KKAQqtdtDS8cwsZ5MUoChiqnDye41svbZ5CeUaggYg,8312
 uipath_langchain_client/settings.py,sha256=mcSXWzZ_jkG_y-1XJOJpDZH9YVHZd6MMbNMMHXujGjw,920
@@ -13,7 +13,7 @@ uipath_langchain_client/clients/bedrock/chat_models.py,sha256=F1LuKnwWjiZSeV1Zbs
 uipath_langchain_client/clients/bedrock/embeddings.py,sha256=rLv2bWG46i3CJDfd7X9dB3acASQwTNfcxksZ7TRQjtg,1141
 uipath_langchain_client/clients/bedrock/utils.py,sha256=YYWyerreQj4X2iR-Lsg1i7ELLUi9HLk63AXuzuHp4KM,3129
 uipath_langchain_client/clients/google/__init__.py,sha256=oWzq3pprWR-si-mZt40VrQ5f53f-O7AKSWzqP60VDHE,271
-uipath_langchain_client/clients/google/chat_models.py,sha256=OinMNzBZkmNWUcIqX1wu2EyjZmrbyCyss5H_ijKBJg0,7668
+uipath_langchain_client/clients/google/chat_models.py,sha256=6KQufo3mt1lDW3o4iR1OMYhG1uZKoAOmCkyvTWsmxWg,2377
 uipath_langchain_client/clients/google/embeddings.py,sha256=jx2TmzvcxIcKA0iitm5-iqSIj0hqy5RmLnNXMXbkXns,1705
 uipath_langchain_client/clients/normalized/__init__.py,sha256=r0tJ3QvhZPcbNjZJz8ADVCVTcyV1nb0zluwhb-tAAMg,257
 uipath_langchain_client/clients/normalized/chat_models.py,sha256=kIs8kyW5OKWiNWnxn7o8hz6W-yizaFVvd3V7KZAC8Sw,16639
@@ -22,7 +22,7 @@ uipath_langchain_client/clients/openai/__init__.py,sha256=f6O7otwFkSc5CxVOHpZ1GH
 uipath_langchain_client/clients/openai/chat_models.py,sha256=fAaDeR5bQ6FFgufP0Qq8Hs3f-1QKHvLNvghGw18pGzQ,4029
 uipath_langchain_client/clients/openai/embeddings.py,sha256=3saz1Do0rylJBVUWLcF8h651VaIED-ivUiiUnLPLJmE,3310
 uipath_langchain_client/clients/vertexai/__init__.py,sha256=g_rPq14ij0DPxUC4tX7vNDB5KVvfULxOUXyReM97PZw,132
-uipath_langchain_client/clients/vertexai/chat_models.py,sha256=K1yG_uXnjryFz6tg4UZepNnRVB6IQd6zhdDo4Sp9ubQ,1880
-uipath_langchain_client-1.0.1.dist-info/METADATA,sha256=nWG-sEQslliVc4lmHROCtdEE3EuMlRB0dw1eXo3UB5k,8762
-uipath_langchain_client-1.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-uipath_langchain_client-1.0.1.dist-info/RECORD,,
+uipath_langchain_client/clients/vertexai/chat_models.py,sha256=BpLGbDZ6YP6mOgxXZp8lXf9tId-JAuYhEFUcrqFejHE,2022
+uipath_langchain_client-1.0.2.dist-info/METADATA,sha256=CiEtjIYvU7UI_4aYFn4c2MUlgg4lpX4Wi-e5xefgVEs,8762
+uipath_langchain_client-1.0.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+uipath_langchain_client-1.0.2.dist-info/RECORD,,