enpaiva commited on
Commit
7da5893
·
verified ·
1 Parent(s): 2dc9932

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def reformat_rag(results_rag):
89
 
90
  def chat_stream_completion(message, history):
91
 
92
- context = reformat_rag(RAG.search(message, None, k=2))
93
  context = " \n ".join(context)
94
 
95
  full_prompt = prompt.format(context,message,"")
 
89
 
90
  def chat_stream_completion(message, history):
91
 
92
+ context = reformat_rag(RAG.search(message, None, k=1))
93
  context = " \n ".join(context)
94
 
95
  full_prompt = prompt.format(context,message,"")