当前位置：首页 > article >正文

【GPT入门】第2课跑通第一openAI程序

article 2025/3/9 8:03:30

【GPT入门】第2课跑通第一openAI程序

1.环境变量中配置openAI key
2.代码

1.环境变量中配置openAI key

国内无法直接访问openai,可以用代理方式，可以从下面网站注册一个：
https://key.wenwen-ai.com/

把下面两个变量配置到环境变量
OPENAI_API_KEY=“sk-xxx” #改为自己的key
OPENAI_BASE_URL=“https://key.wenwen-ai.com/v1”

2.代码

from openai import OpenAI

# 加载 .env 文件到环境变量
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())

# 初始化 OpenAI 服务。会自动从环境变量加载 OPENAI_API_KEY 和 OPENAI_BASE_URL
client = OpenAI()

# 消息
messages = [
    {
        "role": "system",
        "content": "你是AI助手小瓜，是 AGIClass.ai 的助教。这门课每周二、四上课。"  # 注入新知识
    },
    {
        "role": "user",
        "content": "周末上课吗？"  # 问问题。可以改改试试
    },

]

# 调用 GPT-4o-mini 模型
chat_completion = client.chat.completions.create(
    model="gpt-4o-mini",
    messages=messages
)

# 输出回复
print(chat_completion.choices[0].message.content)