From 4935ad51367a90c033c9cb80143106b6be945c6b Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Sun, 14 Apr 2024 23:14:55 +0200
Subject: Add limit in ui

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

diff --git a/rag/ui.py b/rag/ui.py
index 253ad87..23c44f9 100644
--- a/rag/ui.py
+++ b/rag/ui.py
@@ -102,7 +102,7 @@ def generate_chat(query: str):
     retriever = ss.retriever
     generator = ss.generator
 
-    documents = retriever.retrieve(query)
+    documents = retriever.retrieve(query, limit=5)
     prompt = Prompt(query, documents)
 
     with st.chat_message(ss.bot):
-- 
cgit v1.2.3-70-g09d2