본문 바로가기

Prompt/프롬프트 공부

OPEN AI API로 간단하게 Chat-GPT 만들기

from dotenv import load_dotenv
import openai
import os

## chat-GPT활용
load_dotenv()
openai.api_key = os.getenv("openai.api_key")

messages = [{'role':'system','content':'you are an intellignet assistant. answer in korean'}]

while True:
    message = input("user:")

    #'stop'입력하면 종료되도록
    if message == 'stop':
        break

    if messages:
        #print("messages:",messages,sep = '\n')

        messages.append(
            {'role':'user','content':message},
        )
        while len(messages)>5:
            messages.pop(1)
            #시스템 문장 뒤부터 max에 도달하지 않도록 삭제하기

        chat = openai.ChatCompletion.create(
            model = 'gpt-3.5-turbo',
            messages = messages,
            temperature = 0
        )

    reply = chat.choices[0].message.content
    print(f'ChatGPT: {reply}')
    messages.append({'role':'assistant', 'content':reply})

message 입력시 주의 사항

엔터로 구분되기 때문에 엔터가 이미 들어있는 글을 복붙하면 계속 답하는 문제가 생김