Update agentic.py

agentic.py

@@ -288,7 +288,7 @@ search_tools = [
 
 download_file_tool = [ download_input_file ]
 
-web_search_node_agent = big_model.bind_tools(search_tools, parallel_tool_calls=False)
+web_search_node_agent = general_model.bind_tools(search_tools, parallel_tool_calls=False)
 
 def thinking_node(state: State) -> dict:
     """
@@ -371,7 +371,7 @@ Now provide your response immediately without any preamble in text but not in ma
 
     sys_msg = SystemMessage(content=prompt)
 
-    code_node_response = [big_model.invoke([sys_msg])]
+    code_node_response = [general_model.invoke([sys_msg])]
 
     code_node_response[-1].pretty_print()
 
@@ -919,8 +919,8 @@ if __name__ == "__main__":
 
     agent_graph = build_graph()
 
-    with open("image.png", "wb") as png:
-        png.write(agent_graph.get_graph(xray=True).draw_mermaid_png())
+    # with open("image.png", "wb") as png:
+    #     png.write(agent_graph.get_graph(xray=True).draw_mermaid_png(draw_method=MermaidDrawMethod.PYPPETEER))
     
     # print(vision_node.__doc__)
 
@@ -928,10 +928,9 @@ if __name__ == "__main__":
         json_responses = json.loads(responses.read())
 
     json_questions =  [{
-        "task_id": "840bfca7-4f7b-481a-8794-c560c340185d",
-        "question": "On June 6, 2023, an article by Carolyn Collins Petersen was published in Universe Today. This article mentions a team that produced a paper about their observations, linked at the bottom of the article. Find this paper. Under what NASA award number was the work performed by R. G. Arendt supported by?",
-        "Level": "1",
-        "file_name": ""
+        "question": "The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.",
+        "file_name": "7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx", 
+        "task_id": "7bd855d8-463d-4ed5-93ca-5fe35145f733"
     }]
 
     # with open("questions.json", "r") as questions: