Update agent.py

agent.py

@@ -1,5 +1,4 @@
 import os
-from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_groq import ChatGroq
 from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition
@@ -8,10 +7,10 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from tools import *
 from typing import TypedDict
 
-tools = [add, multiply, divide, subtract]
+tools = [add, multiply, divide, subtract, wikipedia_search_tool]
 
 def build_agent():
-    llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+    llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
     chat_with_tools = llm.bind_tools(tools)
 
     def assistant(state: MessagesState):