Skip to content

Commit ee9dc73

Browse files
committed
fix: shorten trick assistant summaries
1 parent 747d007 commit ee9dc73

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

backend/wiki/assistant.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -652,14 +652,14 @@ def extract_trick_query_tokens(query: str):
652652
return useful[:10]
653653

654654

655-
def format_trick_item(entry: TrickEntry):
655+
def format_trick_item(entry: TrickEntry, *, include_summary: bool = True):
656656
raw_title = collapse_text(entry.title)
657-
title = shorten_text(raw_title, 36)
657+
title = shorten_text(raw_title, 30)
658658
body = collapse_text(strip_markdown(entry.content_md))
659659
if body.startswith(raw_title):
660660
body = body[len(raw_title) :].lstrip(" ::,,.。")
661-
summary = shorten_text(body, 44)
662-
if summary and summary != title:
661+
summary = shorten_text(body, 34)
662+
if include_summary and summary and summary != title:
663663
return f"- {title}\uff1a{summary}"
664664
return f"- {title}"
665665

@@ -718,7 +718,9 @@ def build_trick_digest(query: str, *, current_path: str = "", current_title: str
718718
answer_lines = ["\u5e08\u5144\uff0c\u548c\u4f60\u95ee\u7684\u5185\u5bb9\u6700\u50cf\u7684 trick \u5148\u770b\u8fd9\u51e0\u6761\uff1a"]
719719
else:
720720
answer_lines = ["\u5e08\u5144\uff0ctrick \u9875\u91cc\u5df2\u6536\u5f55\u7684\u6280\u5de7\u53ef\u4ee5\u5148\u770b\u8fd9\u51e0\u6761\uff1a"]
721-
answer_lines.extend(format_trick_item(entry) for entry in selected[:3])
721+
answer_lines.extend(
722+
format_trick_item(entry, include_summary=bool(tokens)) for entry in selected[:3]
723+
)
722724

723725
sources = []
724726
for entry in selected[:3]:

0 commit comments

Comments
 (0)