Update agent.py

agent.py

@@ -7,7 +7,7 @@ from langchain_core.messages import SystemMessage, HumanMessage
 from tools import *
 from typing import TypedDict
 
-tools = [add, multiply, divide, subtract, get_studio_album_count]
+tools = [add, multiply, divide, subtract, search_wikipedia]
 
 def build_agent():
     llm = ChatGroq(model="qwen-qwq-32b", temperature=0)