Simple LangGraph Agent Initial Commit

.chainlit/config.toml

@@ -0,0 +1,120 @@
+[project]
+# Whether to enable telemetry (default: true). No personal data is collected.
+enable_telemetry = true
+
+
+# List of environment variables to be provided by each user to use the app.
+user_env = []
+
+# Duration (in seconds) during which the session is saved when the connection is lost
+session_timeout = 3600
+
+# Duration (in seconds) of the user session expiry
+user_session_timeout = 1296000  # 15 days
+
+# Enable third parties caching (e.g., LangChain cache)
+cache = false
+
+# Authorized origins
+allow_origins = ["*"]
+
+[features]
+# Process and display HTML in messages. This can be a security risk (see https://stackoverflow.com/questions/19603097/why-is-it-dangerous-to-render-user-generated-html-or-javascript)
+unsafe_allow_html = false
+
+# Process and display mathematical expressions. This can clash with "$" characters in messages.
+latex = false
+
+# Autoscroll new user messages at the top of the window
+user_message_autoscroll = true
+
+# Automatically tag threads with the current chat profile (if a chat profile is used)
+auto_tag_thread = true
+
+# Allow users to edit their own messages
+edit_message = true
+
+# Authorize users to spontaneously upload files with messages
+[features.spontaneous_file_upload]
+    enabled = true
+    # Define accepted file types using MIME types
+    # Examples:
+    # 1. For specific file types:
+    #    accept = ["image/jpeg", "image/png", "application/pdf"]
+    # 2. For all files of certain type:
+    #    accept = ["image/*", "audio/*", "video/*"]
+    # 3. For specific file extensions:
+    #    accept = { "application/octet-stream" = [".xyz", ".pdb"] }
+    # Note: Using "*/*" is not recommended as it may cause browser warnings
+    accept = ["*/*"]
+    max_files = 20
+    max_size_mb = 500
+
+[features.audio]
+    # Sample rate of the audio
+    sample_rate = 24000
+
+[features.mcp.sse]
+    enabled = true
+
+[features.mcp.stdio]
+    enabled = true
+    # Only the executables in the allow list can be used for MCP stdio server.
+    # Only need the base name of the executable, e.g. "npx", not "/usr/bin/npx".
+    # Please don't comment this line for now, we need it to parse the executable name.
+    allowed_executables = [ "npx", "uvx" ]
+
+[UI]
+# Name of the assistant.
+name = "Assistant"
+
+# default_theme = "dark"
+
+# layout = "wide"
+
+# default_sidebar_state = "open"
+
+# Description of the assistant. This is used for HTML tags.
+# description = ""
+
+# Chain of Thought (CoT) display mode. Can be "hidden", "tool_call" or "full".
+cot = "full"
+
+# Specify a CSS file that can be used to customize the user interface.
+# The CSS file can be served from the public directory or via an external link.
+# custom_css = "/public/test.css"
+
+# Specify additional attributes for a custom CSS file
+# custom_css_attributes = "media=\"print\""
+
+# Specify a JavaScript file that can be used to customize the user interface.
+# The JavaScript file can be served from the public directory.
+# custom_js = "/public/test.js"
+
+# Specify additional attributes for custom JS file
+# custom_js_attributes = "async type = \"module\""
+
+# Custom login page image, relative to public directory or external URL
+# login_page_image = "/public/custom-background.jpg"
+
+# Custom login page image filter (Tailwind internal filters, no dark/light variants)
+# login_page_image_filter = "brightness-50 grayscale"
+# login_page_image_dark_filter = "contrast-200 blur-sm"
+
+# Specify a custom meta image url.
+# custom_meta_image_url = "https://chainlit-cloud.s3.eu-west-3.amazonaws.com/logo/chainlit_banner.png"
+
+# Specify a custom build directory for the frontend.
+# This can be used to customize the frontend code.
+# Be careful: If this is a relative path, it should not start with a slash.
+# custom_build = "./public/build"
+
+# Specify optional one or more custom links in the header.
+# [[UI.header_links]]
+#     name = "Issues"
+#     display_name = "Report Issue"
+#     icon_url = "https://avatars.githubusercontent.com/u/128686189?s=200&v=4"
+#     url = "https://github.com/Chainlit/chainlit/issues"
+
+[meta]
+generated_by = "2.5.5"

.chainlit/translations/en-US.json

@@ -0,0 +1,214 @@
+{
+    "common": {
+        "actions": {
+            "cancel": "Cancel",
+            "confirm": "Confirm",
+            "continue": "Continue",
+            "goBack": "Go Back",
+            "reset": "Reset",
+            "submit": "Submit"
+        },
+        "status": {
+            "loading": "Loading...",
+            "error": {
+                "default": "An error occurred",
+                "serverConnection": "Could not reach the server"
+            }
+        }
+    },
+    "auth": {
+        "login": {
+            "title": "Login to access the app",
+            "form": {
+                "email": {
+                    "label": "Email address",
+                    "required": "email is a required field"
+                },
+                "password": {
+                    "label": "Password",
+                    "required": "password is a required field"
+                },
+                "actions": {
+                    "signin": "Sign In"
+                },
+                "alternativeText": {
+                    "or": "OR"
+                }
+            },
+            "errors": {
+                "default": "Unable to sign in",
+                "signin": "Try signing in with a different account",
+                "oauthSignin": "Try signing in with a different account",
+                "redirectUriMismatch": "The redirect URI is not matching the oauth app configuration",
+                "oauthCallback": "Try signing in with a different account",
+                "oauthCreateAccount": "Try signing in with a different account",
+                "emailCreateAccount": "Try signing in with a different account",
+                "callback": "Try signing in with a different account",
+                "oauthAccountNotLinked": "To confirm your identity, sign in with the same account you used originally",
+                "emailSignin": "The e-mail could not be sent",
+                "emailVerify": "Please verify your email, a new email has been sent",
+                "credentialsSignin": "Sign in failed. Check the details you provided are correct",
+                "sessionRequired": "Please sign in to access this page"
+            }
+        },
+        "provider": {
+            "continue": "Continue with {{provider}}"
+        }
+    },
+    "chat": {
+        "input": {
+            "placeholder": "Type your message here...",
+            "actions": {
+                "send": "Send message",
+                "stop": "Stop Task",
+                "attachFiles": "Attach files"
+            }
+        },
+        "speech": {
+            "start": "Start recording",
+            "stop": "Stop recording",
+            "connecting": "Connecting"
+        },
+        "fileUpload": {
+            "dragDrop": "Drag and drop files here",
+            "browse": "Browse Files",
+            "sizeLimit": "Limit:",
+            "errors": {
+                "failed": "Failed to upload",
+                "cancelled": "Cancelled upload of"
+            }
+        },
+        "messages": {
+            "status": {
+                "using": "Using",
+                "used": "Used"
+            },
+            "actions": {
+                "copy": {
+                    "button": "Copy to clipboard",
+                    "success": "Copied!"
+                }
+            },
+            "feedback": {
+                "positive": "Helpful",
+                "negative": "Not helpful",
+                "edit": "Edit feedback",
+                "dialog": {
+                    "title": "Add a comment",
+                    "submit": "Submit feedback"
+                },
+                "status": {
+                    "updating": "Updating",
+                    "updated": "Feedback updated"
+                }
+            }
+        },
+        "history": {
+            "title": "Last Inputs",
+            "empty": "Such empty...",
+            "show": "Show history"
+        },
+        "settings": {
+            "title": "Settings panel"
+        },
+        "watermark": "Built with"
+    },
+    "threadHistory": {
+        "sidebar": {
+            "title": "Past Chats",
+            "filters": {
+                "search": "Search",
+                "placeholder": "Search conversations..."
+            },
+            "timeframes": {
+                "today": "Today",
+                "yesterday": "Yesterday",
+                "previous7days": "Previous 7 days",
+                "previous30days": "Previous 30 days"
+            },
+            "empty": "No threads found",
+            "actions": {
+                "close": "Close sidebar",
+                "open": "Open sidebar"
+            }
+        },
+        "thread": {
+            "untitled": "Untitled Conversation",
+            "menu": {
+                "rename": "Rename",
+                "delete": "Delete"
+            },
+            "actions": {
+                "delete": {
+                    "title": "Confirm deletion",
+                    "description": "This will delete the thread as well as its messages and elements. This action cannot be undone",
+                    "success": "Chat deleted",
+                    "inProgress": "Deleting chat"
+                },
+                "rename": {
+                    "title": "Rename Thread",
+                    "description": "Enter a new name for this thread",
+                    "form": {
+                        "name": {
+                            "label": "Name",
+                            "placeholder": "Enter new name"
+                        }
+                    },
+                    "success": "Thread renamed!",
+                    "inProgress": "Renaming thread"
+                }
+            }
+        }
+    },
+    "navigation": {
+        "header": {
+            "chat": "Chat",
+            "readme": "Readme",
+            "theme": {
+                "light": "Light Theme",
+                "dark": "Dark Theme",
+                "system": "Follow System"
+            }
+        },
+        "newChat": {
+            "button": "New Chat",
+            "dialog": {
+                "title": "Create New Chat",
+                "description": "This will clear your current chat history. Are you sure you want to continue?",
+                "tooltip": "New Chat"
+            }
+        },
+        "user": {
+            "menu": {
+                "settings": "Settings",
+                "settingsKey": "S",
+                "apiKeys": "API Keys",
+                "logout": "Logout"
+            }
+        }
+    },
+    "apiKeys": {
+        "title": "Required API Keys",
+        "description": "To use this app, the following API keys are required. The keys are stored on your device's local storage.",
+        "success": {
+            "saved": "Saved successfully"
+        }
+    },
+    "alerts": {
+        "info": "Info",
+        "note": "Note",
+        "tip": "Tip",
+        "important": "Important",
+        "warning": "Warning",
+        "caution": "Caution",
+        "debug": "Debug",
+        "example": "Example",
+        "success": "Success",
+        "help": "Help",
+        "idea": "Idea",
+        "pending": "Pending",
+        "security": "Security",
+        "beta": "Beta",
+        "best-practice": "Best Practice"
+    }
+}

.dockerignore

@@ -0,0 +1,35 @@
+# Version control
+.git
+.gitignore
+
+# Virtual environments
+.venv
+venv
+env
+
+# Python cache files
+__pycache__/
+*.py[cod]
+*$py.class
+.pytest_cache
+.coverage
+htmlcov/
+
+# Build directories
+dist/
+build/
+*.egg-info/
+
+# IDE directories
+.idea/
+.vscode/
+
+# Environment files (security - don't include real .env in production)
+# .env
+
+# Logs
+logs/
+*.log
+
+# Chainlit cache
+.chainlit/cache 

.gitignore

@@ -0,0 +1,10 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv

.python-version

@@ -0,0 +1 @@
+3.11

Dockerfile

@@ -0,0 +1,30 @@
+# Get a distribution that has uv already installed
+FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim
+
+# Add user - this is the user that will run the app
+# If you do not set user, the app will run as root (undesirable)
+RUN useradd -m -u 1000 user
+USER user
+
+# Set the home directory and path
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH        
+
+# Set the working directory
+WORKDIR $HOME/app
+
+# Copy the app to the container
+COPY --chown=user . $HOME/app/
+
+# Install the dependencies
+# RUN uv sync --frozen
+RUN uv sync
+
+# Set the port for Hugging Face
+ENV PORT=7860
+# Expose the port Chainlit runs on
+EXPOSE 7860
+
+# Command to run the application
+# Run the app
+CMD ["uv", "run", "chainlit", "run", "main.py", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -0,0 +1,59 @@
+# LangGraph Tool-Calling Agent
+
+A research assistant powered by LangGraph and Chainlit that can search the web and query arXiv papers to answer questions.
+
+## Features
+
+- Multi-tool agent with web search capabilities (Tavily and DuckDuckGo)
+- Academic research with arXiv integration
+- Interactive Chainlit web interface
+- Streaming responses with real-time tool usage visibility
+
+## Local Development
+
+1. Clone the repository
+2. Create a virtual environment: `python -m venv .venv`
+3. Activate the virtual environment:
+   - Windows: `.venv\Scripts\activate`
+   - macOS/Linux: `source .venv/bin/activate`
+4. Install dependencies: `pip install -r requirements.txt`
+5. Create a `.env` file with your API keys:
+   ```
+   OPENAI_API_KEY=your_openai_key_here
+   TAVILY_API_KEY=your_tavily_key_here
+   LANGCHAIN_API_KEY=your_langchain_key_here
+   LANGCHAIN_TRACING_V2=true
+   LANGCHAIN_PROJECT=tool-calling-agent
+   ```
+6. Run the application: `chainlit run main.py`
+7. Open your browser to http://localhost:8501
+
+## Deploying to Hugging Face Spaces
+
+1. Create a new Space on Hugging Face (https://huggingface.co/new-space)
+2. Choose "Docker" as the Space SDK
+3. Clone your Space repository
+4. Copy your project files to the cloned repository
+5. Add your API keys as repository secrets in the Space settings
+6. Push your changes to Hugging Face
+7. Your app will build and deploy automatically
+
+### Environment Variables for Hugging Face
+
+Make sure to add these environment variables in your Hugging Face Space settings:
+
+- `OPENAI_API_KEY`
+- `TAVILY_API_KEY`
+- `LANGCHAIN_API_KEY`
+- `LANGCHAIN_TRACING_V2`
+- `LANGCHAIN_PROJECT`
+
+## Project Structure
+
+- `main.py` - Entry point and Chainlit handler
+- `config.py` - Configuration management
+- `tools.py` - Tool definitions and setup
+- `graph.py` - LangGraph agent implementation
+- `chainlit.yaml` - Chainlit configuration
+- `Dockerfile` - Container definition for deployment
+

chainlit.md

@@ -0,0 +1,14 @@
+# Welcome to Chainlit! 🚀🤖
+
+Hi there, Developer! 👋 We're excited to have you on board. Chainlit is a powerful tool designed to help you prototype, debug and share applications built on top of LLMs.
+
+## Useful Links 🔗
+
+- **Documentation:** Get started with our comprehensive [Chainlit Documentation](https://docs.chainlit.io) 📚
+- **Discord Community:** Join our friendly [Chainlit Discord](https://discord.gg/k73SQ3FyUh) to ask questions, share your projects, and connect with other developers! 💬
+
+We can't wait to see what you create with Chainlit! Happy coding! 💻😊
+
+## Welcome screen
+
+To modify the welcome screen, edit the `chainlit.md` file at the root of your project. If you do not want a welcome screen, just leave this file empty.

chainlit.yaml

@@ -0,0 +1,10 @@
+# chainlit.yaml
+chainlit_server:
+  port: 8501
+  cors:
+    allowed_origins: ["*"]
+
+ui:
+  name: "LangGraph Agent"
+  default_expand_messages: true
+  theme: light

config.py

@@ -0,0 +1,32 @@
+"""
+Configuration module for the LangGraph application.
+"""
+import os
+from uuid import uuid4
+from typing import Dict, Any
+
+def load_config() -> Dict[str, Any]:
+    """
+    Load configuration from environment variables.
+    
+    Returns:
+        Dict[str, Any]: Configuration dictionary
+    """
+    config = {
+        "openai_api_key": os.getenv("OPENAI_API_KEY"),
+        "tavily_api_key": os.getenv("TAVILY_API_KEY"),
+        "langchain_api_key": os.getenv("LANGCHAIN_API_KEY"),
+        "langchain_tracing_v2": os.getenv("LANGCHAIN_TRACING_V2"),
+        "langchain_project": f"AIE6 - HW5 Advanced Build - {uuid4().hex[0:8]}",
+        "model_name": "gpt-4o-mini",
+        "temperature": 0,
+        "tavily_max_results": 5
+    }
+    
+    # Validate required configuration
+    required_keys = ["openai_api_key", "tavily_api_key"]
+    for key in required_keys:
+        if not config[key]:
+            raise ValueError(f"Missing required environment variable for {key}")
+            
+    return config 

graph.py

@@ -0,0 +1,102 @@
+"""
+Module for creating and configuring the LangGraph agent workflow.
+"""
+from typing import List, Dict, Any
+
+from langsmith import traceable
+from pydantic import BaseModel
+from typing import Annotated
+
+from langgraph.graph import StateGraph, END
+from langgraph.graph.message import add_messages
+from langgraph.prebuilt import ToolNode
+from langchain_openai import ChatOpenAI
+
+class AgentState(BaseModel):
+    """
+    State model for the LangGraph agent.
+    """
+    messages: Annotated[list, add_messages]
+
+@traceable (
+    metadata=
+    {
+        "model_name": "gpt-4o-mini", 
+        "model_provider": "ChatOpenAI"
+    }
+)
+def call_model(state: AgentState) -> AgentState:
+    """
+    Node function that calls the LLM to generate a response.
+    
+    Args:
+        state: Current state containing messages
+        
+    Returns:
+        Updated state with model response
+    """
+    # Access the model from the global context
+    # The model is bound to this function when the graph is created
+    #model = call_model.__globals__["_model"]
+    model = get_model()
+    messages = state.messages
+    response = model.invoke(messages)
+    return {"messages": [response]}
+
+def should_continue(state: AgentState) -> str:
+    """
+    Conditional edge function to determine the next node.
+    
+    Args:
+        state: Current agent state
+        
+    Returns:
+        String indicating the next node or END
+    """
+    last_message = state.messages[-1]
+
+    if last_message.tool_calls:
+        return "action"
+
+    return END
+
+def create_agent_graph(tools: List, model: ChatOpenAI) -> StateGraph:
+    """
+    Create the LangGraph agent workflow.
+    
+    Args:
+        tools: List of LangChain tools
+        model: ChatOpenAI model with tools bound
+        
+    Returns:
+        Compiled StateGraph
+    """
+    # Create tool node to execute tools
+    tool_node = ToolNode(tools)
+    
+    # Set model in the global context for call_model function
+    # This avoids issues with serialization when the graph is compiled
+    global _model
+    _model = model
+ 
+    
+    # Create the graph
+    graph = StateGraph(AgentState)
+    
+    # Add nodes
+    graph.add_node("action", tool_node)
+    graph.add_node("agent", call_model)
+    
+    # Add edges
+    graph.set_entry_point("agent")
+    graph.add_edge("action", "agent")
+    graph.add_conditional_edges(
+        "agent",
+        should_continue
+    )
+    
+    # Compile the graph
+    return graph.compile() 
+
+def get_model():
+       return get_model.__globals__["_model"]

main.py

@@ -0,0 +1,139 @@
+from dotenv import load_dotenv
+import os
+
+from langchain_core.messages import HumanMessage
+from langchain_core.runnables.config import RunnableConfig
+
+from config import load_config
+from tools import create_tools
+from graph import create_agent_graph
+
+import chainlit as cl
+import re
+
+# Initialize resources at module level for Chainlit
+load_dotenv()
+config = load_config()
+tools, model = create_tools(config)
+graph = create_agent_graph(tools, model)
+
+def clean_content(content):
+    """Clean and format content for better readability.
+    
+    Args:
+        content: The content to clean
+        
+    Returns:
+        Cleaned content string
+    """
+    if not content:
+        return ""
+    
+    # Replace escaped newlines with actual newlines
+    content = content.replace('\\n', '\n')
+    
+    # Remove excessive backslashes
+    content = content.replace('\\\\', '\\')
+    
+    # Format URLs properly
+    content = re.sub(r'(?<![\[\("])https?://[^\s\'"]+', r'[Link](\g<0>)', content)
+    
+    # Remove raw markdown links that didn't render
+    content = re.sub(r'\\*\[([^\]]+)\]\\*\(([^)]+)\)', r'[\1](\2)', content)
+    
+    # Clean up repeated newlines
+    content = re.sub(r'\n{3,}', '\n\n', content)
+    
+    return content
+
+@cl.on_chat_start
+async def on_chat_start():
+    """Set up the session state when a new chat starts."""
+    cl.user_session.set("graph", graph)
+    # Send welcome message
+    welcome = cl.Message(content="👋 Hello! I'm a research assistant powered by LangGraph. I can search the web using Tavily and query arXiv papers. How can I help you today?")
+    await welcome.send()
+
+@cl.on_message
+async def on_message(message: cl.Message):
+    """Process incoming user messages with the LangGraph agent."""
+    # Get the compiled graph from session
+    graph = cl.user_session.get("graph")
+    
+    # Create input with the user's message
+    inputs = {
+        "messages": [
+            HumanMessage(content=message.content)
+        ]
+    }
+    
+    # Set up callback handler for streaming
+    cb = cl.LangchainCallbackHandler()
+    runnable_config = RunnableConfig(callbacks=[cb])
+    
+    # Send thinking indicator
+    thinking = cl.Message(content="🤔 Thinking...")
+    await thinking.send()
+    
+    # Send initial response message that will be streamed to
+    response_message = cl.Message(content="")
+    await response_message.send()
+    
+    # Remove thinking indicator once we start getting results
+    await thinking.remove()
+    
+    # Process with agent and stream results
+    last_tool_name = None
+    async for chunk in graph.astream(inputs, stream_mode="values", config=runnable_config):
+        if "messages" in chunk and chunk["messages"]:
+            last_message = chunk["messages"][-1]
+            
+            # Handle main content
+            if hasattr(last_message, "content") and last_message.content:
+                clean_text = clean_content(last_message.content)
+                await response_message.stream_token(clean_text)
+            
+            # Handle tool calls - avoid duplicates and clean up display
+            if hasattr(last_message, "tool_calls") and last_message.tool_calls:
+                tool_call = last_message.tool_calls[0]
+                tool_name = tool_call.get('name', '')
+                
+                # Only show each tool call once
+                if tool_name != last_tool_name:
+                    last_tool_name = tool_name
+                    
+                    # Format tool call message
+                    formatted_name = f"🔧 Using {tool_name.replace('_', ' ').title()}"
+                    tool_message = cl.Message(
+                        content=formatted_name,
+                        author="Agent"
+                    )
+                    await tool_message.send()
+            
+            # Handle tool results - clean up and format nicely
+            if hasattr(last_message, "name") and last_message.name:
+                tool_name = last_message.name
+                tool_content = last_message.content
+                
+                # Truncate long tool results and format them
+                if len(tool_content) > 300:
+                    tool_content = tool_content[:297] + "..."
+                
+                tool_content = clean_content(tool_content)
+                
+                # Add emoji based on tool type
+                emoji = "🔍"
+                if "arxiv" in tool_name.lower():
+                    emoji = "📚"
+                elif "search" in tool_name.lower() or "tavily" in tool_name.lower():
+                    emoji = "🌐"
+                
+                formatted_title = f"{emoji} {tool_name.replace('_', ' ').title()} Results:"
+                tool_result = cl.Message(
+                    content=f"{formatted_title}\n\n{tool_content}",
+                    author="Tool"
+                )
+                await tool_result.send()
+# Only used when running the script directly, not with Chainlit
+if __name__ == "__main__":
+    print("Please run this app with Chainlit: chainlit run main.py")

pyproject.toml

@@ -0,0 +1,20 @@
+[project]
+name = "tool-calling-agent"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11, <3.12"
+dependencies = [
+    "arxiv>=2.2.0",
+    "chainlit>=2.5.5",
+    "duckduckgo-search>=8.0.1",
+    "langchain>=0.3.23",
+    "langchain-community>=0.3.21",
+    "langchain-core>=0.3.52",
+    "langchain-openai>=0.3.13",
+    "langgraph>=0.3.30",
+    "langgraph-prebuilt>=0.1.8",
+    "pydantic>=2.11.3",
+    "python-dotenv>=1.1.0",
+    "tavily-python>=0.5.4",
+]

requirements.txt

@@ -0,0 +1,606 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml -o requirements.txt
+aiofiles==24.1.0
+    # via chainlit
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.11.16
+    # via
+    #   langchain-community
+    #   traceloop-sdk
+aiosignal==1.3.2
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+anthropic==0.49.0
+    # via opentelemetry-instrumentation-bedrock
+anyio==4.9.0
+    # via
+    #   anthropic
+    #   asyncer
+    #   httpx
+    #   mcp
+    #   openai
+    #   sse-starlette
+    #   starlette
+    #   watchfiles
+arxiv==2.2.0
+    # via tool-calling-agent (pyproject.toml)
+asyncer==0.0.7
+    # via chainlit
+attrs==25.3.0
+    # via aiohttp
+backoff==2.2.1
+    # via posthog
+bidict==0.23.1
+    # via python-socketio
+certifi==2025.1.31
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+chainlit==2.5.5
+    # via tool-calling-agent (pyproject.toml)
+charset-normalizer==3.4.1
+    # via requests
+chevron==0.14.0
+    # via literalai
+click==8.1.8
+    # via
+    #   chainlit
+    #   duckduckgo-search
+    #   uvicorn
+colorama==0.4.6
+    # via traceloop-sdk
+dataclasses-json==0.6.7
+    # via
+    #   chainlit
+    #   langchain-community
+deprecated==1.2.18
+    # via
+    #   opentelemetry-api
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+    #   opentelemetry-semantic-conventions
+    #   traceloop-sdk
+distro==1.9.0
+    # via
+    #   anthropic
+    #   openai
+    #   posthog
+duckduckgo-search==8.0.1
+    # via tool-calling-agent (pyproject.toml)
+fastapi==0.115.12
+    # via chainlit
+feedparser==6.0.11
+    # via arxiv
+filelock==3.18.0
+    # via huggingface-hub
+filetype==1.2.0
+    # via chainlit
+frozenlist==1.6.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2025.3.2
+    # via huggingface-hub
+googleapis-common-protos==1.70.0
+    # via
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+grpcio==1.71.0
+    # via opentelemetry-exporter-otlp-proto-grpc
+h11==0.14.0
+    # via
+    #   httpcore
+    #   uvicorn
+    #   wsproto
+httpcore==1.0.8
+    # via httpx
+httpx==0.28.1
+    # via
+    #   anthropic
+    #   chainlit
+    #   langgraph-sdk
+    #   langsmith
+    #   literalai
+    #   mcp
+    #   openai
+    #   tavily-python
+httpx-sse==0.4.0
+    # via
+    #   langchain-community
+    #   mcp
+huggingface-hub==0.30.2
+    # via tokenizers
+idna==3.10
+    # via
+    #   anyio
+    #   httpx
+    #   requests
+    #   yarl
+importlib-metadata==8.6.1
+    # via opentelemetry-api
+inflection==0.5.1
+    # via opentelemetry-instrumentation-llamaindex
+jinja2==3.1.6
+    # via traceloop-sdk
+jiter==0.9.0
+    # via
+    #   anthropic
+    #   openai
+jsonpatch==1.33
+    # via langchain-core
+jsonpointer==3.0.0
+    # via jsonpatch
+langchain==0.3.23
+    # via
+    #   tool-calling-agent (pyproject.toml)
+    #   langchain-community
+langchain-community==0.3.21
+    # via tool-calling-agent (pyproject.toml)
+langchain-core==0.3.54
+    # via
+    #   tool-calling-agent (pyproject.toml)
+    #   langchain
+    #   langchain-community
+    #   langchain-openai
+    #   langchain-text-splitters
+    #   langgraph
+    #   langgraph-checkpoint
+    #   langgraph-prebuilt
+langchain-openai==0.3.14
+    # via tool-calling-agent (pyproject.toml)
+langchain-text-splitters==0.3.8
+    # via langchain
+langgraph==0.3.31
+    # via tool-calling-agent (pyproject.toml)
+langgraph-checkpoint==2.0.24
+    # via
+    #   langgraph
+    #   langgraph-prebuilt
+langgraph-prebuilt==0.1.8
+    # via
+    #   tool-calling-agent (pyproject.toml)
+    #   langgraph
+langgraph-sdk==0.1.61
+    # via langgraph
+langsmith==0.3.32
+    # via
+    #   langchain
+    #   langchain-community
+    #   langchain-core
+lazify==0.4.0
+    # via chainlit
+literalai==0.1.201
+    # via chainlit
+lxml==5.3.2
+    # via duckduckgo-search
+markupsafe==3.0.2
+    # via jinja2
+marshmallow==3.26.1
+    # via dataclasses-json
+mcp==1.6.0
+    # via chainlit
+monotonic==1.6
+    # via posthog
+multidict==6.4.3
+    # via
+    #   aiohttp
+    #   yarl
+mypy-extensions==1.0.0
+    # via typing-inspect
+nest-asyncio==1.6.0
+    # via chainlit
+numpy==2.2.4
+    # via langchain-community
+openai==1.75.0
+    # via langchain-openai
+opentelemetry-api==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-alephalpha
+    #   opentelemetry-instrumentation-anthropic
+    #   opentelemetry-instrumentation-bedrock
+    #   opentelemetry-instrumentation-chromadb
+    #   opentelemetry-instrumentation-cohere
+    #   opentelemetry-instrumentation-crewai
+    #   opentelemetry-instrumentation-google-generativeai
+    #   opentelemetry-instrumentation-groq
+    #   opentelemetry-instrumentation-haystack
+    #   opentelemetry-instrumentation-lancedb
+    #   opentelemetry-instrumentation-langchain
+    #   opentelemetry-instrumentation-llamaindex
+    #   opentelemetry-instrumentation-logging
+    #   opentelemetry-instrumentation-marqo
+    #   opentelemetry-instrumentation-milvus
+    #   opentelemetry-instrumentation-mistralai
+    #   opentelemetry-instrumentation-ollama
+    #   opentelemetry-instrumentation-openai
+    #   opentelemetry-instrumentation-pinecone
+    #   opentelemetry-instrumentation-qdrant
+    #   opentelemetry-instrumentation-replicate
+    #   opentelemetry-instrumentation-requests
+    #   opentelemetry-instrumentation-sagemaker
+    #   opentelemetry-instrumentation-sqlalchemy
+    #   opentelemetry-instrumentation-threading
+    #   opentelemetry-instrumentation-together
+    #   opentelemetry-instrumentation-transformers
+    #   opentelemetry-instrumentation-urllib3
+    #   opentelemetry-instrumentation-vertexai
+    #   opentelemetry-instrumentation-watsonx
+    #   opentelemetry-instrumentation-weaviate
+    #   opentelemetry-sdk
+    #   opentelemetry-semantic-conventions
+    #   traceloop-sdk
+    #   uptrace
+opentelemetry-exporter-otlp==1.31.1
+    # via uptrace
+opentelemetry-exporter-otlp-proto-common==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+opentelemetry-exporter-otlp-proto-grpc==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp
+    #   traceloop-sdk
+opentelemetry-exporter-otlp-proto-http==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp
+    #   traceloop-sdk
+opentelemetry-instrumentation==0.52b1
+    # via
+    #   opentelemetry-instrumentation-alephalpha
+    #   opentelemetry-instrumentation-anthropic
+    #   opentelemetry-instrumentation-bedrock
+    #   opentelemetry-instrumentation-chromadb
+    #   opentelemetry-instrumentation-cohere
+    #   opentelemetry-instrumentation-crewai
+    #   opentelemetry-instrumentation-google-generativeai
+    #   opentelemetry-instrumentation-groq
+    #   opentelemetry-instrumentation-haystack
+    #   opentelemetry-instrumentation-lancedb
+    #   opentelemetry-instrumentation-langchain
+    #   opentelemetry-instrumentation-llamaindex
+    #   opentelemetry-instrumentation-logging
+    #   opentelemetry-instrumentation-marqo
+    #   opentelemetry-instrumentation-milvus
+    #   opentelemetry-instrumentation-mistralai
+    #   opentelemetry-instrumentation-ollama
+    #   opentelemetry-instrumentation-openai
+    #   opentelemetry-instrumentation-pinecone
+    #   opentelemetry-instrumentation-qdrant
+    #   opentelemetry-instrumentation-replicate
+    #   opentelemetry-instrumentation-requests
+    #   opentelemetry-instrumentation-sagemaker
+    #   opentelemetry-instrumentation-sqlalchemy
+    #   opentelemetry-instrumentation-threading
+    #   opentelemetry-instrumentation-together
+    #   opentelemetry-instrumentation-transformers
+    #   opentelemetry-instrumentation-urllib3
+    #   opentelemetry-instrumentation-vertexai
+    #   opentelemetry-instrumentation-watsonx
+    #   opentelemetry-instrumentation-weaviate
+    #   uptrace
+opentelemetry-instrumentation-alephalpha==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-anthropic==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-bedrock==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-chromadb==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-cohere==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-crewai==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-google-generativeai==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-groq==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-haystack==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-lancedb==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-langchain==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-llamaindex==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-logging==0.52b1
+    # via traceloop-sdk
+opentelemetry-instrumentation-marqo==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-milvus==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-mistralai==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-ollama==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-openai==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-pinecone==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-qdrant==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-replicate==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-requests==0.52b1
+    # via traceloop-sdk
+opentelemetry-instrumentation-sagemaker==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-sqlalchemy==0.52b1
+    # via traceloop-sdk
+opentelemetry-instrumentation-threading==0.52b1
+    # via traceloop-sdk
+opentelemetry-instrumentation-together==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-transformers==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-urllib3==0.52b1
+    # via traceloop-sdk
+opentelemetry-instrumentation-vertexai==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-watsonx==0.39.2
+    # via traceloop-sdk
+opentelemetry-instrumentation-weaviate==0.39.2
+    # via traceloop-sdk
+opentelemetry-proto==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp-proto-common
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+opentelemetry-sdk==1.31.1
+    # via
+    #   opentelemetry-exporter-otlp-proto-grpc
+    #   opentelemetry-exporter-otlp-proto-http
+    #   traceloop-sdk
+    #   uptrace
+opentelemetry-semantic-conventions==0.52b1
+    # via
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-alephalpha
+    #   opentelemetry-instrumentation-anthropic
+    #   opentelemetry-instrumentation-bedrock
+    #   opentelemetry-instrumentation-chromadb
+    #   opentelemetry-instrumentation-cohere
+    #   opentelemetry-instrumentation-crewai
+    #   opentelemetry-instrumentation-google-generativeai
+    #   opentelemetry-instrumentation-groq
+    #   opentelemetry-instrumentation-haystack
+    #   opentelemetry-instrumentation-lancedb
+    #   opentelemetry-instrumentation-langchain
+    #   opentelemetry-instrumentation-llamaindex
+    #   opentelemetry-instrumentation-marqo
+    #   opentelemetry-instrumentation-milvus
+    #   opentelemetry-instrumentation-mistralai
+    #   opentelemetry-instrumentation-ollama
+    #   opentelemetry-instrumentation-openai
+    #   opentelemetry-instrumentation-pinecone
+    #   opentelemetry-instrumentation-qdrant
+    #   opentelemetry-instrumentation-replicate
+    #   opentelemetry-instrumentation-requests
+    #   opentelemetry-instrumentation-sagemaker
+    #   opentelemetry-instrumentation-sqlalchemy
+    #   opentelemetry-instrumentation-together
+    #   opentelemetry-instrumentation-transformers
+    #   opentelemetry-instrumentation-urllib3
+    #   opentelemetry-instrumentation-vertexai
+    #   opentelemetry-instrumentation-watsonx
+    #   opentelemetry-instrumentation-weaviate
+    #   opentelemetry-sdk
+opentelemetry-semantic-conventions-ai==0.4.3
+    # via
+    #   opentelemetry-instrumentation-alephalpha
+    #   opentelemetry-instrumentation-anthropic
+    #   opentelemetry-instrumentation-bedrock
+    #   opentelemetry-instrumentation-chromadb
+    #   opentelemetry-instrumentation-cohere
+    #   opentelemetry-instrumentation-crewai
+    #   opentelemetry-instrumentation-google-generativeai
+    #   opentelemetry-instrumentation-groq
+    #   opentelemetry-instrumentation-haystack
+    #   opentelemetry-instrumentation-lancedb
+    #   opentelemetry-instrumentation-langchain
+    #   opentelemetry-instrumentation-llamaindex
+    #   opentelemetry-instrumentation-marqo
+    #   opentelemetry-instrumentation-milvus
+    #   opentelemetry-instrumentation-mistralai
+    #   opentelemetry-instrumentation-ollama
+    #   opentelemetry-instrumentation-openai
+    #   opentelemetry-instrumentation-pinecone
+    #   opentelemetry-instrumentation-qdrant
+    #   opentelemetry-instrumentation-replicate
+    #   opentelemetry-instrumentation-sagemaker
+    #   opentelemetry-instrumentation-together
+    #   opentelemetry-instrumentation-transformers
+    #   opentelemetry-instrumentation-vertexai
+    #   opentelemetry-instrumentation-watsonx
+    #   opentelemetry-instrumentation-weaviate
+    #   traceloop-sdk
+opentelemetry-util-http==0.52b1
+    # via
+    #   opentelemetry-instrumentation-requests
+    #   opentelemetry-instrumentation-urllib3
+orjson==3.10.16
+    # via
+    #   langgraph-sdk
+    #   langsmith
+ormsgpack==1.9.1
+    # via langgraph-checkpoint
+packaging==24.2
+    # via
+    #   chainlit
+    #   huggingface-hub
+    #   langchain-core
+    #   langsmith
+    #   literalai
+    #   marshmallow
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-sqlalchemy
+posthog==3.25.0
+    # via traceloop-sdk
+primp==0.15.0
+    # via duckduckgo-search
+propcache==0.3.1
+    # via
+    #   aiohttp
+    #   yarl
+protobuf==5.29.4
+    # via
+    #   googleapis-common-protos
+    #   opentelemetry-proto
+pydantic==2.11.3
+    # via
+    #   tool-calling-agent (pyproject.toml)
+    #   anthropic
+    #   chainlit
+    #   fastapi
+    #   langchain
+    #   langchain-core
+    #   langsmith
+    #   literalai
+    #   mcp
+    #   openai
+    #   pydantic-settings
+    #   traceloop-sdk
+pydantic-core==2.33.1
+    # via pydantic
+pydantic-settings==2.9.1
+    # via
+    #   langchain-community
+    #   mcp
+pyjwt==2.10.1
+    # via chainlit
+python-dateutil==2.9.0.post0
+    # via posthog
+python-dotenv==1.1.0
+    # via
+    #   tool-calling-agent (pyproject.toml)
+    #   chainlit
+    #   pydantic-settings
+python-engineio==4.12.0
+    # via python-socketio
+python-multipart==0.0.18
+    # via chainlit
+python-socketio==5.13.0
+    # via chainlit
+pyyaml==6.0.2
+    # via
+    #   huggingface-hub
+    #   langchain
+    #   langchain-community
+    #   langchain-core
+regex==2024.11.6
+    # via tiktoken
+requests==2.32.3
+    # via
+    #   arxiv
+    #   huggingface-hub
+    #   langchain
+    #   langchain-community
+    #   langsmith
+    #   opentelemetry-exporter-otlp-proto-http
+    #   posthog
+    #   requests-toolbelt
+    #   tavily-python
+    #   tiktoken
+requests-toolbelt==1.0.0
+    # via langsmith
+sgmllib3k==1.0.0
+    # via feedparser
+simple-websocket==1.1.0
+    # via python-engineio
+six==1.17.0
+    # via
+    #   posthog
+    #   python-dateutil
+sniffio==1.3.1
+    # via
+    #   anthropic
+    #   anyio
+    #   openai
+sqlalchemy==2.0.40
+    # via
+    #   langchain
+    #   langchain-community
+sse-starlette==2.2.1
+    # via mcp
+starlette==0.41.3
+    # via
+    #   chainlit
+    #   fastapi
+    #   mcp
+    #   sse-starlette
+syncer==2.0.3
+    # via chainlit
+tavily-python==0.5.4
+    # via tool-calling-agent (pyproject.toml)
+tenacity==9.1.2
+    # via
+    #   langchain-community
+    #   langchain-core
+    #   traceloop-sdk
+tiktoken==0.9.0
+    # via
+    #   langchain-openai
+    #   opentelemetry-instrumentation-openai
+    #   tavily-python
+tokenizers==0.21.1
+    # via opentelemetry-instrumentation-bedrock
+tomli==2.2.1
+    # via chainlit
+tqdm==4.67.1
+    # via
+    #   huggingface-hub
+    #   openai
+traceloop-sdk==0.39.2
+    # via literalai
+typing-extensions==4.13.2
+    # via
+    #   anthropic
+    #   anyio
+    #   fastapi
+    #   huggingface-hub
+    #   langchain-core
+    #   openai
+    #   opentelemetry-sdk
+    #   pydantic
+    #   pydantic-core
+    #   sqlalchemy
+    #   typing-inspect
+    #   typing-inspection
+typing-inspect==0.9.0
+    # via dataclasses-json
+typing-inspection==0.4.0
+    # via
+    #   pydantic
+    #   pydantic-settings
+uptrace==1.31.0
+    # via chainlit
+urllib3==2.4.0
+    # via requests
+uvicorn==0.34.1
+    # via
+    #   chainlit
+    #   mcp
+watchfiles==0.20.0
+    # via chainlit
+wrapt==1.17.2
+    # via
+    #   deprecated
+    #   opentelemetry-instrumentation
+    #   opentelemetry-instrumentation-sqlalchemy
+    #   opentelemetry-instrumentation-threading
+    #   opentelemetry-instrumentation-urllib3
+wsproto==1.2.0
+    # via simple-websocket
+xxhash==3.5.0
+    # via langgraph
+yarl==1.20.0
+    # via aiohttp
+zipp==3.21.0
+    # via importlib-metadata
+zstandard==0.23.0
+    # via langsmith

tools.py

@@ -0,0 +1,53 @@
+"""
+Module for configuring and creating tools for the LangGraph agent.
+"""
+from typing import Dict, Any, List, Tuple
+
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_community.tools.arxiv.tool import ArxivQueryRun
+from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_openai import ChatOpenAI
+
+
+def create_tools(config: Dict[str, Any]) -> Tuple[List, ChatOpenAI]:
+    """
+    Create LangChain tools and model for use in the agent.
+    
+    Args:
+        config: Configuration dictionary
+        
+    Returns:
+        Tuple containing:
+        - List of tools 
+        - ChatOpenAI model
+    """
+    # Initialize Tavily search tool
+    tavily_tool = TavilySearchResults(
+        max_results=config["tavily_max_results"],
+        api_key=config["tavily_api_key"]
+    )
+    
+    # Initialize ArXiv tool
+    arxiv_tool = ArxivQueryRun()
+
+    # Initialize DuckDuckGo search tool
+    duckduckgo_tool = DuckDuckGoSearchRun()
+    
+    # Create the tool belt
+    tool_belt = [
+        tavily_tool,
+        arxiv_tool,
+        duckduckgo_tool,
+    ]
+    
+    # Initialize the OpenAI model
+    model = ChatOpenAI(
+        model=config["model_name"], 
+        temperature=config["temperature"],
+        api_key=config["openai_api_key"]
+    )
+    
+    # Bind tools to the model
+    model_with_tools = model.bind_tools(tool_belt)
+    
+    return tool_belt, model_with_tools