首页 > Python > 如何用Python调用ChatGPT,实现一个本地化聊天机器

如何用Python调用ChatGPT,实现一个本地化聊天机器

Python 2023-03-01

众所周知,学习人工智能,绕不开Python。如果我想在Python中调用这个chatGPT,实现本地化处理应该怎么办呢?
注册chatGPT,并获取 OpenAI API Key 。

安装 openai 模块

使用如下代码安装 openai 模块。

pip install openai

调用接口获取返回信息

代码

调用 openai.Completion.create 函数获取返回信息。

import os
import openai

# 填写注册OpenAI接口账号时获取的 OpenAI API Key
openai.api_key = os.getenv("OPENAI_API_KEY")

# 提问
issue = '你是谁?'

# 访问OpenAI接口
response = openai.Completion.create(
  model='text-davinci-003',
  prompt=issue,
  temperature=0.9,
  max_tokens=150,
  top_p=1,
  frequency_penalty=0.0,
  presence_penalty=0.6
)

# 返回信息
resText = response.choices[0].text

print(resText)

参数

  • model:使用的模型,可选内容如下:
  • prompt:提问,提问描述越详细,回答越准确
  • temperature:控制结果的随机性,如果希望结果更有差异性 0.9,或者希望有固定结果可以尝试 0.0
  • max_tokens:生成结果时的最大 tokens 数。平均一个汉字是 2 个 tokens,text-davinci-003 最多是 4000 个 tokens,也就是 2000 个汉字左右
  • top_p:一个可用于代替 temperature 的参数,对应机器学习中 nucleus sampling,如果设置 0.1 意味着只考虑构成前 10% 概率质量的 tokens
  • frequency_penalty:控制字符的重复度,取值为 -2.0 ~ 2.0 之间的数字
  • presence_penalty:控制主题的重复度,取值为 -2.0 ~ 2.0 之间的数字

热门标签: Pandas   API   带区卷   图片放大   字体DIY   人工智能   安装   字幕