Yasu777 commited on
Commit
1c1a3cc
·
1 Parent(s): 3e335b2

Update third.py

Browse files
Files changed (1) hide show
  1. third.py +1 -1
third.py CHANGED
@@ -23,7 +23,7 @@ async def main(editable_output2, keyword_id):
23
  tools = [
24
  Tool(
25
  name="Search",
26
- func=lambda input_dict: search.run(input_dict["query"]),
27
  description="useful for when you need to answer questions about current events"
28
  ),
29
  ]
 
23
  tools = [
24
  Tool(
25
  name="Search",
26
+ func=lambda input_dict: search.run(input_dict) if isinstance(input_dict, str) else search.run(input_dict["query"]),
27
  description="useful for when you need to answer questions about current events"
28
  ),
29
  ]