Upload agent.py

agent.py

@@ -9,6 +9,8 @@ from langchain_community.document_loaders import WikipediaLoader
 from langchain_core.messages import HumanMessage
 from langchain_core.tools import tool
 from langchain_mistralai import ChatMistralAI
+from langchain_groq import ChatGroq
+
 
 load_dotenv()
 
@@ -107,11 +109,13 @@ tools = [
 def build_graph():
 
     
-    llm = ChatMistralAI(
+    '''llm = ChatMistralAI(
             model="magistral-small-2506",
             temperature=0,
             max_retries=2,
-        )
+        )'''
+    
+    llm = ChatGroq(model="qwen-qwq-32b", temperature=0) # optional : qwen-qwq-32b gemma2-9b-it
 
     llm_with_tools = llm.bind_tools(tools)
 
@@ -138,7 +142,7 @@ def build_graph():
 if __name__=="__main__":
     #system_prompt = "You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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."
     system_prompt = "You are a general AI assistant. I will ask you a question. Give your answer with the following template: only show your final answer. Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. 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."
-    question = "when was new the appolo expedition?"
+    question = "when was the Eiffel tower built?"
     question = f"{system_prompt} \
     The question: \
     {question} \