removed async calls and WORKS

app.py

@@ -14,9 +14,6 @@ from langchain.agents import ZeroShotAgent, AgentExecutor
 from langchain.chat_models import ChatOpenAI
 from langchain import LLMChain
 
-import os
-os.environ["OPENAI_API_KEY"] = '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'
-
 @cl.author_rename
 def rename(orig_author: str):
     rename_dict = {"RetrievalQA": "Consulting The Barbenheimer"}
@@ -193,7 +190,7 @@ async def init():
         tools=tools,
         verbose=True )
     # create execution agent 
-    barbenheimer_agent_chain = await cl.make_async(AgentExecutor.from_agent_and_tools)(
+    barbenheimer_agent_chain = AgentExecutor.from_agent_and_tools(
         agent=barbenheimer_agent,
         tools=tools, 
         verbose=True )
@@ -210,36 +207,35 @@ async def main(message):
     # await msg.send()
     
     chain = cl.user_session.get("chain")
-    cb = cl.AsyncLangchainCallbackHandler(
+    cb = cl.LangchainCallbackHandler(
         stream_final_answer=False, answer_prefix_tokens=["FINAL", "ANSWER"]
     )
     cb.answer_reached = True
-    res = await chain.acall(message, callbacks=[cb], )
+    res = chain.__call__(message, callbacks=[cb], )
     
-    # msg = cl.Message(content=f"{res}")
-    # await msg.send()
+    # print(res.keys()) # keys are "input" and "output"
     
-    answer = res["result"]
+    answer = res["output"]
     source_elements = []
-    visited_sources = set()
-
-    # Get the documents from the user session
-    docs = res["source_documents"]
-    metadatas = [doc.metadata for doc in docs]
-    all_sources = [m["source"] for m in metadatas]
-
-    for source in all_sources:
-        if source in visited_sources:
-            continue
-        visited_sources.add(source)
-        # Create the text element referenced in the message
-        source_elements.append(
-            cl.Text(content="https://www.imdb.com" + source, name="Review URL")
-        )
-
-    if source_elements:
-        answer += f"\nSources: {', '.join([e.content.decode('utf-8') for e in source_elements])}"
-    else:
-        answer += "\nNo sources found"
+    # visited_sources = set()
+
+    # # Get the documents from the user session
+    # docs = res["source_documents"]
+    # metadatas = [doc.metadata for doc in docs]
+    # all_sources = [m["source"] for m in metadatas]
+
+    # for source in all_sources:
+    #     if source in visited_sources:
+    #         continue
+    #     visited_sources.add(source)
+    #     # Create the text element referenced in the message
+    #     source_elements.append(
+    #         cl.Text(content="https://www.imdb.com" + source, name="Review URL")
+    #     )
+
+    # if source_elements:
+    #     answer += f"\nSources: {', '.join([e.content.decode('utf-8') for e in source_elements])}"
+    # else:
+    #     answer += "\nNo sources found"
 
     await cl.Message(content=answer, elements=source_elements).send()

requirements.txt

@@ -3,4 +3,5 @@ langchain==0.0.265
 tiktoken==0.4.0
 openai==0.27.8
 faiss-cpu==1.7.4
-wikipedia==1.4.0
+wikipedia==1.4.0
+rank_bm25