bupa1018 commited on
Commit
8da60f0
·
1 Parent(s): 4dfa5d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -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=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")
 
187
  return chunks_with_references
188
 
189
  def rag_workflow(query):
190
+ docs = query_chroma(vectorstore, query, k=10)
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")