Update rag.py
Browse files
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)
|