Fixed a crash with knowledge base arguments

2023-10-18 17:20:06 +08:00
parent 31d4f91f6f
commit 71726d680b
1 changed files with 6 additions and 3 deletions
@@ -367,9 +367,12 @@ class AIOS_Shell:
            else:
                return show_text
        if sub_cmd == "journal":
+            try:
                topn = 10 if len(args) == 1 else int(args[1])
                journals = [str(journal) for journal in KnowledgePipline.get_instance().get_latest_journals(topn)]
                print_formatted_text("\r\n".join(journals))
+            except ValueError:
+                return FormattedText([("class:title", f"/knowledge journal failed: {args[1]} is not a valid integer.\n")])
        if sub_cmd == "query":
            if len(args) < 2:
                return show_text