fixup file checker

rmackay9 · Dec 20, 2023 · 0d4ae09 · 0d4ae09
1 parent 5cf5d55
commit 0d4ae09
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/MAVProxy/modules/mavproxy_chat/assistant_setup/assistant_file_checker.py b/MAVProxy/modules/mavproxy_chat/assistant_setup/assistant_file_checker.py
@@ -9,7 +9,7 @@
 OpenAI Assistant Playground: https://platform.openai.com/playground
 '''
 
-import datetime
+import datetime, os
 
 try:
     from openai import OpenAI
@@ -97,8 +97,12 @@ def get_filename_size_date_from_id(files_list, file_id):
 
     # parse command line arguments
     from argparse import ArgumentParser
-    parser = ArgumentParser(description="MAVProxy AI chat module OpenAI Assistant setup script")
+    parser = ArgumentParser(description="MAVProxy Chat Module Assistant File Checker")
     parser.add_argument("--api-key", default=None, help="OpenAI API Key")
     parser.add_argument("--delete-unused", action='store_true', help="delete unused files")
-    args = parser.parse_args()
+    try:
+        args = parser.parse_args()
+    except AttributeError as err:
+        parser.print_help()
+        os.sys.exit(0)
     main(openai_api_key=args.api_key, delete_unused=args.delete_unused)