gpt3-completions.py

import openai

messages = []

def prompt_and_print_response():
    prompt = input(">> ")
    messages.append({"role": "user", "content": prompt})

    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )
    choice = response['choices'][0]
    finish_reason = choice['finish_reason']

    if finish_reason != "stop":
        print(response)
        # prompt to continue
        continue_input = input("Continue? ")
        if continue_input.lower() in ["y", "yes"]:
            return True
        else:
            exit(0)
    
    message = choice['message']
    print(message['content'])
    messages.append(message)

while True:
    try:
        prompt_and_print_response()
    except (KeyboardInterrupt, EOFError):
        print("\nExiting...")
        break