summaryrefslogtreecommitdiff
path: root/plugin/vim-ai.vim
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--plugin/vim-ai.vim6
1 files changed, 5 insertions, 1 deletions
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim
index e02936c..974fa19 100644
--- a/plugin/vim-ai.vim
+++ b/plugin/vim-ai.vim
@@ -1,4 +1,8 @@
-let g:openaiToken = system("cat ~/.config/openai.token")
+if $OPENAI_API_TOKEN != ''
+ let g:openaiToken = $OPENAI_API_TOKEN
+else
+ let g:openaiToken = system("cat ~/.config/openai.token")
+endif
function! AIRun(...) range
let prompt = getline(a:firstline, a:lastline)