diff --git a/src/tests/googleindex_test.py b/src/tests/googleindex_test.py index 25e7156..110dddf 100644 --- a/src/tests/googleindex_test.py +++ b/src/tests/googleindex_test.py @@ -59,19 +59,19 @@ def test_03_initialize_vectorstore(self): vectorstore = self.indexer.initialize_vectorstore(self.index_name) self.assertIsInstance(vectorstore, PineconeVectorStore) - def test_04_retrieve_and_generate(self): - """ - Test initializing the vector store and assert its type. - """ - vector_store = self.indexer.initialize_vectorstore(self.index_name) - response = self.indexer.retrieve_and_generate( - query = "tell me something from the context texts", - vector_store = vector_store, - top_k = 3, - rerank_model = "t5" - ) - print(response) - self.assertIsNotNone(response, "The retriever response should not be None.") + # def test_04_retrieve_and_generate(self): + # """ + # Test initializing the vector store and assert its type. + # """ + # vector_store = self.indexer.initialize_vectorstore(self.index_name) + # response = self.indexer.retrieve_and_generate( + # query = "tell me something from the context texts", + # vector_store = vector_store, + # top_k = 3, + # rerank_model = "t5" + # ) + # print(response) + # self.assertIsNotNone(response, "The retriever response should not be None.") @patch('sys.stdout', new_callable=StringIO) def test_05_delete_index(self, mock_stdout):