Add support for GPT_CMD_TOKEN var

Author: chrisdothtml

README.md

@@ -78,9 +78,13 @@ Override the gpt model used by the tool.
 
 **Default**: `gpt-4o`
 
+### `GPT_CMD_TOKEN`
+
+Provide your OpenAI token via this env var instead of storing it in a file.
+
 ### `GPT_CMD_TOKEN_FILE_PATH`
 
-Override the file path the tool gets your OpenAI token from.
+Override the file path the tool gets your OpenAI token from. Alternatively, you can provide the token directly via `GPT_CMD_TOKEN`.
 
 **Default**: `~/OPENAI_TOKEN`
 

gpt_cmd.py

@@ -37,7 +37,7 @@ def write_file(file_path, content):
   'model': os.environ.get('GPT_CMD_MODEL', 'gpt-4o'),
   'token_file_path': os.environ.get(
     'GPT_CMD_TOKEN_FILE_PATH',
-    os.path.join(os.path.expanduser('~'), 'OPENAI_TOKEN'),
+    os.path.join(os.path.expanduser('~'), 'OPENAI_TOKEN')
   ),
 }
 
@@ -79,7 +79,11 @@ def red(text):
 def call_gpt(messages):
   global OPENAI_CLIENT
   if OPENAI_CLIENT is None:
-    OPENAI_CLIENT = OpenAI(api_key=read_file(OPTIONS['token_file_path']))
+    token = os.environ.get(
+      'GPT_CMD_TOKEN',
+      read_file(OPTIONS['token_file_path'])
+    )
+    OPENAI_CLIENT = OpenAI(api_key=token)
 
   response = OPENAI_CLIENT.chat.completions.create(
     model=OPTIONS['model'],