pyhton对接chatgpt

1.获取api_key

登录openai官网，创建自己的api_key

2.python 下载openai库

pip install openai

 3.代码

不同的模型有不同的请求方式，切返回值也有所不同，这边已gpt-3.5和text-davinci为例

1.gpt-3.5

import openai

openai.api_key = "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"
chatCompletion = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {
            "role": "user",
            "content": "这里输入你的问题"
        }
    ]
)
print(chatCompletion["choices"][0]['message']['content'])

如果需要串聊，就在message中传入上面的聊天记录

import openai

openai.api_key = "xxx" #输入你自己的api_key
chatCompletion = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {
            "role": "user",
            "content": "NBA2016年哪支队伍夺冠了？"
        },
        {
            "role": "assistant",
            "content": "NBA2016年夺冠的是克利夫兰骑士队"
        },
        {
            "role": "user",
            "content": "那2017年呢？"
        },
        
    ]
)
print(chatCompletion["choices"][0]['message']['content'])

2.text-davinci

import openai


openai.api_key = "xxx"

completion = openai.Completion.create(
    engine="text-davinci-003",  #模型
    prompt=prompt, #这里输入你的问题
    temperature=0.5, #这是一个浮点数，控制生成文本的随机性。它的值越高，生成的文本就越随机，值越低，生成的文本就越保守。
    max_tokens=1000, #指定要生成的最大token数，它是生成文本的长度限制。
    n=1, #不管
    stop=None # 用于指示生成文本应该停止的位置
)
print(completion["choices"][0]['text'])