Skip to content

Commit

Permalink
Merge pull request #634 from Jaya-sys/bugfix/azureai
Browse files Browse the repository at this point in the history
text to question parameter in the get_similar_question_sql(azure search)
  • Loading branch information
zainhoda authored Sep 12, 2024
2 parents f629f35 + 61bfdec commit 5872dd0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/vanna/azuresearch/azuresearch_vector.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 5872dd0

Please sign in to comment.