File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 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(
561561 inline_embed_func = getattr (self .embedding_service , "embed_query_inline" , None )
562562 if not embedding and callable (inline_embed_func ) and "query" in kwargs :
563563 query_embedding = self .embedding_service .embed_query_inline (kwargs ["query" ]) # type: ignore
564+ embedding_data_string = f"{ query_embedding } "
564565 else :
565566 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;
568570 """
569571 param_dict = {"query_embedding" : query_embedding , "k" : k }
570572 if filter_dict :
You can’t perform that action at this time.
0 commit comments