From 6d77f57ae9e4e11fc4785d7d285f893c939818b6 Mon Sep 17 00:00:00 2001
From: jiangyinzuo <jiangyinzuo@foxmail.com>
Date: Fri, 8 Mar 2024 14:26:19 +0800
Subject: feat: add an option to customize api key file location

---
 README.md                  | 6 ++++++
 autoload/vim_ai_config.vim | 3 +++
 py/utils.py                | 3 ++-
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index da76953..18ac762 100644
--- a/README.md
+++ b/README.md
@@ -40,6 +40,12 @@ echo "YOUR_OPENAI_API_KEY,YOUR_OPENAI_ORG_ID" > ~/.config/openai.token
 export OPENAI_API_KEY="YOUR_OPENAI_API_KEY,YOUR_OPENAI_ORG_ID"
 ```
 
+The default api key file location is `~/.config/openai.token`, but you can change it by setting the `g:vim_ai_token_file_path` in your `.vimrc` file:
+
+```vim
+let g:vim_ai_token_file_path = '~/.config/openai.token'
+```
+
 ### Using `vim-plug`
 
 ```vim
diff --git a/autoload/vim_ai_config.vim b/autoload/vim_ai_config.vim
index 6119e4f..ef3c863 100644
--- a/autoload/vim_ai_config.vim
+++ b/autoload/vim_ai_config.vim
@@ -70,6 +70,9 @@ endif
 if !exists("g:vim_ai_debug_log_file")
   let g:vim_ai_debug_log_file = "/tmp/vim_ai_debug.log"
 endif
+if !exists("g:vim_ai_token_file_path")
+  let g:vim_ai_token_file_path = "~/.config/openai.token"
+endif
 
 function! vim_ai_config#ExtendDeep(defaults, override) abort
   let l:result = a:defaults
diff --git a/py/utils.py b/py/utils.py
index a5c489f..1e2ae95 100644
--- a/py/utils.py
+++ b/py/utils.py
@@ -19,7 +19,8 @@ class KnownError(Exception):
     pass
 
 def load_api_key():
-    config_file_path = os.path.join(os.path.expanduser("~"), ".config/openai.token")
+    config_file_path = os.path.expanduser(vim.eval("g:vim_ai_token_file_path"))
+    print(config_file_path)
     api_key_param_value = os.getenv("OPENAI_API_KEY")
     try:
         with open(config_file_path, 'r') as file:
-- 
cgit v1.2.3