diff options
| author | jiangyinzuo <jiangyinzuo@foxmail.com> | 2024-03-08 14:26:19 +0800 |
|---|---|---|
| committer | jiangyinzuo <jiangyinzuo@foxmail.com> | 2024-03-08 14:26:19 +0800 |
| commit | 6d77f57ae9e4e11fc4785d7d285f893c939818b6 (patch) | |
| tree | 65d6ba9c4a688f973fe635439a9c5ef1ebedf218 | |
| parent | e0e03cf9ca218cd3f8f9afe0a219aca7470fa4f5 (diff) | |
| download | vim-ai-6d77f57ae9e4e11fc4785d7d285f893c939818b6.tar.gz | |
feat: add an option to customize api key file location
Diffstat (limited to '')
| -rw-r--r-- | README.md | 6 | ||||
| -rw-r--r-- | autoload/vim_ai_config.vim | 3 | ||||
| -rw-r--r-- | py/utils.py | 3 |
3 files changed, 11 insertions, 1 deletions
@@ -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: |