Thanh Vinh Vo commited on
Commit
ae5b170
·
1 Parent(s): 2326b05
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -32,7 +32,12 @@ class BasicAgent:
32
  DuckDuckGoSearchTool(),
33
  ],
34
  model=InferenceClientModel(),
35
- additional_authorized_imports=["requests", "bs4", "markdownify"],
 
 
 
 
 
36
  max_steps=20,
37
  )
38
 
@@ -42,7 +47,7 @@ class BasicAgent:
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}")
 
32
  DuckDuckGoSearchTool(),
33
  ],
34
  model=InferenceClientModel(),
35
+ additional_authorized_imports=[
36
+ "requests",
37
+ "bs4",
38
+ "markdownify",
39
+ "wikipedia",
40
+ ],
41
  max_steps=20,
42
  )
43
 
 
47
  Answer the following question: `{question}`.
48
 
49
  Please follow following rules:
50
+ 1. `wikipedia` Python package is provided to extract data from Wikipedia, please use it when possible.
51
  """
52
  result = self.agent.run(prompt)
53
  print(f"Agent responded with: {result}")