bstraehle commited on
Commit
20848ce
·
1 Parent(s): b4be805

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -116,7 +116,7 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
116
  return_source_documents = True,
117
  verbose = False)
118
 
119
- completion = rag_chain({"query": prompt}, include_run_info: True)
120
  print("###" + completion)
121
 
122
  return completion, rag_chain
 
116
  return_source_documents = True,
117
  verbose = False)
118
 
119
+ completion = rag_chain({"query": prompt}, include_run_info = True)
120
  print("###" + completion)
121
 
122
  return completion, rag_chain