Update agent.py

agent.py

@@ -625,13 +625,14 @@ def combine_images(
 #with open("system_prompt.txt", "r", encoding="utf-8") as f:
 #   system_prompt = f.read()
 
-system_prompt = """You are a helpful assistant tasked with answering questions using a set of tools. 
+system_prompt = """
+You are a helpful assistant tasked with answering questions using a set of tools. 
 Now, 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.
-Your answer should only start with "FINAL ANSWER: ", then follows with the answer."""
+Your answer should only start with "FINAL ANSWER: ", then follows with the answer.""".strip()
     
 # System message
 sys_msg = SystemMessage(content=system_prompt)
@@ -667,17 +668,17 @@ tools = [
     wiki_search,
     web_search,
     arvix_search,
-    save_and_read_file,
-    download_file_from_url,
-    extract_text_from_image,
-    analyze_csv_file,
-    analyze_excel_file,
-    execute_code_multilang,
-    analyze_image,
-    transform_image,
-    draw_on_image,
-    generate_simple_image,
-    combine_images,
+    #save_and_read_file,
+    #download_file_from_url,
+    #extract_text_from_image,
+    #analyze_csv_file,
+    #analyze_excel_file,
+    #execute_code_multilang,
+    #analyze_image,
+    #transform_image,
+    #draw_on_image,
+    #generate_simple_image,
+    #combine_images,
 ]
 
 # Build graph function
@@ -714,7 +715,9 @@ def build_graph(provider: str = "huggingface"):
 
     elif provider == "google":
         # Google Gemini
-        llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+        #llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
+        llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash", temperature=0)
+        
     else:
         raise ValueError("Invalid provider. Choose 'huggingface'.")
         
@@ -751,8 +754,8 @@ def build_graph(provider: str = "huggingface"):
         "assistant",
         tools_condition,
     )
-    #builder.add_edge("tools", "retriever")
-    builder.add_edge("tools", "assistant")
+    builder.add_edge("tools", "retriever")
+    #builder.add_edge("tools", "assistant")
 
     # Compile graph
     return builder.compile()