【无标题】

最新推荐文章于 2024-08-12 14:30:00 发布

m0_37764106

最新推荐文章于 2024-08-12 14:30:00 发布

阅读量45

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/m0_37764106/article/details/128917982

版权

chatgpt的python接口

import os
import openai
from enum import Enum


class MODE(Enum):
  SingleLine = 0 # 单行输入，单次回车结束
  MultiLine = 1 # 多行输入，两次回车结束


class test():
  def __init__(self,MODE):
      self.MODE = MODE

  def quest(self):
    openai.api_key = os.getenv("OPENAI_API_KEY")

    text = input("\n\nQ: ")
    if self.MODE == 1:
      line_text = ''
      while True:
        line_text = input()
        if line_text == '':
          break
        text += line_text

    try:
      response = openai.Completion.create(
        model="text-davinci-003",
        prompt=text,
        max_tokens=4096 - len(text) * 4,
        top_p=1,
        frequency_penalty=0.0,
        presence_penalty=0.0,
      )
      print("\n\nA: " + response["choices"][0]["text"].strip())
    except Exception as exc:
      print(exc)

if __name__=="__main__":
  a = test(MODE.SingleLine)
  a.quest()