bstraehle commited on
Commit
090c3ab
·
1 Parent(s): 349ba06

Update rag.py

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