配置工作目录环境变量
在同级目录下创建一个.env的文件,.env文件通常用于存储敏感信息或配置参数,例如数据库连接字符串、API密钥等。例如:
OPENAI_API_TYPE="xxx"
OPENAI_API_BASE="xxx"
OPENAI_API_KEY="xxx"
OPENAI_API_VERSION="xxx"
加载环境变量
load_dotenv函数是python-dotenv库中的一个函数,用于加载.env文件中的环境变量。它会在调用时自动查找当前工作目录下的.env文件,并将其中的环境变量设置到os.environ中。这样,就可以使用os.getenv来获取这些环境变量的值了。
import os
from dotenv import load_dotenv
# 加载.env文件中的环境变量
load_dotenv()
# 使用os.getenv获取环境变量的值
openai.api_type = os.getenv("OPENAI_API_TYPE")
openai.api_base = os.getenv("OPENAI_API_BASE")
openai.api_key = os.getenv("OPENAI_API_KEY")
openai.api_version = os.getenv("OPENAI_API_VERSION")