Thanh Vinh Vo commited on
Commit
2326b05
·
1 Parent(s): ba7db54
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ class BasicAgent:
32
  DuckDuckGoSearchTool(),
33
  ],
34
  model=InferenceClientModel(),
35
- additional_authorized_imports=["requests", "bs4", "wikipediaapi"],
36
  max_steps=20,
37
  )
38
 
@@ -42,7 +42,7 @@ class BasicAgent:
42
  Answer the following question: `{question}`.
43
 
44
  Please follow following rules:
45
- 1. `wikipedia-api` Python package is provided to extract data from Wikipedia pages, please use it when possible.
46
  """
47
  result = self.agent.run(prompt)
48
  print(f"Agent responded with: {result}")
 
32
  DuckDuckGoSearchTool(),
33
  ],
34
  model=InferenceClientModel(),
35
+ additional_authorized_imports=["requests", "bs4", "markdownify"],
36
  max_steps=20,
37
  )
38
 
 
42
  Answer the following question: `{question}`.
43
 
44
  Please follow following rules:
45
+ 1. `markdownify` Python package is provided to extract data from webpage, please use it to clean up HTML content before processing.
46
  """
47
  result = self.agent.run(prompt)
48
  print(f"Agent responded with: {result}")