()
| 72 | print() |
| 73 | |
| 74 | def main(): |
| 75 | parser = argparse.ArgumentParser() |
| 76 | parser.add_argument("prompt", nargs="*", help="Natural language prompt") |
| 77 | args = parser.parse_args() |
| 78 | prompt = " ".join(args.prompt).strip() |
| 79 | |
| 80 | if not prompt or prompt == "--help": |
| 81 | print_help_message() |
| 82 | sys.exit(0) |
| 83 | |
| 84 | print(f"{BLUE}Prompt: {prompt}{RESET}", flush=True) |
| 85 | one_shot_mode(prompt) |
| 86 | |
| 87 | if __name__ == "__main__": |
| 88 | # We call multiprocessing.freeze_support() because we are using PyInstaller to build a frozen binary. |
no test coverage detected