uipath-langchain 0.0.115__tar.gz → 0.0.117__tar.gz

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.115
+Version: 0.0.117
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -24,7 +24,7 @@ Requires-Dist: langgraph>=0.2.70
 Requires-Dist: openai>=1.65.5
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.1.0,>=2.0.71
+Requires-Dist: uipath<2.1.0,>=2.0.79
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/docs/chat_models.md

@@ -1,6 +1,6 @@
 # Chat Models
 
-UiPath provides two chat models `UiPathAzureChatOpenAI` and `UiPathChat`. These are compatible with langgraph as drop in replacements. You do not need to add tokens from OpenAI or Anthropic, usage of these chat models will consume `AI Units` on your account.
+UiPath provides two chat models `UiPathAzureChatOpenAI` and `UiPathChat`. These are compatible with LangGraph as drop in replacements. You do not need to add tokens from OpenAI or Anthropic, usage of these chat models will consume `Agent Units` on your account.
 
 ## UiPathAzureChatOpenAI
 
@@ -86,7 +86,7 @@ Currently the following models can be used with `UiPathChat` (this list can be u
 -   `anthropic.claude-3-5-sonnet-20240620-v1:0`, `anthropic.claude-3-5-sonnet-20241022-v2:0`, `anthropic.claude-3-7-sonnet-20250219-v1:0`, `anthropic.claude-3-haiku-20240307-v1:0`, `gemini-1.5-pro-001`, `gemini-2.0-flash-001`, `gpt-4o-2024-05-13`, `gpt-4o-2024-08-06`, `gpt-4o-2024-11-20`, `gpt-4o-mini-2024-07-18`, `o3-mini-2025-01-31`
 
 /// warning
-Please note that that you may get errors related to data residency, as some models are not available on all regions.
+Please note that you may get errors related to data residency, as some models are not available on all regions.
 
 Example: `[Enforced Region] No model configuration found for product uipath-python-sdk in EU using model anthropic.claude-3-opus-20240229-v1:0`.
 

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/docs/human_in_the_loop.md

@@ -2,7 +2,7 @@
 
 Guide for **Human-In-The-Loop** scenarios within the UiPath-Langchain integration.
 It focuses on the **interrupt(model)** functionality, illustrating its role as a symbolic representation of an agent's
-wait state within the Langchain framework.
+wait state within the LangGraph framework.
 
 ## Models Overview
 
@@ -49,7 +49,7 @@ action_output = interrupt(WaitAction(action=my_action_instance))
 
 ---
 
-> 💡 UiPath Langchain sdk also supports **Robot/Agent-in-the-loop** scenarios. In this context, the execution of one agent
+> 💡The UiPath-LangChain SDK also supports **Robot/Agent-in-the-loop** scenarios. In this context, the execution of one agent
 > can be suspended until another robot or agent finishes its execution.
 
 ### 3. InvokeProcess

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.115"
+version = "0.0.117"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.71, <2.1.0",
+    "uipath>=2.0.79, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",
@@ -49,6 +49,7 @@ dev = [
     "pytest>=7.4.0",
     "pytest-cov>=4.1.0",
     "pytest-mock>=3.11.1",
+    "pytest-asyncio>=1.0.0",
     "pre-commit>=4.1.0",
     "numpy>=1.24.0",
     "pytest_httpx>=0.35.0"
@@ -100,6 +101,8 @@ disallow_untyped_defs = false
 testpaths = ["tests"]
 python_files = "test_*.py"
 addopts = "-ra -q"
+asyncio_default_fixture_loop_scope = "function"
+asyncio_mode = "auto"
 
 [[tool.uv.index]]
 name = "testpypi"

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_cli/_runtime/_context.py

@@ -11,7 +11,7 @@ class LangGraphRuntimeContext(UiPathRuntimeContext):
     """Context information passed throughout the runtime execution."""
 
     langgraph_config: Optional[LangGraphConfig] = None
-    state_graph: Optional[StateGraph] = None
+    state_graph: Optional[StateGraph[Any, Any]] = None
     output: Optional[Any] = None
     state: Optional[Any] = (
         None  # TypedDict issue, the actual type is: Optional[langgraph.types.StateSnapshot]

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -271,7 +271,9 @@ class LangGraphRuntime(UiPathBaseRuntime):
         if hasattr(self, "graph_config") and self.graph_config:
             await self.graph_config.cleanup()
 
-    def _extract_graph_result(self, final_chunk, graph: CompiledStateGraph):
+    def _extract_graph_result(
+        self, final_chunk, graph: CompiledStateGraph[Any, Any, Any]
+    ):
         """
         Extract the result from a LangGraph output chunk according to the graph's output channels.
 

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_cli/_utils/_graph.py

@@ -21,14 +21,18 @@ class GraphConfig:
     path: str
     file_path: str
     graph_var: str
-    _graph: Optional[Union[StateGraph, CompiledStateGraph]] = None
+    _graph: Optional[Union[StateGraph[Any, Any], CompiledStateGraph[Any, Any, Any]]] = (
+        None
+    )
 
     @classmethod
     def from_config(cls, name: str, path: str) -> "GraphConfig":
         file_path, graph_var = path.split(":")
         return cls(name=name, path=path, file_path=file_path, graph_var=graph_var)
 
-    async def load_graph(self) -> Union[StateGraph, CompiledStateGraph]:
+    async def load_graph(
+        self,
+    ) -> Union[StateGraph[Any, Any], CompiledStateGraph[Any, Any, Any]]:
         """Load graph from the specified path"""
         try:
             cwd = os.path.abspath(os.getcwd())

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_cli/cli_init.py

@@ -54,7 +54,9 @@ def process_nullable_types(
     return schema
 
 
-def generate_schema_from_graph(graph: CompiledStateGraph) -> Dict[str, Any]:
+def generate_schema_from_graph(
+    graph: CompiledStateGraph[Any, Any, Any],
+) -> Dict[str, Any]:
     """Extract input/output schema from a LangGraph graph"""
     schema = {
         "input": {"type": "object", "properties": {}, "required": []},

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_utils/_request_mixin.py

@@ -17,6 +17,7 @@ from tenacity import (
     stop_after_attempt,
     wait_exponential_jitter,
 )
+from uipath._utils._ssl_context import get_httpx_client_kwargs
 
 from uipath_langchain._utils._settings import (
     UiPathClientFactorySettings,
@@ -39,6 +40,8 @@ class UiPathRequestMixin(BaseModel):
 
     default_headers: Optional[Mapping[str, str]] = {
         "X-UiPath-Streaming-Enabled": "false",
+        "X-UiPath-JobKey": os.getenv("UIPATH_JOB_KEY", ""),
+        "X-UiPath-ProcessKey": os.getenv("UIPATH_PROCESS_KEY", ""),
     }
     model_name: Optional[str] = Field(
         default_factory=lambda: os.getenv("UIPATH_MODEL_NAME", "gpt-4o-2024-08-06"),
@@ -135,11 +138,13 @@ class UiPathRequestMixin(BaseModel):
         """Run an asynchronous call to the LLM."""
         # if self.logger:
         #     self.logger.info(f"Completion request: {request_body['messages'][:2]}")
+        client_kwargs = get_httpx_client_kwargs()
         with httpx.Client(
+            **client_kwargs,  # Apply SSL configuration
             event_hooks={
                 "request": [self._log_request_duration],
                 "response": [self._log_response_duration],
-            }
+            },
         ) as client:
             response = client.post(
                 url,
@@ -212,11 +217,13 @@ class UiPathRequestMixin(BaseModel):
     ) -> Dict[str, Any]:
         # if self.logger:
         #     self.logger.info(f"Completion request: {request_body['messages'][:2]}")
+        client_kwargs = get_httpx_client_kwargs()
         async with httpx.AsyncClient(
+            **client_kwargs,  # Apply SSL configuration
             event_hooks={
                 "request": [self._alog_request_duration],
                 "response": [self._alog_response_duration],
-            }
+            },
         ) as client:
             response = await client.post(
                 url,

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/_utils/_settings.py

@@ -1,6 +1,5 @@
 # mypy: disable-error-code="syntax"
 import os
-from enum import Enum
 from typing import Any, Optional
 
 import httpx
@@ -47,14 +46,6 @@ class UiPathClientSettings(BaseSettings):
     action_id: str = Field(default="DefaultActionId", alias="UIPATH_ACTION_ID")
 
 
-class UiPathEndpoints(Enum):
-    NORMALIZED_COMPLETION_ENDPOINT = "llmgateway_/api/chat/completions"
-    PASSTHROUGH_COMPLETION_ENDPOINT = "llmgateway_/openai/deployments/{model}/chat/completions?api-version={api_version}"
-    EMBEDDING_ENDPOINT = (
-        "llmgateway_/openai/deployments/{model}/embeddings?api-version={api_version}"
-    )
-
-
 def get_uipath_token_header(
     settings: Any = None,
 ) -> str:

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/chat/models.py

@@ -1,4 +1,6 @@
 import json
+import logging
+from os import environ as env
 from typing import Any, Dict, List, Literal, Optional, Union
 
 from langchain_core.callbacks import (
@@ -12,9 +14,11 @@ from langchain_core.outputs import ChatGeneration, ChatResult
 from langchain_core.runnables import Runnable
 from langchain_openai.chat_models import AzureChatOpenAI
 from pydantic import BaseModel
+from uipath.utils import EndpointManager
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
-from uipath_langchain._utils._settings import UiPathEndpoints
+
+logger = logging.getLogger(__name__)
 
 
 class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
@@ -71,7 +75,9 @@ class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.PASSTHROUGH_COMPLETION_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_passthrough_endpoint()
+        logger.debug("Using endpoint: %s", endpoint)
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )
 
@@ -79,6 +85,15 @@ class UiPathAzureChatOpenAI(UiPathRequestMixin, AzureChatOpenAI):
 class UiPathChat(UiPathRequestMixin, AzureChatOpenAI):
     """Custom LLM connector for LangChain integration with UiPath Normalized."""
 
+    def __init__(self, *args: Any, **kwargs: Any):
+        """Initialize the UiPath Azure Chat OpenAI model."""
+
+        super().__init__(*args, **kwargs)
+        self.default_headers = {
+            "X-UiPath-JobKey": env.get("UIPATH_JOB_KEY", ""),
+            "X-UiPath-ProcessKey": env.get("UIPATH_PROCESS_KEY", ""),
+        }
+
     def _create_chat_result(
         self,
         response: Union[Dict[str, Any], BaseModel],
@@ -252,7 +267,11 @@ class UiPathChat(UiPathRequestMixin, AzureChatOpenAI):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.NORMALIZED_COMPLETION_ENDPOINT.value
+        endpoint = EndpointManager.get_passthrough_endpoint()
+        logger.debug("Using endpoint: %s", endpoint)
+        return endpoint.format(
+            model=self.model_name, api_version=self.openai_api_version
+        )
 
     @property
     def is_normalized(self) -> bool:

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/src/uipath_langchain/embeddings/embeddings.py

@@ -5,9 +5,9 @@ import httpx
 from langchain_community.callbacks.manager import openai_callback_var
 from langchain_openai.embeddings import AzureOpenAIEmbeddings, OpenAIEmbeddings
 from pydantic import Field
+from uipath.utils import EndpointManager
 
 from uipath_langchain._utils._request_mixin import UiPathRequestMixin
-from uipath_langchain._utils._settings import UiPathEndpoints
 
 
 class UiPathAzureOpenAIEmbeddings(UiPathRequestMixin, AzureOpenAIEmbeddings):
@@ -43,7 +43,8 @@ class UiPathAzureOpenAIEmbeddings(UiPathRequestMixin, AzureOpenAIEmbeddings):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.EMBEDDING_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_embeddings_endpoint()
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )
 
@@ -59,7 +60,7 @@ class UiPathOpenAIEmbeddings(UiPathRequestMixin, OpenAIEmbeddings):
     )
 
     def embed_documents(
-        self, texts: List[str], chunk_size: Optional[int] = None
+        self, texts: List[str], chunk_size: Optional[int] = None, **kwargs
     ) -> List[List[float]]:
         """Embed a list of documents using the UiPath."""
         embeddings = []
@@ -83,6 +84,7 @@ class UiPathOpenAIEmbeddings(UiPathRequestMixin, OpenAIEmbeddings):
         self,
         texts: List[str],
         chunk_size: Optional[int] = None,
+        **kwargs,
     ) -> List[List[float]]:
         """Async version of embed_documents."""
         embeddings = []
@@ -104,6 +106,7 @@ class UiPathOpenAIEmbeddings(UiPathRequestMixin, OpenAIEmbeddings):
 
     @property
     def endpoint(self) -> str:
-        return UiPathEndpoints.EMBEDDING_ENDPOINT.value.format(
+        endpoint = EndpointManager.get_embeddings_endpoint()
+        return endpoint.format(
             model=self.model_name, api_version=self.openai_api_version
         )

{uipath_langchain-0.0.115 → uipath_langchain-0.0.117}/tests/hitl/mocks/api_trigger_hitl.py

@@ -1,26 +1,20 @@
-import dataclasses
+from typing import TypedDict
 
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph import END, START, StateGraph
 from langgraph.types import interrupt
 
 
-@dataclasses.dataclass
-class Input:
-    pass
-
-
-@dataclasses.dataclass
-class Output:
+class State(TypedDict):
     message: str
 
 
-def main_node(input: Input) -> Output:
+def main_node(state: State) -> State:
     response = interrupt("interrupt message")
-    return Output(message=response)
+    return {"message": response}
 
 
-builder = StateGraph(input=Input, output=Output)
+builder: StateGraph[State] = StateGraph(State)
 
 builder.add_node("main_node", main_node)