Update modules/index_func.py

modules/index_func.py

@@ -116,7 +116,9 @@ def construct_index(
                                           model=os.environ["AZURE_EMBEDDING_MODEL_NAME"], openai_api_base=os.environ["AZURE_OPENAI_API_BASE_URL"], openai_api_type="azure")
     if os.path.exists(index_path):
         logging.info("找到了缓存的索引文件，加载中……")
-        return FAISS.load_local(index_path, embeddings)
+        index = FAISS.load_local(index_path, embeddings)
+        os.environ["OPENAI_API_KEY"] = ""
+        return index
     else:
         try:
             documents = get_documents(file_src)
@@ -127,10 +129,12 @@ def construct_index(
             os.makedirs("./index", exist_ok=True)
             index.save_local(index_path)
             logging.debug("索引已保存至本地!")
+            os.environ["OPENAI_API_KEY"] = ""
             return index
 
         except Exception as e:
             import traceback
             logging.error("索引构建失败！%s", e)
             traceback.print_exc()
+            os.environ["OPENAI_API_KEY"] = ""
             return None