Upload agent.py

agent.py

@@ -108,7 +108,7 @@ def build_graph():
 
     
     llm = ChatMistralAI(
-            model="mistral-small-2503",
+            model="open-mistral-nemo",
             temperature=0,
             max_retries=2,
         )
@@ -138,7 +138,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 kremlin built?"
+    question = "when was new yourk city founded?"
     question = f"{system_prompt} \
     The question: \
     {question} \