diff options
| author | Martin Bielik <mx.bielik@gmail.com> | 2024-03-09 13:13:53 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-09 13:13:53 +0100 |
| commit | bb018930969076cd3cbd6f283eb851fc4e7d2985 (patch) | |
| tree | 441e0fdfa3d5ddb05d87fbccca1b92dbd7d762b3 | |
| parent | 2c68098308cb59c2a5dd13c3b3f4323ae1b602e2 (diff) | |
| parent | 6d77f57ae9e4e11fc4785d7d285f893c939818b6 (diff) | |
| download | vim-ai-bb018930969076cd3cbd6f283eb851fc4e7d2985.tar.gz | |
Merge pull request #80 from jiangyinzuo/main
feat: add an option to customize api key file location
| -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: |