uipath-langchain 0.0.89__tar.gz → 0.0.91__tar.gz

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/.github/workflows/ci.yml

@@ -11,6 +11,9 @@ on:
 
 
 jobs:
+  commit-lint:
+    if: ${{ github.event_name == 'pull_request' }}
+    uses: ./.github/workflows/commitlint.yml
   lint:
     uses: ./.github/workflows/lint.yml
 

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/.github/workflows/commitlint.yml

@@ -1,9 +1,7 @@
 name: Commit Lint
 
 on:
-  pull_request:
-    branches:
-      - main
+  workflow_call:
 
 jobs:
   commit-lint:

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.89
+Version: 0.0.91
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -25,7 +25,7 @@ Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
 Requires-Dist: requests>=2.23.3
 Requires-Dist: types-requests>=2.32.0.20241016
-Requires-Dist: uipath<2.1.0,>=2.0.7
+Requires-Dist: uipath<2.1.0,>=2.0.13
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.89"
+version = "0.0.91"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.0.7, <2.1.0",
+    "uipath>=2.0.13, <2.1.0",
     "langgraph>=0.2.70",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/multi-agent-planner-researcher-coder-distributed/README.md

@@ -208,16 +208,16 @@ uipath run coder '{"messages":[{"content":"Let me help you state the Pythagorean
 
 6. **Create Agent Processes in Orchestrator**
    - **Planner Agent**
-     ![planner-agent-package-overview](../../docs/sample_images/planner-agent-package-overview.png)
-     ![planner-agent-process-configuration](../../docs/sample_images/planner-agent-process-configuration.png)
+     ![planner-agent-package-overview](../../docs/sample_images/multi-agent-distributed/planner-agent-package-overview.png)
+     ![planner-agent-process-configuration](../../docs/sample_images/multi-agent-distributed/planner-agent-process-configuration.png)
 
    - **Researcher Agent**
-     ![researcher-agent-package-overview](../../docs/sample_images/researcher-agent-package-overview.png)
-     ![researcher-agent-process-configuration](../../docs/sample_images/researcher-agent-process-configuration.png)
+     ![researcher-agent-package-overview](../../docs/sample_images/multi-agent-distributed/researcher-agent-package-overview.png)
+     ![researcher-agent-process-configuration](../../docs/sample_images/multi-agent-distributed/researcher-agent-process-configuration.png)
 
    - **Coder Agent**
-     ![coder-agent-package-overview](../../docs/sample_images/coder-agent-package-overview.png)
-     ![coder-agent-process-configuration](../../docs/sample_images/coder-agent-process-configuration.png)
+     ![coder-agent-package-overview](../../docs/sample_images/multi-agent-distributed/coder-agent-package-overview.png)
+     ![coder-agent-process-configuration](../../docs/sample_images/multi-agent-distributed/coder-agent-process-configuration.png)
 
    > **Note:** Ensure that the display names for the coder and researcher agent processes are *coder-agent* and *researcher-agent*,
         and that all 3 agents are placed in the same folder.

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/ticket-classification/.env.example

@@ -1,4 +1,3 @@
 UIPATH_URL=https://alpha.uipath.com/ada/byoa
 UIPATH_ACCESS_TOKEN=xxx
-AZURE_OPENAI_API_KEY=xxx
-AZURE_OPENAI_ENDPOINT=xxx
+OPENAI_API_KEY=xxx

uipath_langchain-0.0.91/samples/ticket-classification/README.md

@@ -0,0 +1,206 @@
+# Support Ticket Classification System
+
+Use LangGraph with Azure OpenAI to automatically classify support tickets into predefined categories with confidence scores. UiPath Action Center integration for human approval step.
+
+## Debug
+
+1. Clone the repository:
+```bash
+git clone
+cd samples\ticket-classification
+```
+
+2. Install dependencies:
+```bash
+pip install uv
+uv venv -p 3.11 .venv
+.venv\Scripts\activate
+uv sync
+```
+
+3. Create a `.env` file in the project root with the following configuration:
+```env
+UIPATH_URL=https://alpha.uipath.com/ada/byoa
+UIPATH_ACCESS_TOKEN=xxx
+AZURE_OPENAI_API_KEY=xxx
+AZURE_OPENAI_ENDPOINT=xxx
+```
+
+```bash
+uipath run <entrypoint> <input> [--resume]
+```
+
+### Run
+
+To classify a ticket, run the script using UiPath CLI:
+
+```bash
+uipath run agent '{"message": "GET Assets API does not enforce proper permissions Assets.View", "ticket_id": "TICKET-2345"}'
+```
+
+### Resume
+
+To resume the graph with approval:
+
+```bash
+uipath run agent true --resume
+```
+
+### Input Format
+
+The input ticket should be in the following format:
+```json
+{
+    "message": "The ticket message or description",
+    "ticket_id": "Unique ticket identifier",
+    "assignee"[optional]: "username or email of the person assigned to handle escalations"
+}
+```
+
+### Output Format
+
+The script outputs JSON with the classification results:
+```json
+{
+    "label": "security",
+    "confidence": 0.9
+}
+```
+
+## Deployment Guide
+
+This guide walks you through deploying and running the ticket classification agent on the UiPath Platform.
+
+### Prerequisites
+- Access to UiPath Platform
+- Python 3.11 or higher
+- Git
+
+### 1. Repository Setup
+```bash
+# Clone the repository
+git clone https://github.com/UiPath/uipath-langchain-python.git
+cd uipath-langchain-python/samples/ticket-classification
+```
+
+### 2. Action App Deployment
+
+The Ticket Classification Agent utilizes HITL (Human In The Loop) technology, allowing the system to incorporate feedback directly from supervisory personnel. We'll leverage UiPath [Action Center](https://docs.uipath.com/action-center/automation-suite/2023.4/user-guide/introduction) for this functionality.
+
+Follow these steps to deploy the pre-built application using [UiPath Solutions Management](https://docs.uipath.com/solutions-management/automation-cloud/latest/user-guide/solutions-management-overview):
+
+1. **Upload Solution Package**
+   - Navigate to UiPath Solutions Management
+   - Drag and drop [generic-escalation-app-solution-1.0.0.zip](escalation_app_solution/generic-escalation-app-solution-1.0.0.zip) to the upload area
+   - Click the _Upload_ button
+
+   ![upload-solution-package](../../docs/sample_images/ticket-classification/upload-solution-package.png)
+
+2. **Initiate Deployment**
+   - Wait for the package to be processed and ready for deployment
+
+   ![deploy-solution-package](../../docs/sample_images/ticket-classification/deploy-solution-package.png)
+
+3. **Select Destination**
+   - Choose a destination folder or install as root folder under tenant
+
+   ![choose-destination-folder](../../docs/sample_images/ticket-classification/solution-destination-folder.png)
+
+4. **Complete Configuration**
+   - Follow the solution configuration wizard prompts
+
+   ![solution-configuration-wizard](../../docs/sample_images/ticket-classification/deploy-solution-package-wizard.png)
+
+5. **Activate the Apps**
+   - After deployment, activate the apps following the [UiPath documentation](https://docs.uipath.com/apps/automation-cloud/latest/user-guide/apps-in-solutions-management)
+
+   ![activate-deployment](../../docs/sample_images/ticket-classification/activate-deployment.png)
+   ![activate-apps](../../docs/sample_images/ticket-classification/activate-apps.png)
+
+6. **Verify and Configure**
+   - Navigate to the solution folder to verify the escalation app creation
+
+   ![navigate-to-solution-folder](../../docs/sample_images/ticket-classification/navigate-to-solution-folder.png)
+
+   - Copy the folder path for configuration
+
+   ![copy-folder-path](../../docs/sample_images/ticket-classification/copy-folder-path.png)
+
+   - Update the `_PLACEHOLDER` string in `main.py` (line 148) with the copied folder path
+
+### 3. Agent Setup and Publishing
+
+1. **Set Up Python Environment**
+```bash
+# Install UV package manager
+pip install uv
+
+# Create and activate virtual environment
+uv venv -p 3.11 .venv
+
+# Windows
+.venv\Scripts\activate
+
+# Unix-like Systems
+source .venv/bin/activate
+
+# Install dependencies
+uv sync
+```
+
+2. **UiPath Authentication**
+```bash
+uipath auth
+```
+After successful authentication in the browser, select the tenant for publishing the agent package:
+```
+Received log data
+Received authentication information
+Available tenants:
+  0: cosmin
+  1: DefaultTenant
+  2: Demo
+  3: lucian
+  4: Solutions
+  5: SolutionsTest
+  6: Test
+  7: TestRoles
+Select tenant: 2
+```
+
+3. **Package and Publish**
+```bash
+# Create and publish the package
+uipath pack
+uipath publish
+```
+Select your preferred feed type when prompted:
+```
+Select feed type:
+  0: Tenant package feed
+  1: Personal workspace
+Select feed: 1
+```
+Note: When publishing to personal workspace feed, the process will be auto-created for you.
+
+### 4. Running and Monitoring the Agent
+
+1. **Start the Agent**
+   - Navigate to your agent in the UiPath workspace
+   - Click "Run" to start a new job
+
+   ![run the agent](../../docs/sample_images/ticket-classification/run-agent.png)
+   ![start job](../../docs/sample_images/ticket-classification/start-job.png)
+
+2. **Monitor Progress**
+   - Track the agent's progress using the details side panel
+
+   ![monitor agent](../../docs/sample_images/ticket-classification/monitor-agent.png)
+
+3. **Handle Human-in-the-Loop Tasks**
+   - When a ticket is classified, a _Resume Condition_ tab will appear
+   - Use this link to navigate to UiPath Action Center for human intervention
+
+   ![resume condition](../../docs/sample_images/ticket-classification/resume-condition.png)
+
+For detailed information about UiPath Action Center and its features, refer to the [official documentation](https://docs.uipath.com/action-center/automation-suite/2024.10/user-guide/introduction).

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/ticket-classification/main.py

@@ -3,7 +3,7 @@ import os
 from typing import Literal, Optional, List
 
 from langchain_core.messages import HumanMessage, SystemMessage
-from langchain_openai import AzureChatOpenAI
+from langchain_openai import ChatOpenAI
 from langchain_core.output_parsers import PydanticOutputParser
 from langgraph.graph import START, END, StateGraph, MessagesState
 from langgraph.types import interrupt, Command
@@ -70,21 +70,6 @@ def prepare_input(graph_input: GraphInput) -> GraphState:
         human_approval=None,
     )
 
-def get_azure_openai_api_key() -> str:
-    """Get Azure OpenAI API key from environment or UiPath."""
-    api_key = os.getenv("AZURE_OPENAI_API_KEY")
-
-    if not api_key:
-        try:
-            api_key = uipath.assets.retrieve_credential("AZURE_OPENAI_API_KEY")
-            if not api_key:
-                raise ValueError("No API key found in credentials")
-        except Exception as e:
-            logger.error(f"Failed to get API key: {str(e)}")
-            raise RuntimeError("Failed to get Azure OpenAI API key")
-
-    return api_key
-
 def decide_next_node(state: GraphState) -> Literal["classify", "notify_team"]:
     if state["human_approval"] is True:
         return "notify_team"
@@ -93,12 +78,7 @@ def decide_next_node(state: GraphState) -> Literal["classify", "notify_team"]:
 
 async def classify(state: GraphState) -> Command:
     """Classify the support ticket using LLM."""
-    llm = AzureChatOpenAI(
-        azure_deployment="gpt-4o-mini",
-        api_key=get_azure_openai_api_key(),
-        azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
-        api_version="2024-10-21"
-    )
+    llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
     if state.get("last_predicted_category", None):
         predicted_category = state["last_predicted_category"]
@@ -133,7 +113,7 @@ async def wait_for_human(state: GraphState) -> Command:
     ticket_message = state["messages"][1].content
     label = state["label"]
     confidence = state["confidence"]
-    action_data = interrupt(CreateAction(name="escalation_agent_app",
+    action_data = interrupt(CreateAction(app_name="escalation_agent_app",
                                          title="Action Required: Review classification",
                                          data={
                                              "AgentOutput": (
@@ -145,6 +125,7 @@ async def wait_for_human(state: GraphState) -> Command:
                                              "AgentName": "ticket-classification "},
                                          app_version=1,
                                          assignee=state.get("assignee", None),
+                                         app_folder_path="Shared/escalation_app",
                                          ))
 
     return Command(

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/ticket-classification/pyproject.toml

@@ -9,7 +9,7 @@ dependencies = [
     "langchain-openai>=0.3.9",
     "langgraph-checkpoint-sqlite>=2.0.3",
     "python-dotenv>=1.0.1",
-    "uipath-langchain==0.0.88",
+    "uipath-langchain==0.0.90",
     "pydantic>=2.10.6",
     "aiohttp>=3.11.12",
     "typing-extensions>=4.12.2",

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/ticket-classification/uipath.json

@@ -2,7 +2,7 @@
   "entryPoints": [
     {
       "filePath": "agent",
-      "uniqueId": "525ea50d-c54f-4185-94e2-501fe9ab341a",
+      "uniqueId": "f5678e12-7aa8-4a4c-bf62-b98dad8d9145",
       "type": "agent",
       "input": {
         "type": "object",
@@ -54,23 +54,6 @@
   ],
   "bindings": {
     "version": "2.0",
-    "resources": [
-      {
-        "resource": "asset",
-        "key": "AZURE_OPENAI_API_KEY",
-        "value": {
-          "name": {
-            "defaultValue": "AZURE_OPENAI_API_KEY",
-            "isExpression": false,
-            "displayName": "Name"
-          }
-        },
-        "metadata": {
-          "ActivityName": "retrieve_credential",
-          "BindingsVersion": "2.1",
-          "DisplayLabel": "FullName"
-        }
-      }
-    ]
+    "resources": []
   }
 }

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/samples/ticket-classification/uv.lock

@@ -1653,7 +1653,7 @@ wheels = [
 
 [[package]]
 name = "ticket-classification-agent"
-version = "0.0.2"
+version = "0.0.1"
 source = { virtual = "." }
 dependencies = [
     { name = "aiohttp" },
@@ -1679,7 +1679,7 @@ requires-dist = [
     { name = "pydantic", specifier = ">=2.10.6" },
     { name = "python-dotenv", specifier = ">=1.0.1" },
     { name = "typing-extensions", specifier = ">=4.12.2" },
-    { name = "uipath-langchain", specifier = "==0.0.88" },
+    { name = "uipath-langchain", specifier = "==0.0.90" },
 ]
 
 [[package]]
@@ -1826,7 +1826,7 @@ wheels = [
 
 [[package]]
 name = "uipath"
-version = "2.0.4"
+version = "2.0.9"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "click" },
@@ -1839,14 +1839,14 @@ dependencies = [
     { name = "tomli" },
     { name = "types-requests" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/6c/0d/4ae746ca61ec42ca6ab449692bacfbb34e013a97224e6f175715b448ab9b/uipath-2.0.4.tar.gz", hash = "sha256:9ac7abbc2e1f53207e11cb1235e99efc6cf7c265595e0f1ac981512a443e70bc", size = 207150 }
+sdist = { url = "https://files.pythonhosted.org/packages/ca/b2/4be1b45c55be25428fd4b229cf9554a46f6ecd2abc076c6cd7bed9903137/uipath-2.0.9.tar.gz", hash = "sha256:5de6678bd3dd4cc4e97f0e5635a8c7f41a9736b1a7a40ebd3888753764f85be3", size = 210455 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/0f/d9/e4c2d3ba7d87e59937ca5980f4bc2b7d06c68578df33a9a0d4eb99764d44/uipath-2.0.4-py3-none-any.whl", hash = "sha256:698680eb2f8fd885dbc9e13856028cdd10cfac525698375022ff484629f3f778", size = 85424 },
+    { url = "https://files.pythonhosted.org/packages/d7/74/f34b9b7db516e0c2e569618bd693a116f37e7b5f5bdd73e3db3cdcdc34c7/uipath-2.0.9-py3-none-any.whl", hash = "sha256:37212e2494c300aef08a593f4dcbbb1234dc7c26d2d665c2fe671dcd486ee286", size = 86475 },
 ]
 
 [[package]]
 name = "uipath-langchain"
-version = "0.0.88"
+version = "0.0.90"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "httpx" },
@@ -1863,9 +1863,9 @@ dependencies = [
     { name = "types-requests" },
     { name = "uipath" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/80/0e/e05ecfc5a5b6b7f277dca18e61724c1a923ae4c9af39f71f2c8dce52343d/uipath_langchain-0.0.88.tar.gz", hash = "sha256:f06cc80c9499ae4309275119a9abf5d74c881aaef6731c6efb5b525a1291ef33", size = 1188502 }
+sdist = { url = "https://files.pythonhosted.org/packages/e7/d0/945364a318dd04363364be56db5e771e0996ac0ae8487ebc28518da1274a/uipath_langchain-0.0.90.tar.gz", hash = "sha256:9cc19b71c6a05223771f36f1906737852a0b1f607757b740e0afad7e131872ed", size = 1475658 }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/3a/7a/68ec6fcb4698af7f27c30a2107c349fb6115b84373dba38413346e8c5a57/uipath_langchain-0.0.88-py3-none-any.whl", hash = "sha256:78c92bebbd4a3651adcf914d3563a551222bef86f54cb3eb270d93a39ede7d28", size = 50641 },
+    { url = "https://files.pythonhosted.org/packages/20/42/2238adcdad8b1184e0ff716a0d14918ddd5ff9f833504aed99ac8b537c45/uipath_langchain-0.0.90-py3-none-any.whl", hash = "sha256:eddcb8d3e72ed09e89c79300e03b7793d8567edf47f3f3338c4ad55ce18540b6", size = 50995 },
 ]
 
 [[package]]

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/src/uipath_langchain/_cli/_runtime/_input.py

@@ -58,10 +58,12 @@ class LangGraphInputProcessor:
         if not trigger:
             return Command(resume=self.context.input_json)
 
-        type, key = trigger
+        type, key, folder_path, folder_key, payload = trigger
         logger.debug(f"ResumeTrigger: {type} {key}")
         if type == UiPathResumeTriggerType.ACTION.value and key:
-            action = self.uipath.actions.retrieve(key)
+            action = await self.uipath.actions.retrieve_async(
+                key, app_folder_key=folder_key, app_folder_path=folder_path
+            )
             logger.debug(f"Action: {action}")
             if action.data is None:
                 return Command(resume={})
@@ -95,7 +97,7 @@ class LangGraphInputProcessor:
                 return Command(resume=try_convert_to_json_format(job.output_arguments))
         return Command(resume=self.context.input_json)
 
-    async def _get_latest_trigger(self) -> Optional[tuple[str, str]]:
+    async def _get_latest_trigger(self) -> Optional[tuple[str, str, str, str, str]]:
         """Fetch the most recent trigger from the database."""
         if self.context.memory is None:
             return None
@@ -106,7 +108,7 @@ class LangGraphInputProcessor:
                 self.context.memory.conn.cursor() as cur,
             ):
                 await cur.execute(f"""
-                    SELECT type, key
+                    SELECT type, key, folder_path, folder_key, payload
                     FROM {self.context.resume_triggers_table}
                     ORDER BY timestamp DESC
                     LIMIT 1
@@ -114,7 +116,7 @@ class LangGraphInputProcessor:
                 result = await cur.fetchone()
                 if result is None:
                     return None
-                return cast(tuple[str, str], tuple(result))
+                return cast(tuple[str, str, str, str, str], tuple(result))
         except Exception as e:
             raise LangGraphRuntimeError(
                 "DB_QUERY_FAILED",

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/src/uipath_langchain/_cli/_runtime/_output.py

@@ -232,6 +232,9 @@ class LangGraphOutputProcessor:
                             id INTEGER PRIMARY KEY AUTOINCREMENT,
                             type TEXT NOT NULL,
                             key TEXT,
+                            folder_key TEXT,
+                            folder_path TEXT,
+                            payload TEXT,
                             timestamp DATETIME DEFAULT (strftime('%Y-%m-%d %H:%M:%S', 'now', 'utc'))
                         )
                     """)
@@ -280,6 +283,12 @@ class LangGraphOutputProcessor:
                                     app_name=self.interrupt_value.app_name
                                     if self.interrupt_value.app_name
                                     else "",
+                                    app_folder_path=self.interrupt_value.app_folder_path
+                                    if self.interrupt_value.app_folder_path
+                                    else "",
+                                    app_folder_key=self.interrupt_value.app_folder_key
+                                    if self.interrupt_value.app_folder_key
+                                    else "",
                                     app_key=self.interrupt_value.app_key
                                     if self.interrupt_value.app_key
                                     else "",
@@ -295,11 +304,25 @@ class LangGraphOutputProcessor:
                                     self._resume_trigger = UiPathResumeTrigger(
                                         trigger_type=UiPathResumeTriggerType.ACTION,
                                         item_key=action.key,
+                                        payload=self.interrupt_value.model_dump_json(),
+                                        folder_path=self.interrupt_value.app_folder_path
+                                        if self.interrupt_value.app_folder_path
+                                        else None,
+                                        folder_key=self.interrupt_value.app_folder_key
+                                        if self.interrupt_value.app_folder_key
+                                        else None,
                                     )
                             elif isinstance(self.interrupt_value, WaitAction):
                                 self._resume_trigger = UiPathResumeTrigger(
                                     triggerType=UiPathResumeTriggerType.ACTION,
                                     itemKey=self.interrupt_value.action.key,
+                                    payload=self.interrupt_value.model_dump_json(),
+                                    folder_path=self.interrupt_value.app_folder_path
+                                    if self.interrupt_value.app_folder_path
+                                    else None,
+                                    folder_key=self.interrupt_value.app_folder_key
+                                    if self.interrupt_value.app_folder_key
+                                    else None,
                                 )
 
                 except Exception as e:
@@ -324,8 +347,14 @@ class LangGraphOutputProcessor:
                 try:
                     logger.debug(f"ResumeTrigger: {trigger_type} {trigger_key}")
                     await cur.execute(
-                        f"INSERT INTO {self.context.resume_triggers_table} (type, key) VALUES (?, ?)",
-                        (trigger_type, trigger_key),
+                        f"INSERT INTO {self.context.resume_triggers_table} (type, key, payload, folder_path, folder_key) VALUES (?, ?, ?, ?, ?)",
+                        (
+                            trigger_type,
+                            trigger_key,
+                            self.resume_trigger.payload,
+                            self.resume_trigger.folder_path,
+                            self.resume_trigger.folder_key,
+                        ),
                     )
                     await self.context.memory.conn.commit()
                 except Exception as e:

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -15,7 +15,7 @@ from uipath._cli._runtime._contracts import (
     UiPathRuntimeResult,
 )
 
-from ..._utils import _instrument_traceable
+from ..._utils import _instrument_traceable_attributes
 from ...tracers import AsyncUiPathTracer
 from ._context import LangGraphRuntimeContext
 from ._exception import LangGraphRuntimeError
@@ -45,7 +45,7 @@ class LangGraphRuntime(UiPathBaseRuntime):
         Raises:
             LangGraphRuntimeError: If execution fails
         """
-        _instrument_traceable()
+        _instrument_traceable_attributes()
 
         await self.validate()
 
@@ -94,6 +94,14 @@ class LangGraphRuntime(UiPathBaseRuntime):
                     "callbacks": callbacks,
                 }
 
+                recursion_limit = os.environ.get("LANGCHAIN_RECURSION_LIMIT", None)
+                max_concurrency = os.environ.get("LANGCHAIN_MAX_CONCURRENCY", None)
+
+                if recursion_limit is not None:
+                    graph_config["recursion_limit"] = int(recursion_limit)
+                if max_concurrency is not None:
+                    graph_config["max_concurrency"] = int(max_concurrency)
+
                 # Stream the output at debug time
                 if self.context.job_id is None:
                     # Get final chunk while streaming

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/src/uipath_langchain/_utils/__init__.py

@@ -1,4 +1,4 @@
-from ..tracers._instrument_traceable import _instrument_traceable
+from ..tracers._instrument_traceable import _instrument_traceable_attributes
 from ._request_mixin import UiPathRequestMixin
 
-__all__ = ["UiPathRequestMixin", "_instrument_traceable"]
+__all__ = ["UiPathRequestMixin", "_instrument_traceable_attributes"]

{uipath_langchain-0.0.89 → uipath_langchain-0.0.91}/src/uipath_langchain/tracers/__init__.py

@@ -1,5 +1,5 @@
-from ._instrument_traceable import _instrument_traceable
+from ._instrument_traceable import _instrument_traceable_attributes
 from .AsyncUiPathTracer import AsyncUiPathTracer
 from .UiPathTracer import UiPathTracer
 
-__all__ = ["AsyncUiPathTracer", "UiPathTracer", "_instrument_traceable"]
+__all__ = ["AsyncUiPathTracer", "UiPathTracer", "_instrument_traceable_attributes"]