bupa1018 commited on
Commit
506afb0
·
1 Parent(s): 417adb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -187,7 +187,7 @@ def query_chroma(vectorstore, query, k):
187
  return chunks_with_references
188
 
189
  def rag_workflow(query):
190
- docs = query_chroma(vectorstore, query, k=5)
191
  context = "\n\n".join([doc for doc, _ in docs])
192
  references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(docs)])
193
  print(f"Context for the query:\n{context}\n")
@@ -221,6 +221,11 @@ def add_text(history, text):
221
  history = history + [(text, None)]
222
  yield history, ""
223
 
 
 
 
 
 
224
  def bot_kadi(history):
225
  user_query = history[-1][0]
226
  response, references = rag_workflow(user_query)
 
187
  return chunks_with_references
188
 
189
  def rag_workflow(query):
190
+ docs = query_chroma(vectorstore, query, k=20)
191
  context = "\n\n".join([doc for doc, _ in docs])
192
  references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(docs)])
193
  print(f"Context for the query:\n{context}\n")
 
221
  history = history + [(text, None)]
222
  yield history, ""
223
 
224
+
225
+
226
+ import gradio as gr
227
+
228
+
229
  def bot_kadi(history):
230
  user_query = history[-1][0]
231
  response, references = rag_workflow(user_query)