update

app.py

@@ -14,7 +14,7 @@ load_dotenv()
 # Add path to the root of the repo to the system path
 sys.path.append(".")
 
-from rag import retrieval_augmented_qa_pipeline
+from rag import _build_vector_db, retrieval_augmented_qa_pipeline, RetrievalAugmentedQAPipeline
 
 # ChatOpenAI Templates
 system_template = """You are a helpful assistant who always speaks in a pleasant tone!
@@ -48,11 +48,14 @@ async def main(message: cl.Message):
 
     print(message.content)
 
-
-    response = retrieval_augmented_qa_pipeline(client).run_pipeline(message.content)
+    vector_db = _build_vector_db()
+    pipeline = RetrievalAugmentedQAPipeline(
+        llm=client,
+        vector_db_retriever=vector_db
+    )
+    response = pipeline.run_pipeline(message.content)
     
     msg = cl.Message(content="")
-    msg.stream_token(response)
 
     # Update the prompt object with the completion
     msg.prompt = response