From f2846429310452bebbf0d07203b1e53978c439c7 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Tue, 18 Jun 2024 01:38:02 +0200
Subject: Wip ui

---
 rag/ui.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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))
 
-- 
cgit v1.2.3-70-g09d2