search_sql add include_rank option

This commit is contained in:
Jacob Chapman 2022-08-30 04:08:55 -05:00 committed by GitHub
commit 863dd03c67
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -2376,6 +2376,7 @@ class Table(Queryable):
limit: Optional[int] = None,
offset: Optional[int] = None,
where: Optional[str] = None,
include_rank: bool = False,
) -> str:
""" "
Return SQL string that can be used to execute searches against this table.
@ -2427,6 +2428,8 @@ class Table(Queryable):
rank_implementation = "rank_bm25(matchinfo([{}], 'pcnalx'))".format(
fts_table
)
if include_rank:
columns_with_prefix_sql += ",\n " + rank_implementation + " rank"
limit_offset = ""
if limit is not None:
limit_offset += " limit {}".format(limit)