add context to prompt

app.py

@@ -150,11 +150,23 @@ async def main(message: cl.Message):
     The LCEL RAG chain is stored in the user session, and is unique to each user session - this is why we can access it here.
     """
     lcel_rag_chain = cl.user_session.get("lcel_rag_chain")
+    
+    # Retrieve context for the user's query
+    context = hf_retriever.retrieve(query=message.content)
+
+    # Combine context into a single string
+    context_text = "\n".join([doc.page_content for doc in context])
+
+    # Prepare input for the prompt
+    input_dict = {
+        "query": message.content,
+        "context": context_text
+    }
 
     msg = cl.Message(content="")
 
     async for chunk in lcel_rag_chain.astream(
-        {"query": message.content},
+        input_dict,
         config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
     ):
         await msg.stream_token(chunk)