File tree 1 file changed +4
-2
lines changed 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -561,10 +561,12 @@ async def __query_collection(
561
561
inline_embed_func = getattr (self .embedding_service , "embed_query_inline" , None )
562
562
if not embedding and callable (inline_embed_func ) and "query" in kwargs :
563
563
query_embedding = self .embedding_service .embed_query_inline (kwargs ["query" ]) # type: ignore
564
+ embedding_data_string = f"{ query_embedding } "
564
565
else :
565
566
query_embedding = f"{ [float (dimension ) for dimension in embedding ]} "
566
- stmt = f"""SELECT { column_names } , { search_function } ("{ self .embedding_column } ", :query_embedding) as distance
567
- FROM "{ self .schema_name } "."{ self .table_name } " { param_filter } ORDER BY "{ self .embedding_column } " { operator } :query_embedding LIMIT :k;
567
+ embedding_data_string = ":query_embedding"
568
+ stmt = f"""SELECT { column_names } , { search_function } ("{ self .embedding_column } ", { embedding_data_string } ) as distance
569
+ FROM "{ self .schema_name } "."{ self .table_name } " { param_filter } ORDER BY "{ self .embedding_column } " { operator } { embedding_data_string } LIMIT :k;
568
570
"""
569
571
param_dict = {"query_embedding" : query_embedding , "k" : k }
570
572
if filter_dict :
You can’t perform that action at this time.
0 commit comments