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, wikidata_search_tool]
+tools = [add, multiply, divide, subtract, get_studio_album_count]
 
 def build_agent():
     llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
@@ -28,7 +28,6 @@ def build_agent():
                             If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
                             If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
                             Your answer should only start with "FINAL ANSWER: ", then follows with the answer.
-                            Use the tool that is most appropiate on the task you are required to do. Whenever necessary, look for updated information through searches in the web, Wikipedia, Arxiv, etc.
                         """
         return {
             "messages": [sys_msg] + state["messages"]