diff --git a/sciphi/interface/llm/sciphi_interface.py b/sciphi/interface/llm/sciphi_interface.py index 119de8d..cbe8266 100644 --- a/sciphi/interface/llm/sciphi_interface.py +++ b/sciphi/interface/llm/sciphi_interface.py @@ -122,7 +122,6 @@ def get_completion( prompt_with_context, generation_config ).strip() completion += latest_completion - print("latest_completion = ", latest_completion) if not completion.endswith(SciPhiFormatter.RETRIEVAL_TOKEN): break @@ -132,7 +131,6 @@ def get_completion( else f"{SciPhiFormatter.remove_cruft(completion)}" ) context = self.rag_interface.get_contexts([context_query])[0] - print("context = ", context) completion += f"{SciPhiFormatter.INIT_PARAGRAPH_TOKEN}{context}{SciPhiFormatter.END_PARAGRAPH_TOKEN}" return SciPhiFormatter.remove_cruft(completion)