Merge branch 'master' of https://github.com/BillSchumacher/Auto-GPT into plugin-support

autogpt/llm_utils.py

@@ -4,11 +4,12 @@ import time
 from typing import List, Optional
 
 import openai
-from colorama import Fore
+from colorama import Fore, Style
 from openai.error import APIError, RateLimitError
 
 from autogpt.config import Config
 from autogpt.types.openai import Message
+from autogpt.logs import logger
 
 CFG = Config()
 
@@ -70,6 +71,7 @@ def create_chat_completion(
         str: The response from the chat completion
     """
     num_retries = 10
+    warned_user = False
     if CFG.debug_mode:
         print(
             f"{Fore.GREEN}Creating chat completion with model {model}, temperature {temperature}, max_tokens {max_tokens}{Fore.RESET}"
@@ -112,6 +114,12 @@ def create_chat_completion(
                 print(
                     f"{Fore.RED}Error: ", f"Reached rate limit, passing...{Fore.RESET}"
                 )
+            if not warned_user:
+                logger.double_check(
+                    f"Please double check that you have setup a {Fore.CYAN + Style.BRIGHT}PAID{Style.RESET_ALL} OpenAI API Account. "
+                    + f"You can read more here: {Fore.CYAN}https://github.com/Significant-Gravitas/Auto-GPT#openai-api-keys-configuration{Fore.RESET}"
+                )
+                warned_user = True
         except APIError as e:
             if e.http_status != 502:
                 raise
@@ -124,7 +132,17 @@ def create_chat_completion(
             )
         time.sleep(backoff)
     if response is None:
-        raise RuntimeError(f"Failed to get response after {num_retries} retries")
+        logger.typewriter_log(
+            "FAILED TO GET RESPONSE FROM OPENAI",
+            Fore.RED,
+            "Auto-GPT has failed to get a response from OpenAI's services. "
+            + f"Try running Auto-GPT again, and if the problem the persists try running it with `{Fore.CYAN}--debug{Fore.RESET}`.",
+        )
+        logger.double_check()
+        if CFG.debug_mode:
+            raise RuntimeError(f"Failed to get response after {num_retries} retries")
+        else:
+            quit(1)
     resp = response.choices[0].message["content"]
     for plugin in CFG.plugins:
         if not plugin.can_handle_on_response():
@@ -134,7 +152,7 @@ def create_chat_completion(
 
 
 def create_embedding_with_ada(text) -> list:
-    """Create a embedding with text-ada-002 using the OpenAI SDK"""
+    """Create an embedding with text-ada-002 using the OpenAI SDK"""
     num_retries = 10
     for attempt in range(num_retries):
         backoff = 2 ** (attempt + 2)