uipath-langchain 0.1.24__py3-none-any.whl → 0.1.28__py3-none-any.whl

uipath_langchain/chat/vertex.py

@@ -0,0 +1,255 @@
+import os
+from typing import Any, Optional
+
+import httpx
+from uipath._utils._ssl_context import get_httpx_client_kwargs
+from uipath.utils import EndpointManager
+
+from .supported_models import GeminiModels
+
+
+def _check_genai_dependencies() -> None:
+    """Check if required dependencies for UiPathChatVertex are installed."""
+    import importlib.util
+
+    missing_packages = []
+
+    if importlib.util.find_spec("langchain_google_genai") is None:
+        missing_packages.append("langchain-google-genai")
+
+    if importlib.util.find_spec("google.genai") is None:
+        missing_packages.append("google-genai")
+
+    if missing_packages:
+        packages_str = ", ".join(missing_packages)
+        raise ImportError(
+            f"The following packages are required to use UiPathChatVertex: {packages_str}\n"
+            "Please install them using one of the following methods:\n\n"
+            "  # Using pip:\n"
+            f"  pip install uipath-langchain[vertex]\n\n"
+            "  # Using uv:\n"
+            f"  uv add 'uipath-langchain[vertex]'\n\n"
+        )
+
+
+_check_genai_dependencies()
+
+import google.genai
+from google.genai import types as genai_types
+from langchain_google_genai import ChatGoogleGenerativeAI
+from pydantic import PrivateAttr
+
+
+def _rewrite_request_for_gateway(
+    request: httpx.Request, gateway_url: str
+) -> httpx.Request:
+    """Rewrite a request to redirect to the UiPath gateway."""
+    url_str = str(request.url)
+    if "generateContent" in url_str or "streamGenerateContent" in url_str:
+        is_streaming = "alt=sse" in url_str
+
+        headers = dict(request.headers)
+
+        headers["X-UiPath-Streaming-Enabled"] = "true" if is_streaming else "false"
+
+        gateway_url_parsed = httpx.URL(gateway_url)
+        if gateway_url_parsed.host:
+            headers["host"] = gateway_url_parsed.host
+
+        return httpx.Request(
+            method=request.method,
+            url=gateway_url,
+            headers=headers,
+            content=request.content,
+            extensions=request.extensions,
+        )
+    return request
+
+
+class _UrlRewriteTransport(httpx.BaseTransport):
+    """Transport that rewrites URLs to redirect to UiPath gateway."""
+
+    def __init__(self, gateway_url: str):
+        self.gateway_url = gateway_url
+        self._transport = httpx.HTTPTransport()
+
+    def handle_request(self, request: httpx.Request) -> httpx.Response:
+        request = _rewrite_request_for_gateway(request, self.gateway_url)
+        return self._transport.handle_request(request)
+
+    def close(self) -> None:
+        self._transport.close()
+
+
+class _AsyncUrlRewriteTransport(httpx.AsyncBaseTransport):
+    """Async transport that rewrites URLs to redirect to UiPath gateway."""
+
+    def __init__(self, gateway_url: str):
+        self.gateway_url = gateway_url
+        self._transport = httpx.AsyncHTTPTransport()
+
+    async def handle_async_request(self, request: httpx.Request) -> httpx.Response:
+        request = _rewrite_request_for_gateway(request, self.gateway_url)
+        return await self._transport.handle_async_request(request)
+
+    async def aclose(self) -> None:
+        await self._transport.aclose()
+
+
+class UiPathChatVertex(ChatGoogleGenerativeAI):
+    """UiPath Vertex AI Chat model that routes requests through UiPath's LLM Gateway."""
+
+    _vendor: str = PrivateAttr(default="vertexai")
+    _model_name: str = PrivateAttr()
+    _uipath_token: str = PrivateAttr()
+    _uipath_llmgw_url: Optional[str] = PrivateAttr(default=None)
+
+    def __init__(
+        self,
+        org_id: Optional[str] = None,
+        tenant_id: Optional[str] = None,
+        token: Optional[str] = None,
+        model_name: str = GeminiModels.gemini_2_5_flash,
+        temperature: Optional[float] = None,
+        **kwargs: Any,
+    ):
+        org_id = org_id or os.getenv("UIPATH_ORGANIZATION_ID")
+        tenant_id = tenant_id or os.getenv("UIPATH_TENANT_ID")
+        token = token or os.getenv("UIPATH_ACCESS_TOKEN")
+
+        if not org_id:
+            raise ValueError(
+                "UIPATH_ORGANIZATION_ID environment variable or org_id parameter is required"
+            )
+        if not tenant_id:
+            raise ValueError(
+                "UIPATH_TENANT_ID environment variable or tenant_id parameter is required"
+            )
+        if not token:
+            raise ValueError(
+                "UIPATH_ACCESS_TOKEN environment variable or token parameter is required"
+            )
+
+        uipath_url = self._build_base_url(model_name)
+        headers = self._build_headers(token)
+
+        http_options = genai_types.HttpOptions(
+            httpx_client=httpx.Client(
+                transport=_UrlRewriteTransport(uipath_url),
+                headers=headers,
+                **get_httpx_client_kwargs(),
+            ),
+            httpx_async_client=httpx.AsyncClient(
+                transport=_AsyncUrlRewriteTransport(uipath_url),
+                headers=headers,
+                **get_httpx_client_kwargs(),
+            ),
+        )
+
+        if temperature is None and (
+            "gemini-3" in model_name or "gemini-2" in model_name
+        ):
+            temperature = 1.0
+
+        super().__init__(
+            model=model_name,
+            google_api_key="uipath-gateway",
+            temperature=temperature,
+            **kwargs,
+        )
+
+        custom_client = google.genai.Client(
+            api_key="uipath-gateway",
+            http_options=http_options,
+        )
+
+        object.__setattr__(self, "client", custom_client)
+
+        self._model_name = model_name
+        self._uipath_token = token
+        self._uipath_llmgw_url = uipath_url
+
+        if self.temperature is not None and not 0 <= self.temperature <= 2.0:
+            raise ValueError("temperature must be in the range [0.0, 2.0]")
+
+        if self.top_p is not None and not 0 <= self.top_p <= 1:
+            raise ValueError("top_p must be in the range [0.0, 1.0]")
+
+        if self.top_k is not None and self.top_k <= 0:
+            raise ValueError("top_k must be positive")
+
+        additional_headers = self.additional_headers or {}
+        self.default_metadata = tuple(additional_headers.items())
+
+    @staticmethod
+    def _build_headers(token: str) -> dict[str, str]:
+        """Build HTTP headers for UiPath Gateway requests."""
+        headers = {
+            "Authorization": f"Bearer {token}",
+        }
+        if job_key := os.getenv("UIPATH_JOB_KEY"):
+            headers["X-UiPath-JobKey"] = job_key
+        if process_key := os.getenv("UIPATH_PROCESS_KEY"):
+            headers["X-UiPath-ProcessKey"] = process_key
+        return headers
+
+    @staticmethod
+    def _build_base_url(model_name: str) -> str:
+        """Build the full URL for the UiPath LLM Gateway."""
+        env_uipath_url = os.getenv("UIPATH_URL")
+
+        if not env_uipath_url:
+            raise ValueError("UIPATH_URL environment variable is required")
+
+        vendor_endpoint = EndpointManager.get_vendor_endpoint()
+        formatted_endpoint = vendor_endpoint.format(
+            vendor="vertexai",
+            model=model_name,
+        )
+        return f"{env_uipath_url.rstrip('/')}/{formatted_endpoint}"
+
+    def _stream(self, messages, stop=None, run_manager=None, **kwargs):
+        """Streaming fallback - calls _generate and yields single response."""
+        from langchain_core.messages import AIMessageChunk
+        from langchain_core.outputs import ChatGenerationChunk
+
+        result = self._generate(messages, stop=stop, run_manager=run_manager, **kwargs)
+
+        if result.generations:
+            message = result.generations[0].message
+            chunk = AIMessageChunk(
+                content=message.content,
+                additional_kwargs=message.additional_kwargs,
+                response_metadata=getattr(message, "response_metadata", {}),
+                id=message.id,
+                tool_calls=getattr(message, "tool_calls", []),
+                tool_call_chunks=getattr(message, "tool_call_chunks", []),
+            )
+            if hasattr(message, "usage_metadata") and message.usage_metadata:
+                chunk.usage_metadata = message.usage_metadata
+
+            yield ChatGenerationChunk(message=chunk)
+
+    async def _astream(self, messages, stop=None, run_manager=None, **kwargs):
+        """Async streaming fallback - calls _agenerate and yields single response."""
+        from langchain_core.messages import AIMessageChunk
+        from langchain_core.outputs import ChatGenerationChunk
+
+        result = await self._agenerate(
+            messages, stop=stop, run_manager=run_manager, **kwargs
+        )
+
+        if result.generations:
+            message = result.generations[0].message
+            chunk = AIMessageChunk(
+                content=message.content,
+                additional_kwargs=message.additional_kwargs,
+                response_metadata=getattr(message, "response_metadata", {}),
+                id=message.id,
+                tool_calls=getattr(message, "tool_calls", []),
+                tool_call_chunks=getattr(message, "tool_call_chunks", []),
+            )
+            if hasattr(message, "usage_metadata") and message.usage_metadata:
+                chunk.usage_metadata = message.usage_metadata
+
+            yield ChatGenerationChunk(message=chunk)

uipath_langchain/runtime/schema.py

@@ -7,6 +7,8 @@ from langchain_core.runnables.base import Runnable
 from langchain_core.runnables.graph import Graph, Node
 from langgraph.graph.state import CompiledStateGraph
 from langgraph.prebuilt import ToolNode
+from langgraph.pregel._read import PregelNode
+from langgraph.pregel._write import ChannelWrite, ChannelWriteEntry
 from uipath.runtime.schema import (
     UiPathRuntimeEdge,
     UiPathRuntimeGraph,
@@ -193,7 +195,6 @@ def get_graph_schema(
     nodes: list[UiPathRuntimeNode] = []
     for node_id, node in graph.nodes.items():
         subgraph: UiPathRuntimeGraph | None = subgraphs_dict.get(node_id)
-
         nodes.append(
             UiPathRuntimeNode(
                 id=node.id,
@@ -204,15 +205,51 @@ def get_graph_schema(
             )
         )
 
+    # Use a set to track unique edges (source, target)
+    seen_edges: set[tuple[str, str]] = set()
     edges: list[UiPathRuntimeEdge] = []
+
+    # First, add edges from graph.edges (static edges)
     for edge in graph.edges:
-        edges.append(
-            UiPathRuntimeEdge(
-                source=edge.source,
-                target=edge.target,
-                label=getattr(edge, "data", None) or getattr(edge, "label", None),
+        edge_tuple = (edge.source, edge.target)
+        if edge_tuple not in seen_edges:
+            seen_edges.add(edge_tuple)
+            edges.append(
+                UiPathRuntimeEdge(
+                    source=edge.source,
+                    target=edge.target,
+                    label=getattr(edge, "data", None) or getattr(edge, "label", None),
+                )
             )
-        )
+
+    # Build a map of channel -> target node
+    channel_to_target: dict[str, str] = {}
+    node_spec: PregelNode
+    for node_name, node_spec in compiled_graph.nodes.items():
+        for trigger in node_spec.triggers:
+            if isinstance(trigger, str) and trigger.startswith("branch:to:"):
+                channel_to_target[trigger] = node_name
+
+    # Extract edges by looking at what each node writes to (dynamic edges from Command)
+    for source_name, node_spec in compiled_graph.nodes.items():
+        writer: Runnable[Any, Any]
+        for writer in node_spec.writers:
+            if isinstance(writer, ChannelWrite):
+                write_entry: Any
+                for write_entry in writer.writes:
+                    if isinstance(write_entry, ChannelWriteEntry) and isinstance(
+                        write_entry.channel, str
+                    ):
+                        target = channel_to_target.get(write_entry.channel)
+                        if target:
+                            edge_tuple = (source_name, target)
+                            if edge_tuple not in seen_edges:
+                                seen_edges.add(edge_tuple)
+                                edges.append(
+                                    UiPathRuntimeEdge(
+                                        source=source_name, target=target, label=None
+                                    )
+                                )
 
     return UiPathRuntimeGraph(nodes=nodes, edges=edges)
 

{uipath_langchain-0.1.24.dist-info → uipath_langchain-0.1.28.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.1.24
-Summary: UiPath Langchain
+Version: 0.1.28
+Summary: Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
 Project-URL: Documentation, https://uipath.github.io/uipath-python/
@@ -13,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.11
+Requires-Dist: aiosqlite==0.21.0
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: jsonpath-ng>=1.7.0
 Requires-Dist: jsonschema-pydantic-converter>=0.1.5
@@ -24,18 +25,18 @@ Requires-Dist: langgraph<2.0.0,>=1.0.0
 Requires-Dist: openinference-instrumentation-langchain>=0.1.56
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.3.0,>=2.2.28
+Requires-Dist: uipath<2.3.0,>=2.2.30
 Provides-Extra: bedrock
 Requires-Dist: boto3-stubs>=1.41.4; extra == 'bedrock'
 Requires-Dist: langchain-aws>=0.2.35; extra == 'bedrock'
 Provides-Extra: vertex
-Requires-Dist: langchain-community>=0.4.1; extra == 'vertex'
-Requires-Dist: langchain-google-vertexai>=2.1.2; extra == 'vertex'
-Requires-Dist: types-protobuf>=6.32.1.20251105; extra == 'vertex'
+Requires-Dist: google-generativeai>=0.8.0; extra == 'vertex'
+Requires-Dist: langchain-google-genai>=2.0.0; extra == 'vertex'
 Description-Content-Type: text/markdown
 
 # UiPath LangChain Python SDK
 
+[![PyPI - Version](https://img.shields.io/pypi/v/uipath-langchain)](https://pypi.org/project/uipath-langchain/)
 [![PyPI downloads](https://img.shields.io/pypi/dm/uipath-langchain.svg)](https://pypi.org/project/uipath-langchain/)
 [![Python versions](https://img.shields.io/pypi/pyversions/uipath-langchain.svg)](https://pypi.org/project/uipath-langchain/)
 

{uipath_langchain-0.1.24.dist-info → uipath_langchain-0.1.28.dist-info}/RECORD

@@ -17,17 +17,17 @@ uipath_langchain/_utils/_sleep_policy.py,sha256=e9pHdjmcCj4CVoFM1jMyZFelH11Yatsg
 uipath_langchain/agent/exceptions/__init__.py,sha256=GFh8mtsoKYghTHe93rRbIycWOW8dPt0okQjZuc2N-Hc,158
 uipath_langchain/agent/exceptions/exceptions.py,sha256=G7LdeQ9uMljLGTlixWBHvNJ1ULtUIDIkzApjfbsF6_0,216
 uipath_langchain/agent/guardrails/__init__.py,sha256=gWxPFhbxqiBM6phdYeuR2EMVb-qwB71Ny16SzNAQ5kw,610
-uipath_langchain/agent/guardrails/guardrail_nodes.py,sha256=Wkxb1hnxunjQHalPRQHOkGWIh4HPjFHPzYCByQyNUgE,3772
+uipath_langchain/agent/guardrails/guardrail_nodes.py,sha256=qXOnOfXsMY8hHoHzejjQfppuLTk8OsC1vyL98uNodJw,5467
 uipath_langchain/agent/guardrails/guardrails_factory.py,sha256=gkOkNO5uyhNBt1y7j4V3Ao-u1fh1gpTji3-N2bRS_l4,2248
-uipath_langchain/agent/guardrails/guardrails_subgraph.py,sha256=nBOm2U07zPxoE699Iqe4YZECo43C3BQ4uXbj55oUUCE,8739
+uipath_langchain/agent/guardrails/guardrails_subgraph.py,sha256=o38vdDS15LJmXY6FZ0wVHFBHQPm-zFRViG40Xoq9H70,9817
 uipath_langchain/agent/guardrails/types.py,sha256=Z5TS6EHnaanie3s1yfGhE4JXoFKuu01ESsbeFbbPie4,561
 uipath_langchain/agent/guardrails/actions/__init__.py,sha256=PiE5Eb39JYWrQG481aREV2I0tNMPEi7VtChBy7ZJqcc,253
-uipath_langchain/agent/guardrails/actions/base_action.py,sha256=7Q8ArFLYkbTVsKinCDoU1TAI-ki0JLwTc3XknE8dxgA,673
-uipath_langchain/agent/guardrails/actions/block_action.py,sha256=5qBlwqxJdIsTvJ2h5m0N2_p9cIAcWeWas30GsxiBte4,1330
-uipath_langchain/agent/guardrails/actions/escalate_action.py,sha256=O2IY0FiVBQyfiNszzR2JgP1dRJC7hb03eSUIqpW9BFo,9491
-uipath_langchain/agent/guardrails/actions/log_action.py,sha256=ADvP0Sbx3oRbgI5KQfXyZTBUB0CT40QAqY4q8KzswVw,1924
+uipath_langchain/agent/guardrails/actions/base_action.py,sha256=89POG1KQ0x1KPXKP8jT42M04SYoizp5E_co2-Ykq8gc,710
+uipath_langchain/agent/guardrails/actions/block_action.py,sha256=RZPF6xxFqs4G2OByBrE7V5nBdPIsb3AyMl-r-lf95yE,1367
+uipath_langchain/agent/guardrails/actions/escalate_action.py,sha256=BM04L7b0pVG5UOVi9ZA6U2TOMBf_UtpKl278i4TAMHk,18569
+uipath_langchain/agent/guardrails/actions/log_action.py,sha256=mA9qIMErZ7tnkDwGKGY9Q7gh55qiL5C6kn1-ljOGGtU,1961
 uipath_langchain/agent/react/__init__.py,sha256=BhRWUMZ9yLtBr7SyRwS-XMKG9o84DROyedCMpE6j1RU,362
-uipath_langchain/agent/react/agent.py,sha256=A5edIf7C43wCZZzx3FSQoAFION7RBHaz5c4KE9iadSU,3788
+uipath_langchain/agent/react/agent.py,sha256=w--Mu9ZD9P-f4yXeZHTtAwtcDDgRhxCI28pbs215aDw,4002
 uipath_langchain/agent/react/constants.py,sha256=B2yqryh242DETslaRYacUPbVdpjvvApjsBira_qhQwk,61
 uipath_langchain/agent/react/init_node.py,sha256=plhyTecBOBsm-YHfi73ZN3pJhe24tpCMr53IfRzN6SA,566
 uipath_langchain/agent/react/llm_node.py,sha256=4q8JcAwzELyyBaYTaFtx5HP9neq_5UolfFsuqsh4pso,1468
@@ -49,11 +49,11 @@ uipath_langchain/agent/tools/tool_node.py,sha256=TnXsjoShvhsoBuV5RoUVoJCc2zYPKSn
 uipath_langchain/agent/tools/utils.py,sha256=DsFeZ7kDzFaZ0bGHQN6TlGMJ90wYr7P1Vo1rpHPHWws,401
 uipath_langchain/chat/__init__.py,sha256=fAoK-hKuShWDV9tbinJU9TSKRBG2wQQwK3UKZTAyMVM,254
 uipath_langchain/chat/bedrock.py,sha256=eonXWnGh3m9cmMifP49fnXsjmrRQLd3Wda5fFa11568,5863
-uipath_langchain/chat/gemini.py,sha256=Zlz5KVoHdzlemzFTmFVjKCloaFo2lMiBbBq9MCJ9Lds,11903
 uipath_langchain/chat/mapper.py,sha256=XsWFbg6U4kW5Yj_ANvYUZ4HICybHcv9qWdfPPU8UTKA,11950
 uipath_langchain/chat/models.py,sha256=cbIRw-YhEvQcgt0DVTHc84lKIjeDhsQo5oxvBz9luD0,18168
-uipath_langchain/chat/openai.py,sha256=M36pSDeWwnfO5EXcCsYnjfZ5C5jfko6ng_87qS-Z5tA,4594
+uipath_langchain/chat/openai.py,sha256=iZWRFbFGPJo0rgjIL1uGytGgvlYHzYQvuoIgyltOz98,4689
 uipath_langchain/chat/supported_models.py,sha256=KrTO8rex8sjC2NVUnuDoLLvz__ROKtxEtCXSk1xXTOU,1373
+uipath_langchain/chat/vertex.py,sha256=y68TrwJwWewkSNdv_smP9JzCQ9hvJw5o6tYVLV-_9wE,9261
 uipath_langchain/embeddings/__init__.py,sha256=QICtYB58ZyqFfDQrEaO8lTEgAU5NuEKlR7iIrS0OBtc,156
 uipath_langchain/embeddings/embeddings.py,sha256=MGUtnT0pkj97IjHKhROSd786g3ONwwkjgfQBSq3ayhc,6888
 uipath_langchain/retrievers/__init__.py,sha256=rOn7PyyHgZ4pMnXWPkGqmuBmx8eGuo-Oyndo7Wm9IUU,108
@@ -65,12 +65,12 @@ uipath_langchain/runtime/errors.py,sha256=iepiILaDtanusvyYBQSRsPDmKB-6GB94OXKATw
 uipath_langchain/runtime/factory.py,sha256=_fbYViNg6uuApUZiEJjYHQU2noH5dqFBTXWEPy-Oj_k,11043
 uipath_langchain/runtime/graph.py,sha256=1h_eaXkyN0G43BS0WUMoK_UNDao8jrmcyJqGVgwVEjM,5513
 uipath_langchain/runtime/runtime.py,sha256=cK7hewu6Mv8rQbUir03JgPUgv-zqVtYQ1YOfoMLB89I,16904
-uipath_langchain/runtime/schema.py,sha256=sbgf4Z_2T_r2sv1e8uYsAExZ1PNeVCODB9Hm4HZ4gQc,11547
+uipath_langchain/runtime/schema.py,sha256=mzMnxKtRvREGEzmLyAYX3sSYLJTz6xIZe63u6Z6dPNs,13392
 uipath_langchain/runtime/storage.py,sha256=Vem6kz0yGIUgIhTByzguvNOkb6N40b_ba8jcbjGo_eY,3954
 uipath_langchain/vectorstores/__init__.py,sha256=w8qs1P548ud1aIcVA_QhBgf_jZDrRMK5Lono78yA8cs,114
 uipath_langchain/vectorstores/context_grounding_vectorstore.py,sha256=SfRvIenOzn4r7jng_I9GIyknyeDx2hdW63O1rZY-ZLQ,8407
-uipath_langchain-0.1.24.dist-info/METADATA,sha256=STmAiU3rEotgAApOuXj8bk4p3V6jOJcw3JVo7zEAOrA,5658
-uipath_langchain-0.1.24.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-uipath_langchain-0.1.24.dist-info/entry_points.txt,sha256=Wptt1FbvhcDzNQxDlu2Lt1ngBOdDIFPlM1p8WzXf6wQ,171
-uipath_langchain-0.1.24.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
-uipath_langchain-0.1.24.dist-info/RECORD,,
+uipath_langchain-0.1.28.dist-info/METADATA,sha256=rpoHmCK1gULm42RORNwYclg1fapZEA9M6_AbqDhSizo,5818
+uipath_langchain-0.1.28.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+uipath_langchain-0.1.28.dist-info/entry_points.txt,sha256=Wptt1FbvhcDzNQxDlu2Lt1ngBOdDIFPlM1p8WzXf6wQ,171
+uipath_langchain-0.1.28.dist-info/licenses/LICENSE,sha256=JDpt-uotAkHFmxpwxi6gwx6HQ25e-lG4U_Gzcvgp7JY,1063
+uipath_langchain-0.1.28.dist-info/RECORD,,