Skip to content

Commit 4832524

Browse files
fix: allow similarity search on table names with special characters (#65)
1 parent 40babc5 commit 4832524

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/langchain_google_cloud_sql_mysql/vectorstore.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -675,15 +675,15 @@ def _query_collection(
675675
)
676676
if query_options.search_type == SearchType.KNN:
677677
filter = f"WHERE {filter}" if filter else ""
678-
stmt = f"SELECT {column_query}, {distance_function}({self.embedding_column}, string_to_vector('{embedding}')) AS distance FROM {self.table_name} {filter} ORDER BY distance LIMIT {k};"
678+
stmt = f"SELECT {column_query}, {distance_function}({self.embedding_column}, string_to_vector('{embedding}')) AS distance FROM `{self.table_name}` {filter} ORDER BY distance LIMIT {k};"
679679
else:
680680
filter = f"AND {filter}" if filter else ""
681681
num_partitions = (
682682
f",num_partitions={query_options.num_partitions}"
683683
if query_options.num_partitions
684684
else ""
685685
)
686-
stmt = f"SELECT {column_query}, {distance_function}({self.embedding_column}, string_to_vector('{embedding}')) AS distance FROM {self.table_name} WHERE NEAREST({self.embedding_column}) TO (string_to_vector('{embedding}'), 'num_neighbors={k}{num_partitions}') {filter} ORDER BY distance;"
686+
stmt = f"SELECT {column_query}, {distance_function}({self.embedding_column}, string_to_vector('{embedding}')) AS distance FROM `{self.table_name}` WHERE NEAREST({self.embedding_column}) TO (string_to_vector('{embedding}'), 'num_neighbors={k}{num_partitions}') {filter} ORDER BY distance;"
687687

688688
# return self.engine._fetch(stmt)
689689
if map_results:

0 commit comments

Comments
 (0)