uipath-langchain 0.0.120__tar.gz → 0.0.121__tar.gz

uipath_langchain-0.0.121/.github/workflows/integration_tests.yml

@@ -0,0 +1,162 @@
+name: Integration testing
+
+on:
+  push:
+    branches: [ main, develop ]
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  integration-tests:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      pull-requests: write
+
+    steps:
+    - name: Checkout code
+      uses: actions/checkout@v4
+
+    - name: Set up Docker Buildx
+      uses: docker/setup-buildx-action@v3
+
+    - name: Build common Docker image
+      run: |
+        echo "Building common Docker image at $(date)"
+        docker build -f testcases/Dockerfile \
+          -t uipath-langchain-testbase:latest \
+          .
+        echo "Common Docker image built at $(date)"
+
+    - name: Set jobs variable
+      run: |
+        echo "Discovering testcase folders..."
+        
+        # Find all testcase folders (excluding common folders like README, etc.)
+        testcase_dirs=$(find testcases -maxdepth 1 -type d -name "*-*" | sed 's|testcases/||' | sort)
+        
+        echo "Found testcase directories:"
+        echo "$testcase_dirs"
+        
+        # Generate jobs array as a single line with space-separated values
+        jobs_list=""
+        for testcase in $testcase_dirs; do
+          # Add all 4 combinations for each testcase
+          jobs_list="${jobs_list} ${testcase}:true:alpha"
+          jobs_list="${jobs_list} ${testcase}:false:alpha"
+          jobs_list="${jobs_list} ${testcase}:true:cloud"
+          jobs_list="${jobs_list} ${testcase}:false:cloud"
+        done
+        
+        echo "Generated jobs list:"
+        echo "$jobs_list"
+        
+        # Save to GITHUB_ENV as a simple variable
+        echo "JOBS_LIST=$jobs_list" >> $GITHUB_ENV
+
+    - name: Run testcases
+      run: |
+        # Convert the jobs list back to an array
+        read -a jobs <<< "$JOBS_LIST"
+        
+        echo "=== STARTING TESTCASE EXECUTION ==="
+        echo "Total testcases to run: ${#jobs[@]}"
+        echo "Alpha environment with UiPathAzureChatOpenAI: $(echo "${jobs[@]}" | tr ' ' '\n' | grep ':true:alpha' | wc -l)"
+        echo "Alpha environment with UiPathChat: $(echo "${jobs[@]}" | tr ' ' '\n' | grep ':false:alpha' | wc -l)"
+        echo "Cloud environment with UiPathAzureChatOpenAI: $(echo "${jobs[@]}" | tr ' ' '\n' | grep ':true:cloud' | wc -l)"
+        echo "Cloud environment with UiPathChat: $(echo "${jobs[@]}" | tr ' ' '\n' | grep ':false:cloud' | wc -l)"
+        echo "======================================="
+
+        for job in "${jobs[@]}"; do
+          IFS=":" read -r testcase use_azure_chat environment <<< "$job"
+          llm=$([ "$use_azure_chat" = "true" ] && echo "UiPathAzureChatOpenAI" || echo "UiPathChat")
+          log_file="run_${testcase}_${llm}_${environment}.log"
+          
+          # Set environment-specific credentials
+          if [ "$environment" = "alpha" ]; then
+            CLIENT_ID_VAR='${{ secrets.ALPHA_TEST_CLIENT_ID }}'
+            CLIENT_SECRET_VAR='${{ secrets.ALPHA_TEST_CLIENT_SECRET }}'
+            BASE_URL_VAR='${{ secrets.ALPHA_BASE_URL }}'
+          else
+            CLIENT_ID_VAR='${{ secrets.CLOUD_TEST_CLIENT_ID }}'
+            CLIENT_SECRET_VAR='${{ secrets.CLOUD_TEST_CLIENT_SECRET }}'
+            BASE_URL_VAR='${{ secrets.CLOUD_BASE_URL }}'
+          fi
+          
+          (
+            echo "[$(date)] STARTING: $testcase ($llm) [$environment]"
+            
+            # Create log file with header
+            {
+              echo "========================================"
+              echo "TESTCASE: $testcase"
+              echo "LLM: $llm"
+              echo "ENVIRONMENT: $environment"
+              echo "USE_AZURE_CHAT: $use_azure_chat"
+              echo "STARTED_AT: $(date)"
+              echo "========================================"
+              echo ""
+            } > "$log_file"
+            
+            # Run the testcase and append to log file
+            docker run --rm \
+              -e CLIENT_ID="$CLIENT_ID_VAR" \
+              -e CLIENT_SECRET="$CLIENT_SECRET_VAR" \
+              -e BASE_URL="$BASE_URL_VAR" \
+              -e USE_AZURE_CHAT="$use_azure_chat" \
+              uipath-langchain-testbase:latest \
+              bash /app/testcases/$testcase/run.sh >> "$log_file" 2>&1
+            
+            # Capture the exit code immediately
+            exit_code=$?
+            
+            # Add completion status to log file
+            if [ $exit_code -eq 0 ]; then
+              echo "[$(date)] SUCCESS: $testcase ($llm) [$environment]"
+              {
+                echo ""
+                echo "========================================"
+                echo "COMPLETED_AT: $(date)"
+                echo "STATUS: SUCCESS"
+                echo "========================================"
+              } >> "$log_file"
+            else
+              echo "[$(date)] FAILED: $testcase ($llm) [$environment]"
+              {
+                echo ""
+                echo "========================================"
+                echo "COMPLETED_AT: $(date)"
+                echo "STATUS: FAILED"
+                echo "EXIT_CODE: $exit_code"
+                echo "========================================"
+              } >> "$log_file"
+            fi
+          ) &
+        done
+
+        wait
+        echo "All testcases execution completed."
+
+    - name: Display all testcase logs
+      run: |
+        # Convert the jobs list back to an array
+        read -a jobs <<< "$JOBS_LIST"
+        
+        for job in "${jobs[@]}"; do
+          IFS=":" read -r testcase use_azure_chat environment <<< "$job"
+          llm=$([ "$use_azure_chat" = "true" ] && echo "UiPathAzureChatOpenAI" || echo "UiPathChat")
+          log_file="run_${testcase}_${llm}_${environment}.log"
+          
+          # Create a collapsible group for each testcase log
+          echo "::group::${testcase^^} (${llm}) ${environment^^} LOG"
+          echo "Log file: $log_file"
+          
+          if [ -f "$log_file" ]; then
+            cat "$log_file"
+          else
+            echo "ERROR: Log file not found!"
+          fi
+          
+          echo "::endgroup::"
+          echo ""
+        done

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-langchain
-Version: 0.0.120
+Version: 0.0.121
 Summary: UiPath Langchain
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-langchain-python
@@ -20,11 +20,11 @@ Requires-Dist: langchain-core>=0.3.34
 Requires-Dist: langchain-openai>=0.3.3
 Requires-Dist: langchain>=0.3.4
 Requires-Dist: langgraph-checkpoint-sqlite>=2.0.3
-Requires-Dist: langgraph>=0.2.70
+Requires-Dist: langgraph<0.7.0,>=0.5.0
 Requires-Dist: openai>=1.65.5
 Requires-Dist: pydantic-settings>=2.6.0
 Requires-Dist: python-dotenv>=1.0.1
-Requires-Dist: uipath<2.2.0,>=2.1.0
+Requires-Dist: uipath<2.2.0,>=2.1.10
 Provides-Extra: langchain
 Description-Content-Type: text/markdown
 

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/pyproject.toml

@@ -1,12 +1,12 @@
 [project]
 name = "uipath-langchain"
-version = "0.0.120"
+version = "0.0.121"
 description = "UiPath Langchain"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.10"
 dependencies = [
-    "uipath>=2.1.0, <2.2.0",
-    "langgraph>=0.2.70",
+    "uipath>=2.1.10, <2.2.0",
+    "langgraph>=0.5.0, <0.7.0",
     "langchain-core>=0.3.34",
     "langgraph-checkpoint-sqlite>=2.0.3",
     "langchain-community>=0.3.21",

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/src/uipath_langchain/_cli/_runtime/_runtime.py

@@ -101,7 +101,7 @@ class LangGraphRuntime(UiPathBaseRuntime):
                     graph_config["max_concurrency"] = int(max_concurrency)
 
                 # Stream the output at debug time
-                if self.context.job_id is None:
+                if self.is_debug_run():
                     # Get final chunk while streaming
                     final_chunk = None
                     async for stream_chunk in graph.astream(
@@ -115,7 +115,7 @@ class LangGraphRuntime(UiPathBaseRuntime):
 
                     self.context.output = self._extract_graph_result(final_chunk, graph)
                 else:
-                    # Execute the graph normally at runtime
+                    # Execute the graph normally at runtime or eval
                     self.context.output = await graph.ainvoke(
                         processed_input, graph_config
                     )
@@ -310,9 +310,23 @@ class LangGraphRuntime(UiPathBaseRuntime):
             # Check which channels are present
             available_channels = [ch for ch in output_channels if ch in final_chunk]
 
+            # if no available channels, output may contain the last_node name as key
+            unwrapped_final_chunk = {}
+            if not available_channels:
+                if len(final_chunk) == 1 and isinstance(
+                    unwrapped_final_chunk := next(iter(final_chunk.values())), dict
+                ):
+                    available_channels = [
+                        ch for ch in output_channels if ch in unwrapped_final_chunk
+                    ]
+
             if available_channels:
                 # Create a dict with the available channels
-                return {channel: final_chunk[channel] for channel in available_channels}
+                return {
+                    channel: final_chunk.get(channel, None)
+                    or unwrapped_final_chunk[channel]
+                    for channel in available_channels
+                }
 
         # Fallback for any other case
         return final_chunk

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/src/uipath_langchain/_cli/cli_run.py

@@ -31,8 +31,8 @@ def langgraph_run_middleware(
             tracing = bool_map[tracing.lower()]
 
         async def execute():
-            context = LangGraphRuntimeContext.from_config(
-                env.get("UIPATH_CONFIG_PATH", "uipath.json")
+            context: LangGraphRuntimeContext = LangGraphRuntimeContext.from_config(
+                env.get("UIPATH_CONFIG_PATH", "uipath.json"), **kwargs
             )
             context.entrypoint = entrypoint
             context.input = input
@@ -42,6 +42,7 @@ def langgraph_run_middleware(
             context.logs_min_level = env.get("LOG_LEVEL", "INFO")
             context.job_id = env.get("UIPATH_JOB_KEY")
             context.trace_id = env.get("UIPATH_TRACE_ID")
+            context.is_eval_run = kwargs.get("is_eval_run", False)
             context.tracing_enabled = tracing
             context.input_file = kwargs.get("input_file", None)
             context.execution_output_file = kwargs.get("execution_output_file", None)

uipath_langchain-0.0.121/testcases/Dockerfile

@@ -0,0 +1,16 @@
+FROM ghcr.io/astral-sh/uv:python3.12-bookworm
+
+WORKDIR /app
+
+COPY . .
+
+# Install dependencies for all testcases
+RUN uv sync
+
+
+# Set environment variables for runtime
+ENV TAVILY_API_KEY=${TAVILY_API_KEY:-""}
+ENV UIPATH_TENANT_ID=${UIPATH_TENANT_ID:-""}
+ENV UIPATH_JOB_KEY=3a03d5cb-fa21-4021-894d-a8e2eda0afe0
+
+CMD ["bash"]

uipath_langchain-0.0.121/testcases/company-research-agent/run.sh

@@ -0,0 +1,44 @@
+#!/bin/bash
+
+# Authenticate with UiPath
+echo "Authenticating with UiPath..."
+uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL" 
+
+cd /app/testcases/company-research-agent
+
+# Sync dependencies for this specific testcase
+echo "Syncing dependencies..."
+uv sync
+
+# Pack the agent
+echo "Packing agent..."
+uv run uipath pack
+
+# Run the agent
+echo "Running agent with $CHAT_MODE mode..."
+echo "Input from input.json file"
+uv run uipath run agent --file input.json
+
+# Print the output file
+echo "Printing output file..."
+if [ -f "__uipath/output.json" ]; then
+    echo "=== OUTPUT FILE CONTENT ==="
+    cat __uipath/output.json
+    echo "=== END OUTPUT FILE CONTENT ==="
+else
+    echo "ERROR: __uipath/output.json not found!"
+    echo "Checking directory contents:"
+    ls -la
+    if [ -d "__uipath" ]; then
+        echo "Contents of __uipath directory:"
+        ls -la __uipath/
+    else
+        echo "__uipath directory does not exist!"
+    fi
+fi
+
+# Validate output
+echo "Validating output..."
+python src/assert.py
+
+echo "Testcase completed successfully."

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/testcases/company-research-agent/src/graph.py

@@ -7,7 +7,7 @@ from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.prebuilt import create_react_agent
 from pydantic import BaseModel
 
-from uipath_langchain.chat import UiPathAzureChatOpenAI
+from uipath_langchain.chat import UiPathAzureChatOpenAI, UiPathChat 
 
 # Configuration constants
 MAX_SEARCH_RESULTS = 5
@@ -46,11 +46,11 @@ DO NOT do any math as specified in your instructions.
 """
 
 
-def create_llm() -> UiPathAzureChatOpenAI:
-    """Create and configure the language model."""
-    return UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
-    # Alternative model option:
-    # return ChatAnthropic(model=ALTERNATIVE_MODEL)
+def create_llm() -> Union[UiPathAzureChatOpenAI, UiPathChat]:
+    """Create and configure the language model based on an environment variable."""
+    if os.getenv("USE_AZURE_CHAT", "false").lower() == "true":
+        return UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
+    return UiPathChat(model=DEFAULT_MODEL)
 
 
 def create_research_agent():

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/testcases/company-research-agent/uv.lock

@@ -2650,7 +2650,7 @@ wheels = [
 
 [[package]]
 name = "uipath-langchain"
-version = "0.0.118"
+version = "0.0.119"
 source = { editable = "../../" }
 dependencies = [
     { name = "httpx" },

uipath_langchain-0.0.121/testcases/multi-agent-supervisor-researcher-coder/run.sh

@@ -0,0 +1,44 @@
+#!/bin/bash
+
+# Authenticate with UiPath
+echo "Authenticating with UiPath..."
+uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL"
+
+cd /app/testcases/multi-agent-supervisor-researcher-coder
+
+# Sync dependencies for this specific testcase
+echo "Syncing dependencies..."
+uv sync
+
+# Pack the agent
+echo "Packing agent..."
+uv run uipath pack
+
+# Run the agent
+echo "Running agent with $CHAT_MODE mode..."
+echo "Input from input.json file"
+uv run uipath run agent --file input.json
+
+# Print the output file
+echo "Printing output file..."
+if [ -f "__uipath/output.json" ]; then
+    echo "=== OUTPUT FILE CONTENT ==="
+    cat __uipath/output.json
+    echo "=== END OUTPUT FILE CONTENT ==="
+else
+    echo "ERROR: __uipath/output.json not found!"
+    echo "Checking directory contents:"
+    ls -la
+    if [ -d "__uipath" ]; then
+        echo "Contents of __uipath directory:"
+        ls -la __uipath/
+    else
+        echo "__uipath directory does not exist!"
+    fi
+fi
+
+# Validate output
+echo "Validating output..."
+python src/assert.py
+
+echo "Testcase completed successfully."

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/testcases/multi-agent-supervisor-researcher-coder/src/graph.py

@@ -12,7 +12,7 @@ from langgraph.types import Command
 from pydantic import BaseModel
 from typing_extensions import TypedDict
 
-from uipath_langchain.chat import UiPathAzureChatOpenAI
+from uipath_langchain.chat import UiPathAzureChatOpenAI, UiPathChat
 
 # Configuration constants
 MAX_SEARCH_RESULTS = 5
@@ -69,11 +69,11 @@ def create_supervisor_prompt() -> str:
     )
 
 
-def create_llm() -> UiPathAzureChatOpenAI:
-    """Create and configure the language model."""
-    return UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
-    # Alternative model option:
-    # return ChatAnthropic(model=ALTERNATIVE_MODEL)
+def create_llm() -> Union[UiPathAzureChatOpenAI, UiPathChat]:
+    """Create and configure the language model based on an environment variable."""
+    if os.getenv("USE_AZURE_CHAT", "false").lower() == "true":
+        return UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
+    return UiPathChat(model=DEFAULT_MODEL)
 
 
 # Type definitions

uipath_langchain-0.0.121/testcases/simple-local-mcp/run.sh

@@ -0,0 +1,45 @@
+#!/bin/bash
+
+# Authenticate with UiPath
+echo "Authenticating with UiPath..."
+uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL"
+
+cd /app/testcases/simple-local-mcp
+
+# Sync dependencies for this specific testcase
+echo "Syncing dependencies..."
+uv sync
+
+# Pack the agent
+echo "Packing agent..."
+uv run uipath pack
+
+# Run the agent
+echo "Running agent with $CHAT_MODE mode..."
+echo "Input from input.json file"
+uv run uipath run agent --file input.json
+
+# Print the output file
+echo "Printing output file..."
+if [ -f "__uipath/output.json" ]; then
+    echo "=== OUTPUT FILE CONTENT ==="
+    cat __uipath/output.json
+    echo "=== END OUTPUT FILE CONTENT ==="
+else
+    echo "ERROR: __uipath/output.json not found!"
+    echo "Checking directory contents:"
+    ls -la
+    if [ -d "__uipath" ]; then
+        echo "Contents of __uipath directory:"
+        ls -la __uipath/
+    else
+        echo "__uipath directory does not exist!"
+    fi
+fi
+
+
+# Validate output
+echo "Validating output..."
+python src/assert.py
+
+echo "Testcase completed successfully."

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/testcases/simple-local-mcp/src/simple-local-mcp/graph.py

@@ -4,9 +4,12 @@ from contextlib import asynccontextmanager
 
 from langchain_mcp_adapters.client import MultiServerMCPClient
 from langgraph.prebuilt import create_react_agent
-from uipath_langchain.chat import UiPathAzureChatOpenAI
+from uipath_langchain.chat import UiPathAzureChatOpenAI, UiPathChat
 
-model = UiPathAzureChatOpenAI(model="gpt-4o-2024-08-06")
+if os.getenv("USE_AZURE_CHAT", "false").lower() == "true":
+    model = UiPathAzureChatOpenAI(model="gpt-4o-2024-08-06")
+else:
+    model = UiPathChat(model="gpt-4o-2024-08-06")
 
 @asynccontextmanager
 async def make_graph():

uipath_langchain-0.0.121/testcases/ticket-classification/run.sh

@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# Authenticate with UiPath
+echo "Authenticating with UiPath..."
+uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL"
+
+cd /app/testcases/ticket-classification
+
+# Sync dependencies for this specific testcase
+echo "Syncing dependencies..."
+uv sync
+
+# Pack the agent
+echo "Packing agent..."
+uv run uipath pack
+
+# Run the agent
+echo "Running agent with $CHAT_MODE mode..."
+echo "Input from input.json file"
+uv run uipath run agent --file input.json
+
+echo "Resuming agent run by default with {'Answer': true}..."
+uv run uipath run agent '{"Answer": true}' --resume;
+
+
+# Print the output file
+echo "Printing output file..."
+if [ -f "__uipath/output.json" ]; then
+    echo "=== OUTPUT FILE CONTENT ==="
+    cat __uipath/output.json
+    echo "=== END OUTPUT FILE CONTENT ==="
+else
+    echo "ERROR: __uipath/output.json not found!"
+    echo "Checking directory contents:"
+    ls -la
+    if [ -d "__uipath" ]; then
+        echo "Contents of __uipath directory:"
+        ls -la __uipath/
+    else
+        echo "__uipath directory does not exist!"
+    fi
+fi
+
+# Validate output
+echo "Validating output..."
+python src/assert.py
+
+echo "Testcase completed successfully."

{uipath_langchain-0.0.120 → uipath_langchain-0.0.121}/testcases/ticket-classification/src/main.py

@@ -13,7 +13,7 @@ from pydantic import BaseModel, Field
 
 from uipath import UiPath
 from uipath.models import CreateAction
-from uipath_langchain.chat import UiPathAzureChatOpenAI
+from uipath_langchain.chat import UiPathAzureChatOpenAI, UiPathChat
 
 # Configuration
 logger = logging.getLogger(__name__)
@@ -115,8 +115,11 @@ def decide_next_node(state: GraphState) -> NextNode:
 
 async def classify(state: GraphState) -> Command:
     """Classify the support ticket using LLM."""
-    llm = UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
-    
+    if os.getenv("USE_AZURE_CHAT", "false").lower() == "true":
+        llm = UiPathAzureChatOpenAI(model=DEFAULT_MODEL)
+    else:
+        llm = UiPathChat(model=DEFAULT_MODEL)
+
     # Add rejection message if there was a previous prediction
     if state.get("last_predicted_category"):
         predicted_category = state["last_predicted_category"]
@@ -125,16 +128,16 @@ async def classify(state: GraphState) -> Command:
             "Choose another one."
         )
         state["messages"].append(HumanMessage(content=rejection_message))
-    
+
     chain = llm | output_parser
-    
+
     try:
         result = await chain.ainvoke(state["messages"])
         logger.info(
             f"Ticket classified with label: {result.label}, "
             f"confidence score: {result.confidence}"
         )
-        
+
         return Command(
             update={
                 "confidence": result.confidence,