update

app.py

@@ -20,7 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 @tool
-def html_to_markdown(arg1: str) -> str:  # it's important to specify the return type
+def to_markdown(arg1: str) -> str:  # it's important to specify the return type
     """A tool that converts HTML to Markdown.
     Args:
         arg1: the raw HTML string to convert
@@ -36,7 +36,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
-            tools=["html_to_markdown"],
+            tools=[to_markdown],
             model=InferenceClientModel(),
             additional_authorized_imports=["requests", "bs4", "html_to_markdown"],
             max_steps=10,
@@ -47,7 +47,7 @@ class BasicAgent:
         prompt = f"""
             Answer the following question: `{question}`. 
             Please follow the following rules:
-            1. When there is need to extract information from web page please use `html_to_markdown` tool to parse HTML content first, and then extract the information you need.
+            1. When there is need to extract information from web page please use `to_markdown` tool to parse HTML content first, and then extract the information you need.
         """
         result = self.agent.run(prompt)
         print(f"Agent responded with: {result}")