added retrieval-chain workflow

Author: giuliastf

.github/workflows/integration_tests.yml

@@ -0,0 +1,34 @@
+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 Docker image
+      working-directory: ./samples/company-research-agent
+      run: |
+        docker build -t company-research-agent:test \
+          --build-arg CLIENT_ID="${{ secrets.ALPHA_TEST_CLIENT_ID }}" \
+          --build-arg CLIENT_SECRET="${{ secrets.ALPHA_TEST_CLIENT_SECRET }}" \
+          --build-arg BASE_URL="${{ secrets.ALPHA_BASE_URL }}" \
+          .
+          exit 1  # Intentionally fail the build
+      
+      # env:
+      #     ALPHA_BASE_URL: ${{ secrets.ALPHA_BASE_URL }}

samples/company-research-agent/Dockerfile

@@ -0,0 +1,28 @@
+FROM ghcr.io/astral-sh/uv:python3.12-bookworm
+
+WORKDIR /app
+
+COPY . .
+
+RUN uv sync
+
+ARG CLIENT_ID
+ARG CLIENT_SECRET
+ARG BASE_URL
+
+# Validate required environment variables
+RUN if [ -z "$CLIENT_ID" ]; then echo "CLIENT_ID build arg is required" && exit 1; fi
+RUN if [ -z "$CLIENT_SECRET" ]; then echo "CLIENT_SECRET build arg is required" && exit 1; fi
+RUN if [ -z "$BASE_URL" ]; then echo "BASE_URL build arg is required" && exit 1; fi
+
+# Set environment variables for runtime
+ENV CLIENT_ID=$CLIENT_ID
+ENV CLIENT_SECRET=$CLIENT_SECRET
+ENV BASE_URL=$BASE_URL
+ENV TAVILY_API_KEY=${TAVILY_API_KEY:-""}
+
+
+# Authenticate with UiPath during build
+RUN uv run uipath auth --client-id="$CLIENT_ID" --client-secret="$CLIENT_SECRET" --base-url="$BASE_URL"
+
+RUN uv run uipath run agent '{"company_name":"uipath"}'

samples/company-research-agent/graph.py

@@ -1,11 +1,17 @@
+import os
 from langchain_anthropic import ChatAnthropic
 from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.tools import DuckDuckGoSearchResults
 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
 
 # Set up the Tavily search tool
-tavily_tool = TavilySearchResults(max_results=5)
+if os.getenv("TAVILY_API_KEY"):
+    tool = TavilySearchResults(max_results=5)
+else:
+    tool = DuckDuckGoSearchResults()
 
 # Define system prompt
 system_prompt = """You are an advanced AI assistant specializing in corporate research and outreach strategy development. Your primary functions are:
@@ -29,10 +35,10 @@
 
 DO NOT do any math as specified in your instructions.
 """
+llm = UiPathAzureChatOpenAI(model="gpt-4o-2024-08-06")
+# llm = ChatAnthropic(model="claude-3-5-sonnet-latest")
 
-llm = ChatAnthropic(model="claude-3-5-sonnet-latest")
-
-research_agent = create_react_agent(llm, tools=[tavily_tool], prompt=system_prompt)
+research_agent = create_react_agent(llm, tools=[tool], prompt=system_prompt)
 
 
 class GraphState(BaseModel):

samples/company-research-agent/pyproject.toml

@@ -3,12 +3,17 @@ name = "company-research-agent"
 version = "0.0.1"
 description = "Company research agent with Tavily web search"
 authors = [{ name = "John Doe", email = "[email protected]" }]
+
 requires-python = ">=3.10"
 dependencies = [
     "langgraph>=0.2.55",
     "langchain-anthropic>=0.3.8",
     "tavily-python>=0.5.0",
-    "uipath-langchain==0.0.113"
+    "uipath>=2.0.79",
+    "uipath-langchain>=0.0.117",
+    "duckduckgo-search>=8.1.1",
+    "langchain-community>=0.3.21",
+    "debugpy>=1.8.15",
 ]
 
 [project.optional-dependencies]
@@ -41,3 +46,9 @@ lint.ignore = [
 
 [tool.ruff.lint.per-file-ignores]
 "tests/*" = ["D", "UP"]
+
+[[tool.uv.index]]
+name = "testpypi"
+url = "https://test.pypi.org/simple/"
+publish-url = "https://test.pypi.org/legacy/"
+explicit = true