Update main.py

main.py

@@ -217,14 +217,21 @@ 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(k=10)
+    searchAPI = GoogleSearchAPIWrapper()
+    def top5_results(query):
+        return search.results(query, 5)
+
     tool = Tool(
-        name="google_search",
+        name="Google Search Snippets",
         description="Search Google for recent results.",
-        func=search.run,
+        func=top5_results,
     )
-    print(tool.run(str(message.content)))
-
+    query = str(message.content)
+    ref_text = tool.run(query)
+    if 'Result' not in ref_text[0].keys():
+        print(ref_text)
+    else:
+        print('None')
     
     await cl.Message(
         content="Sources : ",