admin 管理员组

文章数量: 887021

使用python调用ChatGPT接口实现多轮连续对话

    • 使用前提
    • 导入需要用的包:
    • 设置好自己的端口,写入自己的key:
    • 发送message,得到gpt回复的函数:
    • 构建message,输入要说的话:
    • 最后进行调用
    • 完整代码在这里

使用前提

如果出现网络连接问题推荐在csdn上看一下这篇博客:《最新关于openai.APIConnectionError: Connection error.的解决方法》,第二点确保自己的openai账户还有余额,一般刚建号会送五美元,但是长时间不用就会过期。

导入需要用的包:

导入os和openai两个包,这里用的是openai=1.6.1.,可以先在命令行pip装一个。

pip install openai==1.6.1
from openai import OpenAI
import os

设置好自己的端口,写入自己的key:

不会设置,或者设置还报错看这位大佬写的博客最新关于openai.APIConnectionError: Connection error.的解决方法,然后把key换成自己的!!!.

os.environ["http_proxy"] = "http://localhost:7890"
#换成你自己的端口
os.environ["https_proxy"] = "http://localhost:7890"
client = OpenAI(api_key ="你自己的key")

发送message,得到gpt回复的函数:

message是在下面的chat函数构建的,gpt_content是我们得到的gpt的回复。

def demo_dev(messages):
  response = client.chat.completions.create(
    model="gpt-3.5-turbo-1106",
    messages=messages
  )
  gpt_content=response.choices[0].message.content
  print(gpt_content)
  return gpt_content

构建message,输入要说的话:

利用message.append实现对对话的累加,从而实现多轮对话的效果,从命令行输入的字符串就是你对gpt所说的话,该字符串赋值给user_content。

def chat(sys_content):
  message = [
    {'role': 'system', 'content': sys_content},
  ]
  while True:
    user_content = input(">>> ")
    if user_content == "exit":
      break
    message.append({'role': 'user', 'content': user_content})
    gpt_content = demo_dev(message)
    message.append({'role': 'system', 'content': "用户刚才对你说了"+user_content+","+"你回复了"+gpt_content})

最后进行调用

if __name__ == "__main__":

  chat('请你可爱的进行回复')

完整代码在这里

使用python调用ChatGPT接口实现多轮连续对话demo

本文标签: 接口 python ChatGpt