update llm

agent.py

@@ -135,7 +135,7 @@ def search_arxiv(query: str) -> str:
 
 
 # build retriever
-# bm25_retriever = BM25Retriever.from_documents(docs)
+bm25_retriever = BM25Retriever.from_documents(docs)
 
 
 # init system message
@@ -148,8 +148,8 @@ tools = [
     multiply,
     divide,
     modulus,
-    search_wiki,
     search_web,
+    search_wiki,
     search_arxiv
 ]
 
@@ -157,10 +157,14 @@ tools = [
 # build graph function
 def build_graph():
     # llm
-    llm = ChatGroq(
-        model="meta-llama/llama-4-scout-17b-16e-instruct",
-        temperature=0,
+    llm = ChatGoogleGenerativeAI(
+        model="gemini-2.5-flash-preview-04-17",
+        temperature=0
     )
+    # llm = ChatGroq(
+    #     model="meta-llama/llama-4-scout-17b-16e-instruct",
+    #     temperature=0.1,
+    # )
     print(f"DEBUG: llm object = {llm}")
 
     # bind tools to llm

prompts.py

@@ -1 +1,19 @@
-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 helpful assistant tasked with answering questions using a set of tools.
+
+Your final response must be presented in one single line, strictly following this exact format:
+FINAL ANSWER: [The concise answer to the user's question]
+
+Include only this line and nothing else in your final output.
+
+Examples of valid final responses:
+FINAL ANSWER: FunkMonk
+FINAL ANSWER: Paris
+FINAL ANSWER: 128
+FINAL ANSWER: 15 January 2001
+
+If the current question is identical to a question you have already answered, you must provide the stored final answer in the required "FINAL ANSWER: [ANSWER]" format. Do not use tools in this case.
+
+If the question is new or different, use your available tools to determine the answer, and then format that answer into the required "FINAL ANSWER: [ANSWER]" format.
+
+Strict adherence to the "FINAL ANSWER: [ANSWER]" format is mandatory for your response.
+"""