From d560cbeea359c81c96589811f412e7426c9ae27b Mon Sep 17 00:00:00 2001 From: KevKibe Date: Tue, 30 Apr 2024 18:36:06 +0300 Subject: [PATCH] fix: vectorstore parameter in test_04_retrieve_and_generate method --- src/tests/googleindex_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tests/googleindex_test.py b/src/tests/googleindex_test.py index 6f64e09..3051db9 100644 --- a/src/tests/googleindex_test.py +++ b/src/tests/googleindex_test.py @@ -63,9 +63,9 @@ def test_04_retrieve_and_generate(self): """ Test initializing the vector store and assert its type. """ - vectorstore = self.indexer.initialize_vectorstore(self.index_name) + vector_store = self.indexer.initialize_vectorstore(self.index_name) response = self.indexer.retrieve_and_generate(query = "give a short summary of the introduction", - vectorstore= vectorstore + vectorstore= vector_store ) print(response) self.assertIsNotNone(response, "The retriever response should not be None.")