openai chat bot base

.env

OPENAI_API_KEY="dummy"

main.py

import openai
import dotenv
import os
from typing import Literal

dotenv.load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")

def create_message(role: Literal["user", "assistant", "system"], content: str):
    return {
        "role": role,
        "content": content
    }

def openai_chat(messages: list):
    res = openai.ChatCompletion.create(
        # model="gpt-3.5-turbo",
#         model="gpt-4-turbo",
        model="gpt-4o",
        messages=messages,
    )
    return res["choices"][0]["message"]["content"]

def user_interaction():
    return input("You: ")

def print_interaction(name: str, message: str):
    print(f"{name}: {message}")

if __name__ == "__main__":
    # read system prompt
    system_prompt = "!!!INPUT SYSTEM PROMPT HERE!!!"
    with open("system_prompt.txt", "r") as f:
        system_prompt = f.read()

    # initial setting
    messages = [
        create_message("system", system_prompt),
        create_message("user", "Please introduce yourself."),
    ]

    while True:
        # request to AI
        msg = openai_chat(messages)
        messages.append(create_message("assistant", msg))
        print_interaction("AI", msg)

        # user input
        msg = user_interaction()
        if msg == "exit":
            break
        messages.append(create_message("user", msg))

system_prompt.txt

...