[LangChain] 간단한 프로그램 만들어보기

간단한 LangChain 프로그램을 만들어보자.

prompt로 "재미있는 이야기 하나 해줘~"라고 질문하면 응답을 받는 형식이다.

여기서 필요한 것은 아래와 같다.

1. Python 3 설치 (python 3.12)
2. OpenAI + LangChain 패키지 설치 (pipenv 사용)
3. OpenAI API key (ChatGPT API 호출을 위해 필요)
4. 코드 에디터 + 터미널

패키지 설치

우선 Pipfile을 생성하고 langchain-openai를 추가한다.

[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"

[packages]
langchain = "==0.1.16"
openai = "==1.21.2"
langchain-openai = "==0.1.3"

[dev-packages]

[requires]
python_version = "3.12"

그리고 패키지를 설치하자.

pipenv install

만일 pipenv를 찾을 수 없다면 pip install pipenv를 통해 설치를 하자.

프로그램 작성

그리고 main.py를 만든다.

from langchain_openai import OpenAI

api_key = "sk-xxxx" # openai_api_key는 api key이며 미리 openai에서 발급을 받아야 한다.

llm = OpenAI(
    openai_api_key=api_key,
)

result = llm.invoke("짧은 농담 하나 해줘~")
print(result)

실행을 해보자. (python main.py)

"너는 이 세상에서 가장 똑똑한 사람이야. 왜냐하면 너는 내가 좋아하는 사람이니까!"

농담이 어색하긴 하지만 GPT 실행은 잘 되는 것을 확인할 수 있다.

참고

ChatGPT and LangChain: The Complete Developer's Masterclass