Wip ui

1 files changed, 2 insertions, 1 deletions

diff --git a/rag/ui.py b/rag/ui.py
index a453f47..9a0f2cf 100644
--- a/rag/ui.py
+++ b/rag/ui.py
@@ -65,7 +65,8 @@ def generate_chat(query: str):
         st.write(query)
 
     rag = ss.rag
-    prompt = rag.retrieve(query)
+    documents = rag.retrieve(query)
+    Prompt(query, documents, self.client)
     with st.chat_message(ss.bot):
         response = st.write_stream(rag.generate(query))