Thanh Vinh Vo commited on
Commit
7fc9205
·
1 Parent(s): 04a1a34
Files changed (2) hide show
  1. app.py +3 -1
  2. requirements.txt +1 -0
app.py CHANGED
@@ -37,6 +37,7 @@ class BasicAgent:
37
  "bs4",
38
  "markdownify",
39
  "wikipedia",
 
40
  ],
41
  max_steps=20,
42
  )
@@ -47,7 +48,8 @@ class BasicAgent:
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. If extraction fails, please try to manually inspect the page!
 
51
  """
52
  result = self.agent.run(prompt)
53
  print(f"Agent responded with: {result}")
 
37
  "bs4",
38
  "markdownify",
39
  "wikipedia",
40
+ "pandas"
41
  ],
42
  max_steps=20,
43
  )
 
48
  Answer the following question: `{question}`.
49
 
50
  Please follow following rules:
51
+ 1. `wikipedia` Python package is provided to interact with Wikipedia.
52
+ 2. `pandas` Python package is provided. It can be used to extract read data from HTML content.
53
  """
54
  result = self.agent.run(prompt)
55
  print(f"Agent responded with: {result}")
requirements.txt CHANGED
@@ -5,3 +5,4 @@ duckduckgo_search
5
  bs4
6
  markdownify
7
  wikipedia
 
 
5
  bs4
6
  markdownify
7
  wikipedia
8
+ pandas