From 8bd362ad5cd8b6902983ca4a30c1baccb5f32f83 Mon Sep 17 00:00:00 2001 From: Ayodeji Ayibiowu Date: Mon, 19 Feb 2024 20:35:29 +0100 Subject: [PATCH] fix(ci): remove debug print statement --- .../retrievers/document_compressors/llmlingua_filter.py | 1 - .../retrievers/document_compressors/test_llmlingua_filter.py | 1 - 2 files changed, 2 deletions(-) diff --git a/libs/community/langchain_community/retrievers/document_compressors/llmlingua_filter.py b/libs/community/langchain_community/retrievers/document_compressors/llmlingua_filter.py index 88d10d6922a8b..c4746fb8515d9 100644 --- a/libs/community/langchain_community/retrievers/document_compressors/llmlingua_filter.py +++ b/libs/community/langchain_community/retrievers/document_compressors/llmlingua_filter.py @@ -170,7 +170,6 @@ def compress_documents( add_instruction=True, **self.additional_compress_kwargs, ) - print(compressed_prompt) compreseed_context = compressed_prompt["compressed_prompt"].split("\n\n")[1:] extracted_metadata = self.extract_ref_id_tuples_and_clean(compreseed_context) diff --git a/libs/community/tests/unit_tests/retrievers/document_compressors/test_llmlingua_filter.py b/libs/community/tests/unit_tests/retrievers/document_compressors/test_llmlingua_filter.py index d2cd83daf5ee3..4eb34bf0b05d0 100644 --- a/libs/community/tests/unit_tests/retrievers/document_compressors/test_llmlingua_filter.py +++ b/libs/community/tests/unit_tests/retrievers/document_compressors/test_llmlingua_filter.py @@ -92,7 +92,6 @@ def test_compress_documents_with_documents( Document(page_content="Content of document 1", metadata={"id": "1"}), ] compressed_docs = llm_lingua_compressor.compress_documents(docs, "query") - print(compressed_docs) assert len(compressed_docs) == 2 assert compressed_docs[0].page_content == "Compressed content for document 0" assert compressed_docs[0].metadata == {"id": "0"}