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

通过python-api使用openai的gpt

article 2024/10/6 18:46:10

目前，OpenAI 提供的 GPT 模型可以通过其提供的 API 进行访问。以下是如何通过 Python 使用 OpenAI GPT API 的详细步骤：

1. 安装 OpenAI Python 库

首先，你需要安装 OpenAI 的 Python 库。可以通过 pip 安装：

pip install openai

2. 获取 API 密钥

要使用 OpenAI 的 API，你需要一个 API 密钥。你可以通过以下步骤获取：

登录 OpenAI 官网。
进入控制台 (dashboard)。
在 API 页面，生成一个 API 密钥。

请确保妥善保管这个密钥，不要泄露给他人。

3. 使用 Python 调用 GPT 模型

以下是一个基本的示例代码，展示如何通过 OpenAI 的 API 调用 GPT 模型。

import openai

# 设置 OpenAI 的 API 密钥
openai.api_key = 'your-api-key-here'

# 调用 GPT-4 模型生成回答
response = openai.Completion.create(
  model="gpt-4",
  prompt="What is the meaning of life?",
  max_tokens=100
)

# 打印生成的回答
print(response.choices[0].text.strip())

4. 更多 API 选项

model：指定使用的模型。常用模型有 gpt-3.5-turbo 和 gpt-4。
prompt：你希望模型回答的问题或提供的上下文信息。
max_tokens：控制生成的回答长度，tokens 包括输入和输出，1 个 token 大约是 4 个字符的英文文本。
temperature：控制生成的随机性。值为 0 会使模型变得更确定性，值为 1 则增加输出的多样性。

5. 聊天模型示例

对于类似 ChatGPT 的对话体验，可以使用 chat-completion 接口。下面是一个示例代码：

import openai

openai.api_key = 'your-api-key-here'

response = openai.ChatCompletion.create(
  model="gpt-4",
  messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Tell me a joke."}
    ]
)

# 输出生成的回答
print(response['choices'][0]['message']['content'])

6. 处理 API 的响应

API 返回的是一个 JSON 对象，其中包括模型生成的文本以及一些元数据。通常你需要从 choices 列表中提取生成的文本。

例如，在上面的代码中，响应可以通过以下方式处理：

response_text = response['choices'][0]['message']['content']
print(response_text)

7. 异常处理

在生产环境中，建议添加异常处理，防止 API 调用失败或速率限制导致程序崩溃：

try:
    response = openai.Completion.create(
        model="gpt-4",
        prompt="Explain the theory of relativity.",
        max_tokens=150
    )
    print(response.choices[0].text.strip())
except Exception as e:
    print(f"API 调用失败: {e}")