Thanh Vinh Vo commited on
Commit
b48b971
·
1 Parent(s): 9150700
Files changed (2) hide show
  1. app.py +4 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -39,8 +39,10 @@ class BasicAgent:
39
  def __call__(self, question: str) -> str:
40
  print(f"Agent received question: {question}")
41
  prompt = f"""
42
- Answer the following question: `{question}`. Please follow following rules:
43
- 1. wikipedia Python package is provided to extract data from Wikipedia pages, please use it when possible.
 
 
44
  """
45
  result = self.agent.run(prompt)
46
  print(f"Agent responded with: {result}")
 
39
  def __call__(self, question: str) -> str:
40
  print(f"Agent received question: {question}")
41
  prompt = f"""
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}")
requirements.txt CHANGED
@@ -4,4 +4,4 @@ smolagents
4
  duckduckgo_search
5
  bs4
6
  markdownify
7
- wikipedia
 
4
  duckduckgo_search
5
  bs4
6
  markdownify
7
+ wikipedia-api