[LLM마스터클래스]코랩_6_OpenAI의 Langchain을 연결해서 질문하기( API 키 불러오는 3가지 방법 정리 )

Hyejin Beck·2024년 5월 25일
0

AI 서비스

목록 보기
6/8

GPT의 Key 불러오는 3가지 방법
(1) 직접 기재
(2) txt파일로 저장해서 불러오기
(3) 구글코랩: Secrets 이용해서 저장하기

Langchain 설치

!pip install openai langchain 

구글 시크릿으로 API키 저장

OpenAI모델과 API키 연결

# from openai import openai
import openai
import os

# API키 불러오기 방법(1)
# os.environ["OPENAI_API_KEY"] = "여기에 직접 작성한다" 

# API키 불러오기 방법(2)  ---> txt파일
# with open('api키를텍스트파일로저장.txt', 'r') as file:
#    os.environ["OPENAI_API_KEY"] = file.read().strip()

# API키 불러오기 방법(3)  ---> 구글시크릿
# from google.colab import userdata 
# userdata.get('저장한이름')

from google.colab import userdata
os.environ["OPENAI_API_KEY"] = userdata.get('openai')  # openai 라는 이름으로 gpt_api를 저장했다.

OpenAI로 질문하기

!pip install langchain_community
from langchain.llms import OpenAI

openai_api_key = userdata.get('openai') # openai 라는 이름으로 gpt_api를 저장했다.

# OpenAI 모델 선정 
llm = OpenAI(api_key = openai_api_key)

# 모델에게 요청하기
response = llm.generate(["한국말로 된 속담 하나 말해줘"])

# 요청답안 출력하기
print(response)

다른모델 OpenAI로 질문하기

error

참고로 위의 있는 방식으로 OpenAI 모델 선정 코드에
, model = "gpt-4" 추가하면 에러납니다.

번거롭더라도 "gpt-4" 모델 선정해서 OpenAI를 연결할 경우 아래 코드로 입력합니다.

!pip install langchain_openai
from langchain_openai import ChatOpenAI
llm = ChatOpenAI(api_key = openai_api_key, model = "gpt-4")
llm.invoke("한국말로 된 속담 하나 말해줘")

profile
데이터기반 스토리텔링을 통해 인사이트를 얻습니다.

0개의 댓글

관련 채용 정보