crodri commited on
Commit
32e1599
·
verified ·
1 Parent(s): 8373b81

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -151,7 +151,7 @@ class RAG:
151
  try:
152
  docs = self.get_context(prompt, model_parameters["NUM_CHUNKS"])
153
  text_context, full_context, source = self.beautiful_context(docs)
154
-
155
  del model_parameters["NUM_CHUNKS"]
156
 
157
  response = self.predict_completion(prompt, text_context, model_parameters)
 
151
  try:
152
  docs = self.get_context(prompt, model_parameters["NUM_CHUNKS"])
153
  text_context, full_context, source = self.beautiful_context(docs)
154
+ print(text_context)
155
  del model_parameters["NUM_CHUNKS"]
156
 
157
  response = self.predict_completion(prompt, text_context, model_parameters)