dlaima commited on
Commit
2a40e49
·
verified ·
1 Parent(s): 77a0c13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -36,8 +36,7 @@ class WikipediaSearchTool(Tool):
36
  super().__init__()
37
  self.searcher = WikipediaSearcher()
38
 
39
- def __call__(self, inputs: dict) -> str:
40
- query = inputs.get("query", "")
41
  return self.searcher.search(query)
42
 
43
  wikipedia_search_tool = WikipediaSearchTool()
 
36
  super().__init__()
37
  self.searcher = WikipediaSearcher()
38
 
39
+ def forward(self, query: str) -> str:
 
40
  return self.searcher.search(query)
41
 
42
  wikipedia_search_tool = WikipediaSearchTool()