Update main.py

main.py

@@ -217,15 +217,11 @@ async def on_message(message: cl.Message):
 
     os.environ["GOOGLE_CSE_ID"] = os.getenv('GOOGLE_CSE_ID')
     os.environ["GOOGLE_API_KEY"] = os.getenv('GOOGLE_API_KEY')
-    searchAPI = GoogleSearchAPIWrapper()
-
-    def top5_results(query):
-        return searchAPI.results(query, 5)
-
+    searchAPI = GoogleSearchAPIWrapper(k=10)
     tool = Tool(
-        name="Google Search Snippets",
+        name="google_search",
         description="Search Google for recent results.",
-        func=top5_results,
+        func=search.run,
     )
     print(tool.run(str(message.content)))