Created --reset-key flag to update OpenAI key

sgpt/app.py

@@ -114,7 +114,19 @@ def main(
         callback=install_shell_integration,
         hidden=True,  # Hiding since should be used only once.
     ),
+    reset_key : bool = typer.Option(
+        False,
+        "--reset-key",
+        help="Resets OpenAI API key in config file. Note: the OPENAI_API_KEY environment variable will still be prioritized over the key in config file.",
+    ),
 ) -> None:
+    if reset_key:
+        if cfg.reset_OPENAI_API_key():
+            typer.secho("Successfully reset OpenAI API key", fg="green")
+        else:
+            typer.secho("Reset OpenAI API key in config file. However, the existing OPENAI_API_KEY environement variable will still be used.", fg="yellow")
+        return
+
     stdin_passed = not sys.stdin.isatty()
 
     if stdin_passed and not repl:

sgpt/client.py

@@ -3,6 +3,7 @@
 from typing import Dict, Generator, List
 
 import requests
+import typer
 
 from .cache import Cache
 from .config import cfg
@@ -58,7 +59,15 @@ def _request(
             timeout=REQUEST_TIMEOUT,
             stream=stream,
         )
-        response.raise_for_status()
+        try:
+            response.raise_for_status()
+        except requests.exceptions.HTTPError as http_err:
+            # Check if the status code is 403 (Forbidden) or 401 (Unauthorized)
+            if http_err.response.status_code == 403 or http_err.response.status_code == 401:
+                typer.secho("Invalid OpenAI API key.", fg="red")
+            else:
+                typer.secho(f"HTTPError occurred: {http_err}", fg="red")
+            raise typer.Exit(code=1)
         # TODO: Optimise.
         # https://github.com/openai/openai-python/blob/237448dc072a2c062698da3f9f512fae38300c1c/openai/api_requestor.py#L98
         if not stream:

sgpt/config.py

@@ -79,5 +79,14 @@ def get(self, key: str) -> str:  # type: ignore
             raise UsageError(f"Missing config key: {key}")
         return value
 
+    def reset_OPENAI_API_key(self):
+        __api_key = getpass(prompt="Please enter your new OpenAI API key: ")
+        self["OPENAI_API_KEY"] = __api_key
+        self._write()
+        #if an env variable exists it will still be used
+        if os.getenv("OPENAI_API_KEY"):
+            return False
+        else:
+            return True
 
 cfg = Config(SHELL_GPT_CONFIG_PATH, **DEFAULT_CONFIG)