test_openai_api.py
import sys
import openai
openai.api_key = "" # your API key
if not openai.api_key:
print("OPENAI API KEY를 지정해주세요.", file=sys.stderr)
sys.exit(1)
# 텍스트 생성 혹은 문서 요약
response = openai.Completion.create(
engine="text-davinci-003",
prompt="""
Fix grammar errors:
- I is a boy
- You is a girl""".strip(),
)
print(response)
print(response.choices[0].text.strip())
# 챗봇 응답 생성
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "당신은 지식이 풍부한 도우미입니다."},
{"role": "user", "content": "세계에서 가장 큰 도시는 어디인가요?"}
],
)
print(response)
print(response["choices"][0]["message"]["content"])
.env 파일 만들어서 변수(’OPENAI_API_KEY’) 생성하고, test-chat.py에다가
import os
import openai
from dotenv import load_dotenv
load_dotenv()
openai.api_key = os.getenv("OPENAI_API_KEY")
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "당신은 영어 학습을 도와주는 챗봇입니다."},
{"role": "user", "content": "대화를 나눠봅시다."},
],
)
print(response)
# 응답 메세지만 출력하기
print(response["choices"][0]["message"]["content"])
Uploaded by Notion2Tistory v1.1.0