Update agent.py

agent.py

@@ -1,6 +1,5 @@
 """LangGraph Agent (No Supabase)"""
 import os
-from dotenv import load_dotenv
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_google_genai import ChatGoogleGenerativeAI
@@ -11,38 +10,30 @@ from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 
-load_dotenv()
-
 @tool
 def multiply(a: int, b: int) -> int:
-    """Multiply two integers and return the result."""
     return a * b
 
 @tool
 def add(a: int, b: int) -> int:
-    """Add two integers and return the result."""
     return a + b
 
 @tool
 def subtract(a: int, b: int) -> int:
-    """Subtract b from a and return the result."""
     return a - b
 
 @tool
 def divide(a: int, b: int) -> float:
-    """Divide a by b and return the result."""
     if b == 0:
         raise ValueError("Cannot divide by zero.")
     return a / b
 
 @tool
 def modulus(a: int, b: int) -> int:
-    """Return the modulus of a and b."""
     return a % b
 
 @tool
 def wiki_search(query: str) -> dict:
-    """Search Wikipedia for a query and return up to 2 results."""
     search_docs = WikipediaLoader(query=query, load_max_docs=2).load()
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content}\n</Document>" for doc in search_docs
@@ -51,7 +42,6 @@ def wiki_search(query: str) -> dict:
 
 @tool
 def web_search(query: str) -> dict:
-    """Search the web via Tavily and return up to 3 results."""
     search_docs = TavilySearchResults(max_results=3).invoke(query=query)
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content}\n</Document>" for doc in search_docs
@@ -60,7 +50,6 @@ def web_search(query: str) -> dict:
 
 @tool
 def arvix_search(query: str) -> dict:
-    """Search Arxiv and return up to 3 truncated results."""
     search_docs = ArxivLoader(query=query, load_max_docs=3).load()
     results = "\n\n---\n\n".join(
         f"<Document>\n{doc.page_content[:500]}\n</Document>" for doc in search_docs
@@ -83,7 +72,10 @@ def build_graph(provider: str = "groq"):
     if provider == "google":
         llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
     elif provider == "groq":
-        llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
+        groq_api_key = os.environ.get("GROQ_API_KEY")
+        if not groq_api_key:
+            raise ValueError("GROQ_API_KEY is not set in the environment.")
+        llm = ChatGroq(model="qwen-qwq-32b", temperature=0, api_key=groq_api_key)
     elif provider == "huggingface":
         llm = ChatHuggingFace(
             llm=HuggingFaceEndpoint(