실습 5 — 오류를 잡아 친절하게 알리기

import os

import openai
from dotenv import find_dotenv, load_dotenv
from langchain_openai import ChatOpenAI

# 과목 폴더의 .env 에서 OPENAI_API_KEY 를 읽어 환경 변수에 넣는다
load_dotenv(find_dotenv(usecwd=True))
# 모델 이름도 .env 에서 바꿀 수 있게 하고, 없으면 gpt-4.1-mini 를 쓴다
MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-mini")


def explain(e):
    """오류 객체를 받아 무엇을 고치면 되는지 한국어로 돌려준다. 좁은 오류부터 차례로 견준다."""
    # isinstance(e, 종류): e 가 그 종류의 오류이거나 그것을 이어받은 오류면 True
    if isinstance(e, openai.AuthenticationError):
        return "401 인증 실패: API 키가 틀렸거나 지운 키입니다. .env 의 키를 다시 붙여 넣으세요."
    if isinstance(e, openai.RateLimitError):
        return "429 한도 초과: 너무 자주 보냈거나 사용 한도를 넘었습니다. 잠시 뒤 다시 하세요."
    if isinstance(e, openai.NotFoundError):
        return "404 찾을 수 없음: 모델 이름이 틀렸거나 이 키로 쓸 수 없는 모델입니다."
    if isinstance(e, openai.APIConnectionError):
        return "연결 오류: 인터넷이나 학교·회사의 방화벽을 확인하세요."
    if isinstance(e, openai.OpenAIError):
        return "그 밖의 오류: 키를 못 읽었다면 .env 의 위치와 OPENAI_API_KEY 이름을 확인하세요."
    return f"모르는 오류: {type(e).__name__}"


print("안내할 오류:", ["401", "429", "404", "연결", "그 밖"])

# 일부러 틀린 키(0 을 늘어놓아 지어낸 시험용 글자)를 넣는다
# max_retries=0: 실패해도 다시 보내지 않는다
bad_key = ChatOpenAI(model=MODEL, api_key="sk-test-0000000000000000", max_retries=0)
try:
    bad_key.invoke("안녕하세요")
except Exception as e:
    # langchain-openai 가 바꿔 낸 이름이 찍히지만, openai.AuthenticationError 로도 잡힌다
    print("오류 이름:", type(e).__name__)
    print("상태 코드:", e.status_code)
    print(explain(e))

# 키는 .env 의 진짜 키를 쓰고, 모델 이름에 일부러 i 를 하나 더 붙인다
typo = ChatOpenAI(model="gpt-4.1-minii", max_retries=0)
try:
    typo.invoke("안녕하세요")
except Exception as e:
    print("오류 이름:", type(e).__name__)
    print("상태 코드:", e.status_code)
    print(explain(e))

import time


def ask_with_retry(model, question, tries=3):
    """429 가 나면 1초, 2초 기다렸다가 다시 묻는다. 다른 오류는 안내만 하고 멈춘다."""
    for k in range(1, tries + 1):
        try:
            reply = model.invoke(question)
            print(f"{k}번째 시도에 성공")
            return reply.content
        except openai.RateLimitError as e:
            print(f"{k}번째 시도:", explain(e))
            # 시도할수록 조금씩 더 오래 기다린다(1초, 2초, …)
            time.sleep(k)
        except openai.OpenAIError as e:
            print(explain(e))
            return None
    return None


# 진짜 키와 맞는 모델 이름으로 부른다. 다시 보내기는 우리 함수가 맡으므로 max_retries=0
model = ChatOpenAI(model=MODEL, temperature=0, max_retries=0)
print(ask_with_retry(model, "제주도를 한 문장으로 소개해 줘."))