Thanh Vinh Vo commited on
Commit
f892bb9
·
1 Parent(s): c79c7c4
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -203,7 +203,7 @@ class BasicAgent:
203
  )
204
 
205
  self.code_agent = CodeAgent(
206
- tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text, extract_table_from_html, WikipediaSearchTool()],
207
  model=InferenceClientModel(
208
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
209
  temperature=0.0,
@@ -239,7 +239,7 @@ class BasicAgent:
239
  This agent follows rules below:
240
  1. Take the question literally! Do not add any additional information or assumptions.
241
  2. `wikipedia` Python library is provided that makes it easy to to interact with Wikipedia pages.
242
- 3. `extract_table_from_html` tool is provided that makes it easy to extract tables from Wikipedia HTML pages.
243
  4. Only use BeautifulSoup to parse HTML at last resort!
244
  """,
245
  verbosity_level=0,
@@ -288,7 +288,6 @@ class BasicAgent:
288
  "{question}""{file}"
289
  Please follow rules below:
290
  1. Take the question literally! Do not add any additional information or assumptions.
291
- 2. Only use BeautifulSoup to parse HTML at last resort!
292
  """
293
  result = self.manager_agent.run(prompt)
294
  print(f"Agent responded with: {result}")
 
203
  )
204
 
205
  self.code_agent = CodeAgent(
206
+ tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text, WikipediaSearchTool()],
207
  model=InferenceClientModel(
208
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
209
  temperature=0.0,
 
239
  This agent follows rules below:
240
  1. Take the question literally! Do not add any additional information or assumptions.
241
  2. `wikipedia` Python library is provided that makes it easy to to interact with Wikipedia pages.
242
+ 3. `pandas` tool is provided that makes it easy to extract table data from Wikipedia HTML pages.
243
  4. Only use BeautifulSoup to parse HTML at last resort!
244
  """,
245
  verbosity_level=0,
 
288
  "{question}""{file}"
289
  Please follow rules below:
290
  1. Take the question literally! Do not add any additional information or assumptions.
 
291
  """
292
  result = self.manager_agent.run(prompt)
293
  print(f"Agent responded with: {result}")