Clean up lcel_rag_chain

app.py

@@ -145,6 +145,7 @@ hf_llm = HuggingFaceEndpoint(
     typical_p=0.95,
     temperature=0.01,
     repetition_penalty=1.03,
+    huggingfacehub_api_token=HF_TOKEN,
 )
 
 @cl.author_rename
@@ -170,7 +171,10 @@ async def start_chat():
     """
 
     ### BUILD LCEL RAG CHAIN THAT ONLY RETURNS TEXT
-    lcel_rag_chain = rag_prompt | hf_llm
+    lcel_rag_chain = (
+        {"context": itemgetter("query") | hf_retriever, "query": itemgetter("query")}
+        | rag_prompt | hf_llm
+    )
 
     cl.user_session.set("lcel_rag_chain", lcel_rag_chain)