bstraehle commited on
Commit
cfda348
·
1 Parent(s): 704c818

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -115,7 +115,7 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
115
  retriever = db.as_retriever(search_kwargs = {"k": config["k"]}),
116
  return_source_documents = True)
117
 
118
- completion = rag_chain({"query": prompt}, include_run_info = true)
119
  print("###" + str(completion))
120
  print("###" + str(completion["run_info"]))
121
 
 
115
  retriever = db.as_retriever(search_kwargs = {"k": config["k"]}),
116
  return_source_documents = True)
117
 
118
+ completion = rag_chain({"query": prompt}, include_run_info = True)
119
  print("###" + str(completion))
120
  print("###" + str(completion["run_info"]))
121