Update agent.py

agent.py

@@ -1,3 +1,4 @@
+import os
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition
 from langgraph.prebuilt import ToolNode
@@ -12,6 +13,7 @@ from langchain_core.tools import tool
 from langchain_google_genai import ChatGoogleGenerativeAI
 
 #load_dotenv()
+google_api_key = os.environ["GOOGLE_API_KEY"]
 
 @tool
 def add(a: int, b: int) -> int:
@@ -96,7 +98,7 @@ tools = [
 
 
 def build_graph(provider: str = "google"):
-  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0, api_key=google_api_key)
 
   llm_with_tools = llm.bind_tools(tools)