Dockerfile for company research agent

Author: giuliastf

samples/company-research-agent/Dockerfile

@@ -0,0 +1,34 @@
+FROM ghcr.io/astral-sh/uv:python3.12-alpine
+
+# Set working directory
+WORKDIR /app
+
+# Copy project files
+COPY pyproject.toml uv.lock ./
+COPY main.py ./
+COPY README.md ./
+
+# Install dependencies
+RUN uv sync
+
+# Create environment file from build args
+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(max_results=5)
 
 # 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,14 @@ name = "company-research-agent"
 version = "0.0.1"
 description = "Company research agent with Tavily web search"
 authors = [{ name = "John Doe", email = "john.doe@myemail.com" }]
+
 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-langchain==0.0.113",
+    "uipath>=2.0.72",
 ]
 
 [project.optional-dependencies]