Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool, MultiStepAgent
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool
 from smolagents.agents import PromptTemplates
 
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
@@ -58,11 +58,6 @@ model = OpenAIServerModel(
 # """
 # )
 
-class CustomMultiStepAgent(MultiStepAgent):
-    def initialize_system_prompt(self):
-        # Implement the method as required
-        pass
-
 prompt = {
     'system_prompt':"""
 You are a helpful assistant tasked with answering questions using a set of tools.
@@ -116,11 +111,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        self.agent = CustomMultiStepAgent(
+        self.agent = CodeAgent(
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             model = model,
             add_base_tools=True,
-            prompt_templates = prompt
+            system_prompt = prompt
         )
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str: