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

---
 py/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'py/utils.py')

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