如何在环境变量中存储API密钥并在Google Colab中调用?以及如何创建包含GPT-3 API密钥的JSON文件并在Google Colab中实现自动代码生成
没问题,我来一步步教你搞定这两个需求——不管是用JSON文件还是环境变量存API密钥,在Google Colab里都很简单:
方法一:使用JSON文件存储API密钥
步骤1:创建包含API密钥的JSON文件
- 先在本地新建一个文本文件,把后缀改成
.json,比如命名为GPT_SECRET_KEY.json - 打开这个文件,写入以下内容(把
your_actual_api_key_here替换成你自己的OpenAI API密钥):
{ "API_KEY": "your_actual_api_key_here" }
- 保存好文件。
步骤2:上传JSON文件到Google Colab
- 打开你的Colab笔记本,点击左侧栏的「文件」图标(文件夹样式)
- 点击「上传到会话存储」按钮,选择刚才创建的
GPT_SECRET_KEY.json,等待上传完成。
步骤3:在Colab中读取并使用API密钥
- 导入需要的库,然后用你提供的代码读取密钥,还可以加个测试验证:
import json import openai # 读取JSON文件里的密钥 with open('GPT_SECRET_KEY.json') as f: data = json.load(f) openai.api_key = data["API_KEY"] # 测试调用:生成一个简单的加法函数 response = openai.ChatCompletion.create( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "写一个Python加法函数,接收两个参数返回它们的和"}] ) print(response.choices[0].message['content'])
方法二:使用Google Colab环境变量存储API密钥
这种方法更安全,不用上传文件,适合长期使用。
步骤1:设置环境变量(两种方式可选)
方式A:用Colab密钥管理(推荐,永久有效)
- 点击Colab顶部菜单栏的「代码执行程序」→「管理会话密钥」
- 在弹出的窗口里,「名称」填
OPENAI_API_KEY,「值」填你的API密钥,点击「添加」即可。这个密钥会被安全存储,每次打开笔记本都会自动加载。
方式B:代码临时设置(仅当前会话有效)
- 在Colab单元格里运行以下代码,输入你的API密钥(输入时不会显示明文):
import os from getpass import getpass os.environ['OPENAI_API_KEY'] = getpass('请输入你的OpenAI API密钥:')
步骤2:从环境变量中调用密钥
- 导入库后直接读取环境变量,同样可以加测试:
import openai import os openai.api_key = os.getenv('OPENAI_API_KEY') # 测试调用:解释Python装饰器的作用 response = openai.ChatCompletion.create( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "用大白话解释Python装饰器是什么"}] ) print(response.choices[0].message['content'])
⚠️ 重要提醒:不管用哪种方法,绝对不要把API密钥分享给他人,也不要把它写到公开的代码仓库里,密钥关联你的账户和费用,一定要妥善保管。
内容的提问来源于stack exchange,提问作者kumaran T
相关产品推荐
相关产品推荐

