Update main.py

main.py

@@ -12,7 +12,7 @@ from operator import itemgetter
 from pinecone import Pinecone
 from langchain_pinecone import PineconeVectorStore
 from langchain_community.chat_message_histories import ChatMessageHistory
-from langchain_google_community import GoogleSearchAPIWrapper
+#from langchain_google_community import GoogleSearchAPIWrapper
 from langchain.memory import ConversationBufferMemory
 from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableConfig, RunnableLambda
 from langchain.callbacks.base import BaseCallbackHandler
@@ -20,7 +20,7 @@ from langchain.chains import (
     StuffDocumentsChain, ConversationalRetrievalChain
 )
 from langchain_core.tracers.context import tracing_v2_enabled
-from langchain_core.tools import Tool
+#from langchain_core.tools import Tool
 
 import chainlit as cl
 from chainlit.input_widget import TextInput, Select, Switch, Slider
@@ -219,23 +219,23 @@ async def on_message(message: cl.Message):
         cl.Text(name="Sources", content=search[0], display="inline")
     ]
 
-    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)
-
-    tool = Tool(
-        name="Google Search Snippets",
-        description="Search Google for recent results.",
-        func=top5_results,
-    )
-    query = str(message.content)
-    ref_text = tool.run(query)
-    if 'Result' not in ref_text[0].keys():
-        print(ref_text)
-    else:
-        print('None')
+    #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)
+
+    #tool = Tool(
+    #    name="Google Search Snippets",
+    #    description="Search Google for recent results.",
+    #    func=top5_results,
+    #)
+    #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 : ",