[백엔드][Django][장고 채널스] Section1. openAI API 살펴보기 (1)

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