（chatGPT）Python调用text-davinci-003接口，进行有记忆的对话

最新推荐文章于 2023-05-14 23:27:35 发布

AwesomeAI

最新推荐文章于 2023-05-14 23:27:35 发布

阅读量1.2k

点赞数 1

文章标签： python 深度学习开发语言

本文链接：https://blog.csdn.net/GeorgeDu/article/details/129049693

版权

import openai
openai.api_key = "YOUR-KEY"

# 建立对话
def start_conversation(prompt):
    completions = openai.Completion.create(
    engine="text-davinci-002",
    prompt=prompt,
    max_tokens=0,
    )
    message = completions.choices[0].text
    return message

# 更新上下文
def append_context(prompt, context):
    if len(context) > 0:
        prompt = f"{prompt}\n\nContext: {context}\n"
    return prompt

# 生成文本
def generate_text(prompt, context):
    prompt = append_context(prompt, context)
    completions = openai.Completion.create(
    engine="text-davinci-003",
    prompt=prompt,
    max_tokens=1024,
    n=1,
    stop=None,
    temperature=0.5,
    )
    message = completions.choices[0].text
    return message

if __name__ == "__main__":
    context = ""
    prompt = ""    
    while True:
        print("########CONVERSATION BLOCK########")
        prompt = append_context(prompt, context)
        user_input = input("You: ")
        if user_input.lower() == "exit":
            break
        prompt += f"\nUser: {user_input}"
        response = generate_text(prompt, context)
        context += f"\nUser: {user_input}\nAI: {response}"
        print(f"AI:{response}")

使用【"text-davinci-002"】