Github에 소스코드 올리기 | api-key는 환경변수로 설정

혼자 실습할 때에는 API key를 직접 코드 안에 넣는 방식으로 했는데, 깃허브에 올리거나 외부에 공개할 때에는 API 키를 노출하면 안되기 때문에 환경변수에 저장시키는 방식으로 코드를 바꿨다. 그리고 Openai 버전으로 바꿔서 깃허브에 올려봤는데, 자질구레한 실수들이 있어서 기록

 

1) llm.py 안에 api key를 넣었는데 OPENAI_API_KEY 를 계속 OPEN_API_KEY로 잘못 써서 인식을 못해서 계속 오류남. 이걸 모르고 계속 헛짓거리를.. ^^ 

from langchain_openai import ChatOpenAI
import getpass
import os

def get_api_key():
    if "OPENAI_API_KEY" not in os.environ:
        api_key = getpass.getpass("OpenAI API 키를 입력하세요: ")
        
    else:
        api_key = os.environ["OPENAI_API_KEY"]
    return api_key

def get_chat_model():
    api_key = get_api_key()
    chat = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key = api_key)
    return chat

 

2) 환경변수를 설정하지 않은 경우에. main.py를 실행시키면 터미널 창안에

OpenAI API 키를 입력하세요:

가 뜨기는 하는데 문제는 바로 입력이 안됨. 엔터인가 스페이스바를 한번 친 다음에 입력할 수 있기도 한데, 복사한 키를 붙여넣기 하면 또 갑자기 오류난다. api키를 따로 파일로 저장해서 그걸 불러오는 방식을 써도 된다는데 그럴 바에야 그냥 환경변수에 입력해 놓는 게 낫지.

환경변수와 관련된 명령어들(맥)

환경변수 설정

 export OPENAI_API_KEY="your-api-key-here"
 
 source ~/.bash_profile  # 또는 해당하는 프로필 파일

환경변수 확인

echo $OPENAI_API_KEY

현재 세션에서 환경변수 삭제

unset OPENAI_API_KEY

 

아래는 환경변수 처리해서 올린 RAG 실습한 모듈 파일들

https://github.com/hannakhw/RAG_practice/tree/main

GitHub - hannakhw/RAG_practice