Merge pull request #84 from alexsunxl/main

Fixed a crash with knowledge base arguments
2023-10-18 10:51:05 -07:00
parent 31d4f91f6f 71726d680b
commit e4be896b7c
1 changed files with 6 additions and 3 deletions
@@ -367,9 +367,12 @@ class AIOS_Shell:
            else:
                return show_text
        if sub_cmd == "journal":
-            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))
+            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