From 61bfdec8bba3dd9cc341083dff73d1baf72e577c Mon Sep 17 00:00:00 2001 From: Jaya Maheshwari <65943514+Jaya-sys@users.noreply.github.com> Date: Wed, 11 Sep 2024 22:38:25 +0530 Subject: [PATCH] text to question parameter --- src/vanna/azuresearch/azuresearch_vector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/vanna/azuresearch/azuresearch_vector.py b/src/vanna/azuresearch/azuresearch_vector.py index 79546db0..bd6002a4 100644 --- a/src/vanna/azuresearch/azuresearch_vector.py +++ b/src/vanna/azuresearch/azuresearch_vector.py @@ -186,10 +186,10 @@ def get_related_documentation(self, text: str) -> List[str]: result = df["document"].tolist() return result - def get_similar_question_sql(self, text: str) -> List[str]: + def get_similar_question_sql(self, question: str) -> List[str]: result = [] # Vectorize the text - vector_query = VectorizedQuery(vector=self.generate_embedding(text), fields="document_vector") + vector_query = VectorizedQuery(vector=self.generate_embedding(question), fields="document_vector") df = pd.DataFrame( self.search_client.search( top=self.n_results_sql,