Fix error print

README.md

@@ -84,11 +84,10 @@ terminalgpt delete
 
 ## Future Plans
 
-1. Support multiline input.
-2. Support optional vim input mode.
-3. Auto-completion for all commands.
-4. Open source the project.
-5. Add more models (???)
-6. Encrypt the conversation data.
+1. Support optional vim input mode.
+2. Auto-completion for all commands.
+3. Add more models
+4. Encrypt the conversation data.
+5. GPT-4 support.
 
 ---

terminalgpt/chat_utils.py

@@ -91,27 +91,27 @@ def get_user_answer(messages):
     """Returns the answer from OpenAI API."""
 
     while True:
-        with yaspin(
-            Spinners.earth,
-            text=Style.BRIGHT + "Assistant:" + Style.RESET_ALL,
-            color="blue",
-            side="right",
-        ):
-            try:
+        try:
+            with yaspin(
+                Spinners.earth,
+                text=Style.BRIGHT + "Assistant:" + Style.RESET_ALL,
+                color="blue",
+                side="right",
+            ):
                 answer = openai.ChatCompletion.create(
                     model=config.MODEL, messages=messages
                 )
                 return answer
-            except openai.error.RateLimitError as error:
-                print_utils.print_slowly(
-                    Back.RED + Style.BRIGHT + str(error) + Style.RESET_ALL
-                )
-                waiting_before_trying_again()
-            except openai.error.InvalidRequestError as error:
-                if "Please reduce the length of the messages" in str(error):
-                    messages.pop(1)
-                else:
-                    raise error
+        except openai.error.RateLimitError as error:
+            print_utils.print_slowly(
+                Back.RED + Style.BRIGHT + str(error) + Style.RESET_ALL
+            )
+            waiting_before_trying_again()
+        except openai.error.InvalidRequestError as error:
+            if "Please reduce the length of the messages" in str(error):
+                messages.pop(1)
+            else:
+                raise error
 
 
 def exceeding_token_limit(total_usage: int, token_limit: int):