From 334a1d2632da5fbc4c2a66c4171e361e94b07227 Mon Sep 17 00:00:00 2001
From: keenborder786 <21110290@lums.edu.pk>
Date: Tue, 11 Apr 2023 01:23:59 +0500
Subject: [PATCH] [feat]: Added a argument which allows gpt4 only mode and
 configure the fast_llm_model accordingly

---
 scripts/main.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/main.py b/scripts/main.py
index 3dfcaa15..6fc6a030 100644
--- a/scripts/main.py
+++ b/scripts/main.py
@@ -265,6 +265,7 @@ def parse_arguments():
     parser.add_argument('--speak', action='store_true', help='Enable Speak Mode')
     parser.add_argument('--debug', action='store_true', help='Enable Debug Mode')
     parser.add_argument('--gpt3only', action='store_true', help='Enable GPT3.5 Only Mode')
+    parser.add_argument('--gpt4only', action='store_true', help='Enable GPT4 Only Mode')
     args = parser.parse_args()
 
     if args.continuous:
@@ -286,6 +287,10 @@ def parse_arguments():
     if args.gpt3only:
         print_to_console("GPT3.5 Only Mode: ", Fore.GREEN, "ENABLED")
         cfg.set_smart_llm_model(cfg.fast_llm_model)
+    
+    if args.gpt4only:
+        print_to_console("GPT4 Only Mode: ", Fore.GREEN, "ENABLED")
+        cfg.set_fast_llm_model(cfg.smart_llm_model)
 
     if args.debug:
         print_to_console("Debug Mode: ", Fore.GREEN, "ENABLED")