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

uipath_langchain/_cli/_templates/langgraph.json.template

@@ -1,7 +1,5 @@
 {
-    "dependencies": ["."],
     "graphs": {
       "agent": "./main.py:graph"
-    },
-    "env": ".env"
-  }
+    }
+}

uipath_langchain/_cli/cli_new.py

@@ -31,7 +31,7 @@ version = "0.0.1"
 description = "{project_name}"
 authors = [{{ name = "John Doe", email = "john.doe@myemail.com" }}]
 dependencies = [
-    "uipath-langchain>=0.1.0",
+    "uipath-langchain>=0.2.0",
 ]
 requires-python = ">=3.11"
 """
@@ -47,7 +47,6 @@ def langgraph_new_middleware(name: str) -> MiddlewareResult:
 
     try:
         with console.spinner(f"Creating new agent {name} in current directory ..."):
-            generate_pyproject(directory, name)
             generate_script(directory)
             console.success("Created 'main.py' file.")
             console.success("Created 'langgraph.json' file.")

uipath_langchain/_utils/_request_mixin.py

@@ -137,6 +137,8 @@ class UiPathRequestMixin(BaseModel):
     max_tokens: int | None = 1000
     frequency_penalty: float | None = None
     presence_penalty: float | None = None
+    agenthub_config: str | None = None
+    byo_connection_id: str | None = None
 
     logger: logging.Logger | None = None
     max_retries: int | None = 5
@@ -748,6 +750,12 @@ class UiPathRequestMixin(BaseModel):
                 "Authorization": f"Bearer {self.access_token}",
                 "X-UiPath-LlmGateway-TimeoutSeconds": str(self.default_request_timeout),
             }
+            if self.agenthub_config:
+                self._auth_headers["X-UiPath-AgentHub-Config"] = self.agenthub_config
+            if self.byo_connection_id:
+                self._auth_headers["X-UiPath-LlmGateway-ByoIsConnectionId"] = (
+                    self.byo_connection_id
+                )
             if self.is_normalized and self.model_name:
                 self._auth_headers["X-UiPath-LlmGateway-NormalizedApi-ModelName"] = (
                     self.model_name

uipath_langchain/_utils/_settings.py

@@ -5,6 +5,7 @@ from typing import Any
 import httpx
 from pydantic import Field
 from pydantic_settings import BaseSettings
+from uipath._utils._ssl_context import get_httpx_client_kwargs
 
 
 class UiPathCachedPathsSettings(BaseSettings):
@@ -58,7 +59,7 @@ def get_uipath_token_header(
             client_secret=settings.client_secret,
             grant_type="client_credentials",
         )
-        with httpx.Client() as client:
+        with httpx.Client(**get_httpx_client_kwargs()) as client:
             res = client.post(url_get_token, data=token_credentials)
             res_json = res.json()
             uipath_token_header = res_json.get("access_token")
@@ -79,7 +80,7 @@ async def get_token_header_async(
             grant_type="client_credentials",
         )
 
-        with httpx.Client() as client:
+        with httpx.Client(**get_httpx_client_kwargs()) as client:
             res_json = client.post(url_get_token, data=token_credentials).json()
             uipath_token_header = res_json.get("access_token")
 

uipath_langchain/agent/guardrails/__init__.py

@@ -1,21 +1,5 @@
-from .guardrail_nodes import (
-    create_agent_guardrail_node,
-    create_llm_guardrail_node,
-    create_tool_guardrail_node,
-)
 from .guardrails_factory import build_guardrails_with_actions
-from .guardrails_subgraph import (
-    create_agent_guardrails_subgraph,
-    create_llm_guardrails_subgraph,
-    create_tool_guardrails_subgraph,
-)
 
 __all__ = [
-    "create_llm_guardrails_subgraph",
-    "create_agent_guardrails_subgraph",
-    "create_tool_guardrails_subgraph",
-    "create_llm_guardrail_node",
-    "create_agent_guardrail_node",
-    "create_tool_guardrail_node",
     "build_guardrails_with_actions",
 ]

uipath_langchain/agent/guardrails/actions/__init__.py

@@ -1,6 +1,7 @@
 from .base_action import GuardrailAction
 from .block_action import BlockAction
 from .escalate_action import EscalateAction
+from .filter_action import FilterAction
 from .log_action import LogAction
 
 __all__ = [
@@ -8,4 +9,5 @@ __all__ = [
     "BlockAction",
     "LogAction",
     "EscalateAction",
+    "FilterAction",
 ]

uipath_langchain/agent/guardrails/actions/block_action.py

@@ -6,7 +6,7 @@ from uipath.runtime.errors import UiPathErrorCategory, UiPathErrorCode
 from uipath_langchain.agent.guardrails.types import ExecutionStage
 
 from ...exceptions import AgentTerminationException
-from ..types import AgentGuardrailsGraphState
+from ...react.types import AgentGuardrailsGraphState
 from .base_action import GuardrailAction, GuardrailActionNode