Update app.py

app.py

@@ -9,7 +9,7 @@ import json
 from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
 import wikipedia
 import torch
-from tooling import ModelMathTool
+from tooling import ModelMathTool, WikiTool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -95,7 +95,7 @@ class BasicAgent:
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
             model=model,
-            tools=[web_search,python_interpreter,visit_webpage_tool], 
+            tools=[web_search,python_interpreter,visit_webpage_tool,wiki_tool], 
             max_steps=5,
             verbosity_level=1,
             grammar=None,