From cba22f4d0a576b40bab430f48f3be73b9478673a Mon Sep 17 00:00:00 2001 From: Andrew Grande Date: Tue, 1 Oct 2024 16:00:45 -0700 Subject: [PATCH] Use named arguments instead of positional Fixed a bug where using both k and pre_filter search_kwargs caused in valid SQL generation. --- src/langchain_google_spanner/vector_store.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/langchain_google_spanner/vector_store.py b/src/langchain_google_spanner/vector_store.py index b818fcf2..675b5b73 100644 --- a/src/langchain_google_spanner/vector_store.py +++ b/src/langchain_google_spanner/vector_store.py @@ -830,7 +830,9 @@ def similarity_search_with_score_by_vector( """ results, column_order_map = self._get_rows_by_similarity_search( - embedding, k, pre_filter + embedding=embedding, + k=k, + pre_filter=pre_filter ) documents = self._get_documents_from_query_results( list(results), column_order_map