Kkanittha commited on
Commit
55d089a
·
verified ·
1 Parent(s): dd8f671

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
7
  #from langchain_community.llms import HuggingFaceEndpoint
8
  #from langchain.agents import initialize_agent, Tool
9
  #from langchain.utilities import DuckDuckGoSearchAPIWrapper
@@ -32,7 +32,13 @@ class BasicAgent:
32
  print(answer)'''
33
 
34
  # Add tools
35
- tools = [DuckDuckGoSearchTool()]
 
 
 
 
 
 
36
 
37
  '''query = "Who won the 2024 Olympic gold in swimming?"
38
 
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, WikipediaSearchTool, InferenceClientModel
7
  #from langchain_community.llms import HuggingFaceEndpoint
8
  #from langchain.agents import initialize_agent, Tool
9
  #from langchain.utilities import DuckDuckGoSearchAPIWrapper
 
32
  print(answer)'''
33
 
34
  # Add tools
35
+ tools = [WikipediaSearchTool(
36
+ user_agent="MyResearchBot (kanittha.kra@gmail.com)",
37
+ language="en",
38
+ content_type="summary", # or "text"
39
+ extract_format="WIKI",
40
+ )
41
+ ]
42
 
43
  '''query = "Who won the 2024 Olympic gold in swimming?"
44